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

Как добавить функциональность к существующему методу стороннего объекта?

Я не уверен, что вопрос сформулирован правильно, поэтому вот пример того, чего я хочу достичь.

Для мигания кнопки используется следующая функция:

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, но оставить существующий метод нетронутым.

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


python pyqt5 qpushbutton pyqt overriding
person Soundwave    schedule 12.01.2017    source источник

Ответы (1)


arrow_upward
1
arrow_downward

Вы можете создать собственный подкласс, а затем использовать его вместо обычного 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_())
person ekhumoro    schedule 12.01.2017