PyQt5 QColorDialog – 为其子按钮设置背景色

当我们在使用Python编写GUI程序时，颜色选择框(QColorDialog)是处理颜色相关任务的重要部分。PyQt5中的QColorDialog提供了许多功能，其中设置子按钮背景色是非常常用且重要的。在这里，我将介绍如何使用PyQt5 QColorDialog为其子按钮设置背景色的完整使用攻略。

准备工作

首先要确保你已经安装了PyQt5库，如果没有请使用以下命令进行安装：

pip install PyQt5

在安装完PyQt5库以后，就可以使用如下代码导入相关库：

from PyQt5.QtCore import Qt
from PyQt5.QtGui import QColor
from PyQt5.QtWidgets import QApplication, QColorDialog, QWidget, QVBoxLayout, QPushButton

使用QColorDialog为子按钮设置背景色

下面，我们将演示如何使用QColorDialog为子按钮设置背景色。首先，我们需要创建一个QWidget和一个QPushButton。使用以下代码实现：

class Example(QWidget):

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

        self.initUI()

    def initUI(self):

        layout = QVBoxLayout()
        self.button = QPushButton('Button', self)
        layout.addWidget(self.button)

        self.setLayout(layout)

        self.setGeometry(300, 300, 300, 200)
        self.setWindowTitle('Color Dialog')
        self.show()

if __name__ == '__main__':
    app = QApplication(sys.argv)
    ex = Example()
    sys.exit(app.exec_())

接着，使用以下代码添加按钮单击事件和弹出颜色选择框：

def showDialog(self):

    col = QColorDialog.getColor()

    if col.isValid():
        self.button.setStyleSheet("background-color: {}".format(col.name()))

self.button.clicked.connect(self.showDialog)

QColorDialog.getColor()方法会返回选择的颜色。使用isValid()方法检查颜色的有效性。如果颜色有效，就使用按钮的setStyleSheet()方法设置背景色。

完整的代码如下：

import sys
from PyQt5.QtCore import Qt
from PyQt5.QtGui import QColor
from PyQt5.QtWidgets import QApplication, QColorDialog, QWidget, QVBoxLayout, QPushButton

class Example(QWidget):

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

        self.initUI()

    def initUI(self):

        layout = QVBoxLayout()
        self.button = QPushButton('Button', self)
        layout.addWidget(self.button)

        self.setLayout(layout)

        self.setGeometry(300, 300, 300, 200)
        self.setWindowTitle('Color Dialog')
        self.show()

        self.button.clicked.connect(self.showDialog)

    def showDialog(self):

        col = QColorDialog.getColor()

        if col.isValid():
            self.button.setStyleSheet("background-color: {}".format(col.name()))

if __name__ == '__main__':
    app = QApplication(sys.argv)
    ex = Example()
    sys.exit(app.exec_())

现在运行程序，单击按钮，就会出现颜色选择框，选择颜色后子按钮背景色就会相应改变。

自定义子按钮的颜色

除了使用系统颜色选择框外，我们还可以自定义子按钮的颜色。例如，我们可以添加另一个按钮，让用户自己定义颜色，再根据选择的颜色设置子按钮的背景色。

使用以下代码创建一个自定义颜色按钮：

class Example(QWidget):

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

        self.initUI()

    def initUI(self):

        layout = QVBoxLayout()
        self.button = QPushButton('Button', self)
        layout.addWidget(self.button)

        self.color_btn = QPushButton('Color', self)
        layout.addWidget(self.color_btn)

        self.setLayout(layout)

        self.setGeometry(300, 300, 300, 200)
        self.setWindowTitle('Color Dialog')
        self.show()

        self.button.clicked.connect(self.showDialog)
        self.color_btn.clicked.connect(self.customColorDialog)

    def customColorDialog(self):

        col = QColorDialog.getColor()

        if col.isValid():
            self.color = col

在customColorDialog()函数中，使用QColorDialog.getColor()方法获取用户选择的颜色，并将其存储在self.color变量中。

然后，我们需要更新我们的showDialog()函数，以便它能够使用self.color变量设置子按钮的背景色：

def showDialog(self):

    if hasattr(self, 'color'):
        self.button.setStyleSheet("background-color: {}".format(self.color.name()))
    else:
        col = QColorDialog.getColor()

        if col.isValid():
            self.button.setStyleSheet("background-color: {}".format(col.name()))

由于我们的自定义颜色按钮可以在弹出颜色选择框之前被单击，所以我们需要使用hasattr()函数检查self.color变量是否已定义。如果已经定义，就使用该颜色设置子按钮的背景色。否则，我们使用上文中的弹出颜色选择框。

完整的代码如下：

import sys
from PyQt5.QtCore import Qt
from PyQt5.QtGui import QColor
from PyQt5.QtWidgets import QApplication, QColorDialog, QWidget, QVBoxLayout, QPushButton

class Example(QWidget):

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

        self.initUI()

    def initUI(self):

        layout = QVBoxLayout()
        self.button = QPushButton('Button', self)
        layout.addWidget(self.button)

        self.color_btn = QPushButton('Color', self)
        layout.addWidget(self.color_btn)

        self.setLayout(layout)

        self.setGeometry(300, 300, 300, 200)
        self.setWindowTitle('Color Dialog')
        self.show()

        self.button.clicked.connect(self.showDialog)
        self.color_btn.clicked.connect(self.customColorDialog)

    def customColorDialog(self):

        col = QColorDialog.getColor()

        if col.isValid():
            self.color = col

    def showDialog(self):

        if hasattr(self, 'color'):
            self.button.setStyleSheet("background-color: {}".format(self.color.name()))
        else:
            col = QColorDialog.getColor()

            if col.isValid():
                self.button.setStyleSheet("background-color: {}".format(col.name()))

if __name__ == '__main__':
    app = QApplication(sys.argv)
    ex = Example()
    sys.exit(app.exec_())

现在运行程序，单击"Color"按钮，选择自定义颜色，再单击"Button"按钮，就可以将子按钮的背景色设置为所选颜色了。