ZetCode

PyQt6 中的自定义小部件

最后修改于 2023 年 1 月 10 日

PyQt6 拥有一套丰富的小部件。然而,没有任何工具包可以提供程序员在其应用程序中可能需要的所有小部件。工具包通常只提供最常见的小部件,例如按钮、文本小部件或滑块。如果需要更专业的小部件,我们必须自己创建它。

自定义小部件是使用工具包提供的绘图工具创建的。有两种基本可能性:程序员可以修改或增强现有的部件,或者他可以从头开始创建自定义小部件。

PyQt6 烧录小部件

这是我们可以在 Nero、K3B 或其他 CD/DVD 烧录软件中看到的小部件。

burning_widget.py
#!/usr/bin/python

"""
ZetCode PyQt6 tutorial

In this example, we create a custom widget.

Author: Jan Bodnar
Website: zetcode.com
"""

from PyQt6.QtWidgets import (QWidget, QSlider, QApplication,
        QHBoxLayout, QVBoxLayout)
from PyQt6.QtCore import QObject, Qt, pyqtSignal
from PyQt6.QtGui import QPainter, QFont, QColor, QPen
import sys


class Communicate(QObject):
    updateBW = pyqtSignal(int)


class BurningWidget(QWidget):

    def __init__(self):
        super().__init__()

        self.initUI()


    def initUI(self):

        self.setMinimumSize(1, 30)
        self.value = 75
        self.num = [75, 150, 225, 300, 375, 450, 525, 600, 675]


    def setValue(self, value):

        self.value = value


    def paintEvent(self, e):

        qp = QPainter()
        qp.begin(self)
        self.drawWidget(qp)
        qp.end()


    def drawWidget(self, qp):

        MAX_CAPACITY = 700
        OVER_CAPACITY = 750

        font = QFont('Serif', 7, QFont.Weight.Light)
        qp.setFont(font)

        size = self.size()
        w = size.width()
        h = size.height()

        step = int(round(w / 10))

        till = int(((w / OVER_CAPACITY) * self.value))
        full = int(((w / OVER_CAPACITY) * MAX_CAPACITY))

        if self.value >= MAX_CAPACITY:

            qp.setPen(QColor(255, 255, 255))
            qp.setBrush(QColor(255, 255, 184))
            qp.drawRect(0, 0, full, h)
            qp.setPen(QColor(255, 175, 175))
            qp.setBrush(QColor(255, 175, 175))
            qp.drawRect(full, 0, till - full, h)

        else:

            qp.setPen(QColor(255, 255, 255))
            qp.setBrush(QColor(255, 255, 184))
            qp.drawRect(0, 0, till, h)

        pen = QPen(QColor(20, 20, 20), 1,
                   Qt.PenStyle.SolidLine)

        qp.setPen(pen)
        qp.setBrush(Qt.BrushStyle.NoBrush)
        qp.drawRect(0, 0, w - 1, h - 1)

        j = 0

        for i in range(step, 10 * step, step):

            qp.drawLine(i, 0, i, 5)
            metrics = qp.fontMetrics()
            fw = metrics.horizontalAdvance(str(self.num[j]))

            x, y = int(i - fw/2), int(h / 2)
            qp.drawText(x, y, str(self.num[j]))
            j = j + 1


class Example(QWidget):

    def __init__(self):
        super().__init__()

        self.initUI()


    def initUI(self):

        OVER_CAPACITY = 750

        sld = QSlider(Qt.Orientations.Horizontal, self)
        sld.setFocusPolicy(Qt.FocusPolicy.NoFocus)
        sld.setRange(1, OVER_CAPACITY)
        sld.setValue(75)
        sld.setGeometry(30, 40, 150, 30)

        self.c = Communicate()
        self.wid = BurningWidget()
        self.c.updateBW[int].connect(self.wid.setValue)

        sld.valueChanged[int].connect(self.changeValue)
        hbox = QHBoxLayout()
        hbox.addWidget(self.wid)
        vbox = QVBoxLayout()
        vbox.addStretch(1)
        vbox.addLayout(hbox)
        self.setLayout(vbox)

        self.setGeometry(300, 300, 390, 210)
        self.setWindowTitle('Burning widget')
        self.show()


    def changeValue(self, value):

        self.c.updateBW.emit(value)
        self.wid.repaint()


def main():

    app = QApplication(sys.argv)
    ex = Example()
    sys.exit(app.exec())


if __name__ == '__main__':
    main()

在我们的例子中,我们有一个 QSlider 和一个自定义小部件。滑块控制自定义小部件。此小部件以图形方式显示介质的总容量和可供我们使用的可用空间。我们自定义小部件的最小值为 1,最大值为 OVER_CAPACITY。如果我们达到值 MAX_CAPACITY,我们开始用红色绘制。这通常表示超刻录。

烧录小部件放置在窗口的底部。这是使用一个 QHBoxLayout 和一个 QVBoxLayout 实现的。

class BurningWidget(QWidget):

    def __init__(self):
        super().__init__()

烧录小部件基于 QWidget 小部件。

self.setMinimumSize(1, 30)

我们更改部件的最小大小(高度)。默认值对我们来说有点小。

font = QFont('Serif', 7, QFont.Weight.Light)
qp.setFont(font)

我们使用比默认字体更小的字体。这更适合我们的需求。

size = self.size()
w = size.width()
h = size.height()

step = int(round(w / 10))


till = int(((w / OVER_CAPACITY) * self.value))
full = int(((w / OVER_CAPACITY) * MAX_CAPACITY))

我们动态绘制小部件。窗口越大,烧录小部件也越大,反之亦然。这就是为什么我们必须计算绘制自定义小部件的小部件的大小。 till 参数决定要绘制的总大小。此值来自滑块小部件。它是整个区域的一部分。 full 参数决定我们开始用红色绘制的点。

实际的绘制包括三个步骤。我们绘制黄色或红色和黄色的矩形。然后,我们绘制将小部件分成几个部分的垂直线。最后,我们绘制指示介质容量的数字。

metrics = qp.fontMetrics()
fw = metrics.horizontalAdvance(str(self.num[j]))

x, y = int(i - fw/2), int(h / 2)
qp.drawText(x, y, str(self.num[j]))

我们使用字体度量来绘制文本。我们必须知道文本的宽度才能将其居中于垂直线周围。

def changeValue(self, value):

    self.c.updateBW.emit(value)
    self.wid.repaint()

当我们移动滑块时,会调用 changeValue 方法。在方法内部,我们发送一个带有参数的自定义 updateBW 信号。该参数是滑块的当前值。该值稍后用于计算要绘制的烧录小部件的容量。然后重新绘制自定义小部件。

The burning widget
图:烧录小部件

在本 PyQt6 教程中,我们创建了一个自定义小部件。