PyQt – 如何使用QItemDelegate在表格视图中设置QComboBox

在本文中，我们将介绍如何使用PyQt中的QItemDelegate来在表格视图中设置QComboBox。QComboBox是一个下拉列表框，它可以用于展示一个预定义的选项列表，用户可以从中选择一个选项。

阅读更多：PyQt 教程

1. QItemDelegate简介

QItemDelegate是PyQt中的一个重要类，在表格视图中进行单元格的绘制和编辑时起到关键的作用。通过使用QItemDelegate，我们可以定制单元格的外观和编辑行为，例如将QComboBox添加到单元格中。

2. 在表格视图中设置QComboBox

为了在表格视图中设置QComboBox，我们需要将QComboBox添加到QItemDelegate的paint()方法中，并根据当前单元格的值设置所选项。下面是一个示例代码：

from PyQt5.QtWidgets import QComboBox, QItemDelegate

class ComboBoxDelegate(QItemDelegate):
    def __init__(self, items):
        super().__init__()
        self.items = items

    def createEditor(self, parent, option, index):
        editor = QComboBox(parent)
        editor.addItems(self.items)
        return editor

    def setEditorData(self, editor, index):
        value = index.model().data(index, role=Qt.EditRole)
        editor.setCurrentText(value)

    def setModelData(self, editor, model, index):
        model.setData(index, editor.currentText(), role=Qt.EditRole)

    def updateEditorGeometry(self, editor, option, index):
        editor.setGeometry(option.rect)

在上面的代码中，我们定义了一个ComboBoxDelegate类，继承自QItemDelegate。在createEditor()方法中，我们创建了一个QComboBox并添加了预定义的选项列表。在setEditorData()方法中，我们将当前单元格的值设置为QComboBox的当前选项。在setModelData()方法中，我们将所选的选项设置为模型的数据。

接下来，我们需要在表格视图中使用这个QItemDelegate来设置QComboBox。下面是一个示例代码：

from PyQt5.QtWidgets import QTableView

items = ['Item 1', 'Item 2', 'Item 3']

table_view = QTableView()
model = QStandardItemModel()
delegate = ComboBoxDelegate(items)

table_view.setItemDelegate(delegate)
table_view.setModel(model)

在上面的代码中，我们创建了一个QTableView和一个QStandardItemModel。然后，我们实例化了我们自定义的ComboBoxDelegate并将其设置为表格视图的项代理。最后，我们将模型设置为表格视图的模型。

现在，当我们运行这段代码时，我们将在表格视图中看到一个QComboBox，并且我们可以从选项列表中选择一个选项。

3. 自定义QComboBox的外观

我们可以通过自定义QComboBox的外观来改变其在表格视图中的显示方式。例如，我们可以更改QComboBox的背景色、前景色、字体等。下面是一个示例代码：

from PyQt5.QtGui import QFont, QColor, QPainter
from PyQt5.QtCore import Qt
from PyQt5.QtWidgets import QStyleOptionComboBox

class CustomComboBoxDelegate(ComboBoxDelegate):
    def paint(self, painter, option, index):
        rect = option.rect
        value = index.model().data(index, role=Qt.DisplayRole)

        # 绘制背景
        painter.save()
        if option.state & QStyleOptionComboBox.State_Enabled:
            if option.state & QStyleOptionComboBox.State_Active:
                bg_color = QColor(255, 255, 200)
            else:
                bg_color = QColor(255, 255, 255)
        else:
            bg_color = QColor(200, 200, 200)
        painter.fillRect(rect, bg_color)
        painter.restore()

        # 绘制文本
        painter.save()
        font = QFont("Arial", 10)
        painter.setFont(font)
        painter.drawText(rect, Qt.AlignCenter, value)
        painter.restore()

在上面的代码中，我们创建了一个CustomComboBoxDelegate类，继承自之前的ComboBoxDelegate。在paint()方法中，我们重写了绘制CustomComboBoxDelegate的逻辑。在绘制背景时，我们根据QStyleOptionComboBox的状态来确定背景色，并使用QPainter将矩形填充为指定的颜色。在绘制文本时，我们设置了字体和对齐方式，并使用QPainter.drawText()方法将文本绘制在矩形中心位置。

为了使用CustomComboBoxDelegate，我们可以按照之前的步骤将其设置为表格视图的项代理：

table_view = QTableView()
model = QStandardItemModel()
delegate = CustomComboBoxDelegate(items)

table_view.setItemDelegate(delegate)
table_view.setModel(model)

现在，当我们运行这段代码时，我们将在表格视图中看到一个自定义外观的QComboBox。