重载开关按钮构造函数

qfluentwidgets/components/widgets/switch_button.py

@@ -6,6 +6,7 @@ from PyQt5.QtGui import QColor, QPainter
 from PyQt5.QtWidgets import QHBoxLayout, QLabel, QToolButton, QWidget
 
 from ...common.style_sheet import FluentStyleSheet
+from ...common.overload import singledispatchmethod
 
 
 class Indicator(QToolButton):
@@ -129,13 +130,11 @@ class SwitchButton(QWidget):
 
     checkedChanged = pyqtSignal(bool)
 
-    def __init__(self, text='Off', parent=None, indicatorPos=IndicatorPosition.LEFT):
+    @singledispatchmethod
+    def __init__(self, parent: QWidget = None, indicatorPos=IndicatorPosition.LEFT):
         """
         Parameters
         ----------
-        text: str
-            the text of switch button
-
         parent: QWidget
             parent widget
 
@@ -143,14 +142,31 @@ class SwitchButton(QWidget):
             the position of indicator
         """
         super().__init__(parent=parent)
-        self.text = text
+        self.text = 'Off'
         self.__spacing = 12
         self.indicatorPos = indicatorPos
         self.hBox = QHBoxLayout(self)
         self.indicator = Indicator(self)
-        self.label = QLabel(text, self)
+        self.label = QLabel(self.text, self)
         self.__initWidget()
 
+    @__init__.register
+    def _(self, text: str = 'Off', parent: QWidget = None, indicatorPos=IndicatorPosition.LEFT):
+        """
+        Parameters
+        ----------
+        text: str
+            the text of switch button
+
+        parent: QWidget
+            parent widget
+
+        indicatorPosition: IndicatorPosition
+            the position of indicator
+        """
+        self.__init__(parent, indicatorPos)
+        self.setText(text)
+
     def __initWidget(self):
         """ initialize widgets """
         self.setAttribute(Qt.WA_StyledBackground)