From a3e3e56f2b7d1e1ea0693d3ae6d3da7b1c7ea3b7 Mon Sep 17 00:00:00 2001
From: RumovZ <gp5glkw78@relay.firefox.com>
Date: Mon, 29 Mar 2021 12:19:48 +0200
Subject: [PATCH] Implement custom qt button Switch

---
 qt/aqt/switch.py | 115 +++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 115 insertions(+)
 create mode 100644 qt/aqt/switch.py

diff --git a/qt/aqt/switch.py b/qt/aqt/switch.py
new file mode 100644
index 000000000..d1a883b44
--- /dev/null
+++ b/qt/aqt/switch.py
@@ -0,0 +1,115 @@
+# -*- coding: utf-8 -*-
+# Copyright: Ankitects Pty Ltd and contributors
+# License: GNU AGPL, version 3 or later; http://www.gnu.org/licenses/agpl.html
+
+from aqt import colors
+from aqt.qt import *
+from aqt.theme import theme_manager
+
+
+class Switch(QAbstractButton):
+    """A horizontal slider to toggle between two states which can be denoted by short strings.
+    The left state is the default and corresponds to isChecked=True.
+    """
+
+    _margin: int = 2
+
+    def __init__(
+        self,
+        radius: int = 10,
+        left_label: str = "",
+        right_label: str = "",
+        parent: QWidget = None,
+    ) -> None:
+        super().__init__(parent=parent)
+        self.setCheckable(True)
+        super().setChecked(True)
+        self.setSizePolicy(QSizePolicy.Fixed, QSizePolicy.Fixed)
+        self._left_label = left_label
+        self._right_label = right_label
+        self._path_radius = radius
+        self._knob_radius = radius - self._margin
+        self._left_position = self._position = self._path_radius + self._margin
+        self._right_position = 3 * self._path_radius + self._margin
+
+    @pyqtProperty(int)  # type: ignore
+    def position(self) -> int:
+        return self._position
+
+    @position.setter  # type: ignore
+    def position(self, position: int) -> None:
+        self._position = position
+        self.update()
+
+    @property
+    def start_position(self) -> int:
+        return self._right_position if self.isChecked() else self._left_position
+
+    @property
+    def end_position(self) -> int:
+        return self._left_position if self.isChecked() else self._right_position
+
+    @property
+    def label(self) -> str:
+        return self._left_label if self.isChecked() else self._right_label
+
+    def sizeHint(self) -> QSize:
+        return QSize(
+            4 * self._path_radius + 2 * self._margin,
+            2 * self._path_radius + 2 * self._margin,
+        )
+
+    def setChecked(self, checked: bool) -> None:
+        super().setChecked(checked)
+        self._position = self.end_position
+        self.update()
+
+    def paintEvent(self, _event: QPaintEvent) -> None:
+        painter = QPainter(self)
+        painter.setRenderHint(QPainter.Antialiasing, True)
+        painter.setPen(Qt.NoPen)
+        self._paint_path(painter)
+        self._paint_knob(painter)
+        self._paint_label(painter)
+
+    def _paint_path(self, painter: QPainter) -> None:
+        painter.setBrush(QBrush(theme_manager.qcolor(colors.FRAME_BG)))
+        rectangle = QRectF(
+            self._margin,
+            self._margin,
+            self.width() - 2 * self._margin,
+            self.height() - 2 * self._margin,
+        )
+        painter.drawRoundedRect(rectangle, self._path_radius, self._path_radius)
+
+    def _current_knob_rectangle(self) -> QRectF:
+        return QRectF(
+            self.position - self._knob_radius,  # type: ignore
+            2 * self._margin,
+            2 * self._knob_radius,
+            2 * self._knob_radius,
+        )
+
+    def _paint_knob(self, painter: QPainter) -> None:
+        painter.setBrush(QBrush(theme_manager.qcolor(colors.HIGHLIGHT_BG)))
+        painter.drawEllipse(self._current_knob_rectangle())
+
+    def _paint_label(self, painter: QPainter) -> None:
+        painter.setPen(QColor("white"))
+        font = painter.font()
+        font.setPixelSize(int(1.5 * self._knob_radius))
+        painter.setFont(font)
+        painter.drawText(self._current_knob_rectangle(), Qt.AlignCenter, self.label)
+
+    def mouseReleaseEvent(self, event: QMouseEvent) -> None:
+        super().mouseReleaseEvent(event)
+        if event.button() == Qt.LeftButton:
+            animation = QPropertyAnimation(self, b"position", self)
+            animation.setDuration(100)
+            animation.setStartValue(self.start_position)
+            animation.setEndValue(self.end_position)
+            animation.start()
+
+    def enterEvent(self, event: QEvent) -> None:
+        self.setCursor(Qt.PointingHandCursor)
+        super().enterEvent(event)