Добавить функциональность в существующий метод сторонней библиотеки

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())

Я хочу, чтобы каждый QPushButton библиотеки PyQt5 мигал при нажатии.

Моя мысль - добавить функцию clickColor в метод clicked.connect QPushButton, но оставить существующий метод без изменений.

Как правильно сделать то, чего я хочу достичь?

Answer 1

Вы можете создать собственный подкласс, а затем использовать его вместо обычного QPushButton везде, где вы хотите эффект моргания. Если вы используете Qt Designer, вы также можете использовать продвижение виджетов, чтобы заменить любые кнопки, добавленные в файл пользовательского интерфейса, на ваш собственный класс (см. Этот ответ для более подробной информации).

Вот основной демонстрационный скрипт:

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_())