Question

如何向第三方对象的现有方法添加功能？

我不确定问题是否正确表达，所以这是我想要实现的一个例子。

以下功能用于闪烁按钮：

def clickColor(button, color):
    beforeColor = button.palette().color(QPalette.Background)
    button.setStyleSheet("background-color: %s" % color)
    QTimer.singleShot(100, lambda: unClickColor(button, beforeColor))

def unClickColor(button, beforeColor):
    button.setStyleSheet("background-color: %s" % beforeColor.name())

我想让PyQt5库的每个QPushButton在点击时闪烁。

我的想法是将clickColor函数添加到QPushButton的 clicked.connect 方法，但保持现有方法不变。

做我想做的事情的正确方法是什么？

Answer 1

你可以创建一个自定义子类，然后在任何你想要闪烁效果的地方使用它而不是普通的QPushButton。如果您正在使用Qt Designer，您还可以使用widget promotion将自定义类替换添加到ui文件的所有按钮（有关详细信息，请参阅this answer）。

这是一个基本的演示脚本：

import sys
from PyQt5 import QtCore, QtGui, QtWidgets

class BlinkButton(QtWidgets.QPushButton):
    def __init__(self, *args, **kwargs):
        super(BlinkButton, self).__init__(*args, **kwargs)
        self.clicked.connect(self._blink)
        self._blink_color = QtGui.QColor()

    def blinkColor(self):
        return QtGui.QColor(self._blink_color)

    def setBlinkColor(self, color=None):
        self._blink_color = QtGui.QColor(color)

    def _blink(self):
        if self._blink_color.isValid():
            self.setStyleSheet(
                'background-color: %s' % self._blink_color.name())
            QtCore.QTimer.singleShot(100, lambda: self.setStyleSheet(''))

class Window(QtWidgets.QWidget):
    def __init__(self):
        super(Window, self).__init__()
        self.button = BlinkButton('Test', self)
        self.button.setBlinkColor('red')
        self.button.clicked.connect(self.handleButton)
        layout = QtWidgets.QVBoxLayout(self)
        layout.addWidget(self.button)
        layout.addWidget(self.button2)

    def handleButton(self):
        print('Hello World!')

if __name__ == '__main__':

    app = QtWidgets.QApplication(sys.argv)
    window = Window()
    window.setGeometry(600, 100, 200, 100)
    window.show()
    sys.exit(app.exec_())

为第三方库的现有方法添加功能

1 个答案: