127 lines
3.6 KiB
Python
127 lines
3.6 KiB
Python
|
from PySide6.QtWidgets import (
|
|||
|
|
QAbstractButton,
|
|||
|
|
QApplication,
|
|||
|
|
QSizePolicy,
|
|||
|
|
QWidget,
|
|||
|
|
QVBoxLayout,
|
|||
|
|
)
|
|||
|
|
from PySide6.QtCore import Qt, QRect, Signal
|
|||
|
|
from PySide6.QtGui import QPainter, QBrush
|
|||
|
|
|
|||
|
|
|
|||
|
|
class SwitchButton(QAbstractButton):
|
|||
|
|
# 开关切换信号
|
|||
|
|
# 开: True, 关: False
|
|||
|
|
switched = Signal(bool)
|
|||
|
|
|
|||
|
|
def __init__(self, parent=None):
|
|||
|
|
super().__init__(parent)
|
|||
|
|
self._checked = False
|
|||
|
|
|
|||
|
|
self.setCursor(Qt.CursorShape.PointingHandCursor)
|
|||
|
|
|
|||
|
|
# 颜色映射
|
|||
|
|
self.color_keymap = {
|
|||
|
|
"slider": "#16ffff",
|
|||
|
|
"text": Qt.GlobalColor.white,
|
|||
|
|
"on_bg": "#008ee8", # 开的时候的背景颜色
|
|||
|
|
"off_bg": "#001c83", # 关的时候的背景颜色
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
self.clicked.connect(self.onClicked)
|
|||
|
|
self._init_style() # 新尺寸的样式表
|
|||
|
|
self._init_size_policy()
|
|||
|
|
|
|||
|
|
def _init_style(self):
|
|||
|
|
self.setStyleSheet(
|
|||
|
|
"""
|
|||
|
|
SwitchButton {
|
|||
|
|
font-family: "Microsoft YaHei";
|
|||
|
|
font-size: 11px;
|
|||
|
|
color: white;
|
|||
|
|
border-radius: 9px;
|
|||
|
|
margin: 2px;
|
|||
|
|
min-width: 39px;
|
|||
|
|
max-width: 39px;
|
|||
|
|
min-height: 18px;
|
|||
|
|
max-height: 18px;
|
|||
|
|
}
|
|||
|
|
"""
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
def _init_size_policy(self):
|
|||
|
|
self.setSizePolicy(QSizePolicy.Policy.Fixed, QSizePolicy.Policy.Fixed)
|
|||
|
|
|
|||
|
|
def paintEvent(self, event):
|
|||
|
|
# 1. 调整滑块间距
|
|||
|
|
slider_space = 1 # 滑块与背景的间距（上下左右各1px）
|
|||
|
|
painter = QPainter(self)
|
|||
|
|
painter.setRenderHint(QPainter.RenderHint.Antialiasing, True)
|
|||
|
|
|
|||
|
|
try:
|
|||
|
|
# 2. 绘制背景
|
|||
|
|
background_rect = self.rect()
|
|||
|
|
painter.setPen(Qt.PenStyle.NoPen)
|
|||
|
|
bg_color = (
|
|||
|
|
self.color_keymap["on_bg"]
|
|||
|
|
if self._checked
|
|||
|
|
else self.color_keymap["off_bg"]
|
|||
|
|
)
|
|||
|
|
painter.setBrush(QBrush(bg_color))
|
|||
|
|
painter.drawRoundedRect(
|
|||
|
|
background_rect, self.height() / 2, self.height() / 2
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
# 3. 计算新滑块尺寸
|
|||
|
|
slider_width = self.height() - slider_space * 2
|
|||
|
|
|
|||
|
|
# 4. 计算新滑块位置
|
|||
|
|
if self._checked:
|
|||
|
|
slider_x = self.width() - slider_width - slider_space
|
|||
|
|
else:
|
|||
|
|
slider_x = slider_space
|
|||
|
|
slider_y = slider_space
|
|||
|
|
|
|||
|
|
# 5. 绘制滑块
|
|||
|
|
slider_rect = QRect(slider_x, slider_y, slider_width, slider_width)
|
|||
|
|
painter.setBrush(QBrush(self.color_keymap["slider"]))
|
|||
|
|
painter.drawEllipse(slider_rect)
|
|||
|
|
finally:
|
|||
|
|
painter.end()
|
|||
|
|
|
|||
|
|
def onClicked(self):
|
|||
|
|
self._checked = not self._checked
|
|||
|
|
self.update()
|
|||
|
|
self.switched.emit(self._checked)
|
|||
|
|
|
|||
|
|
def setChecked(self, checked: bool):
|
|||
|
|
self._checked = checked
|
|||
|
|
self.update()
|
|||
|
|
self.switched.emit(self._checked)
|
|||
|
|
|
|||
|
|
def isChecked(self) -> bool:
|
|||
|
|
return self._checked
|
|||
|
|
|
|||
|
|
|
|||
|
|
# 测试示例
|
|||
|
|
if __name__ == "__main__":
|
|||
|
|
import sys
|
|||
|
|
|
|||
|
|
app = QApplication(sys.argv)
|
|||
|
|
|
|||
|
|
test_window = QWidget()
|
|||
|
|
test_window.setWindowTitle("开关测试")
|
|||
|
|
test_window.resize(200, 100)
|
|||
|
|
|
|||
|
|
layout = QVBoxLayout(test_window)
|
|||
|
|
layout.setContentsMargins(50, 30, 50, 30)
|
|||
|
|
|
|||
|
|
switch = SwitchButton()
|
|||
|
|
switch.switched.connect(
|
|||
|
|
lambda state: print(f"开关状态：{'选中' if state else '未选中'}")
|
|||
|
|
)
|
|||
|
|
layout.addWidget(switch)
|
|||
|
|
|
|||
|
|
test_window.show()
|
|||
|
|
sys.exit(app.exec())
|