问创建自定义时间选择器小部件

EN

from PyQt5 import QtWidgets, QtGui, QtCore
import math, sys


class ClockWidget(QtWidgets.QWidget):   # I want to be able to reuse this class for other programs also, so please don't hard code values of the list, start and end

    def __init__(self, start, end, lst=[], *args, **kwargs):
        super(ClockWidget, self).__init__(*args, **kwargs)

        self.lst = lst

        if not self.lst:
            self.lst = [*range(start, end)]

        self.index_start = 0  # tune this to move the letters in the circle
        self.pointer_angles_multiplier = 9  # just setting the default values

        self.current = None
        self.rects = []

    @property
    def index_start(self):
        return self._index_start

    @index_start.setter
    def index_start(self, index):
        self._index_start = index

    def paintEvent(self, event):

        self.rects = []

        painter = QtGui.QPainter(self)
        pen = QtGui.QPen()
        pen.setColor(QtCore.Qt.red)
        pen.setWidth(2)
        painter.setPen(pen)

        x, y = self.rect().x(), self.rect().y()
        width, height = self.rect().width(), self.rect().height()

        painter.drawEllipse(x, y, x + width, x + height)

        s, t, equal_angles, radius = self.angle_calc()
        radius -= 30

        pen.setColor(QtCore.Qt.green)
        pen.setWidth(2)
        painter.setPen(pen)

        """ pointer angle helps in determining to which position the pointer should be drawn"""
        self.pointer_x, self.pointer_y = s + ((radius-30) * math.cos(self.pointer_angles_multiplier * equal_angles)), t \
                + ((radius-30)  * math.sin(self.pointer_angles_multiplier * equal_angles))

        """ The pendulum like pointer """
        painter.drawLine(QtCore.QPointF(s, t), QtCore.QPointF(self.pointer_x, self.pointer_y))

        painter.drawEllipse(QtCore.QRectF(QtCore.QPointF(self.pointer_x - 20, self.pointer_y - 40),
                                          QtCore.QPointF(self.pointer_x + 30, self.pointer_y + 10)))

        pen.setColor(QtCore.Qt.blue)
        pen.setWidth(3)

        font = self.font()
        font.setPointSize(14)
        painter.setFont(font)

        painter.setPen(pen)

        """ Drawing the number around the circle formula y = t + radius * cos(a)
            y = s + radius * sin(a) where angle is in radians (s, t) are the mid point of the circle """
        for index, char in enumerate(self.lst, start=self.index_start):
            angle = equal_angles * index

            y = t + radius * math.sin(angle)
            x = s + radius * math.cos(angle)

            # print(f"Add: {add_x}, index: {index}; char: {char}")
            rect = QtCore.QRectF(x - 30, y - 40, x + 60, y)  # clickable point

            self.rects.append([index, char, rect])  # appends index, letter, rect

            painter.setPen(QtCore.Qt.blue)
            painter.drawRect(rect)  # helps in visualizing the points where the click can received

            print(f"Rect: {rect}; char: {char}")
            painter.setPen(QtCore.Qt.red)

            points = QtCore.QPointF(x, y)
            painter.drawText(points, str(char))

    def mousePressEvent(self, event):

        for x in self.rects:

            index, char, rect = x

            if event.button() & QtCore.Qt.LeftButton and rect.contains(event.pos()):

                self.pointer_angles_multiplier = index
                self.current = char
                self.update()
                break

    def angle_calc(self):
        """
         This will simply return (midpoints of circle, divides a circle into the len(list) and return the
         angle in radians, radius)
         """
        return ((self.rect().width() - self.rect().x()) / 2, (self.rect().height() - self.rect().y()) / 2,
                (360 / len(self.lst)) * (math.pi / 180), (self.rect().width() / 2))

    def resizeEvent(self, event: QtGui.QResizeEvent):
        """This is supposed to maintain a Square aspect ratio on widget resizing but doesn't work
        correctly as you will see when executing"""

        if event.size().width() > event.size().height():
            self.resize(event.size().height(), event.size().width())

        else:
            self.resize(event.size().width(), event.size().width())


if __name__ == '__main__':
    app = QtWidgets.QApplication(sys.argv)

    message = ClockWidget(1, 13)
    message.index_start = 10
    message.show()

    sys.exit(app.exec())