PyQt5：在QTableView中插入小部件

在本文中，我们将介绍如何在PyQt5的QTableView（表视图）中插入小部件。QTableView是PyQt5中用于显示二维数据的控件，我们可以在其中显示文本、图像等内容。然而，在某些情况下，我们可能需要在特定的单元格中插入自定义小部件，以提供更丰富的交互和功能。

阅读更多：PyQt5 教程

使用委托进行小部件插入

在QTableView中插入小部件的一种常见方法是使用委托（Delegate）。委托允许我们自定义每个单元格的外观和行为，以便可以在其中插入小部件。

首先，我们需要创建一个自定义的委托类，继承自QItemDelegate。然后，我们需要重写委托类的createEditor()方法，该方法用于创建自定义小部件。在createEditor()方法中，我们可以使用任何Qt支持的小部件，例如QLineEdit、QComboBox等。

下面是一个简单的示例，展示了如何在QTableView的单元格中插入一个QPushButton小部件：

import sys
from PyQt5.QtWidgets import QApplication, QMainWindow, QTableView, QPushButton, QStyledItemDelegate

class ButtonDelegate(QStyledItemDelegate):
    def __init__(self, parent=None):
        super().__init__(parent)

    def createEditor(self, parent, option, index):
        button = QPushButton(parent)
        button.setText("Click Me")
        return button

app = QApplication(sys.argv)
window = QMainWindow()
table_view = QTableView(window)

# 创建一个自定义委托对象
delegate = ButtonDelegate()
table_view.setItemDelegate(delegate)

# 设置表格大小和数据
table_view.setRowCount(4)
table_view.setColumnCount(4)

for row in range(4):
    for col in range(4):
        table_view.setIndexWidget(table_view.model().index(row, col), QPushButton("Button"))

window.setCentralWidget(table_view)
window.show()
sys.exit(app.exec_())

在上面的示例中，我们创建了一个ButtonDelegate类，并在其中重写了createEditor()方法，以创建一个带有文本为”Click Me”的QPushButton。然后，我们将该委托对象设置为QTableView的委托，从而实现在每个单元格中插入按钮。

请注意，我们还使用了QStyledItemDelegate，该委托类提供了更好的绘制效果和事件处理能力。

运行上述代码，我们将看到一个带有按钮的表格，每个单元格中都有一个按钮。当我们点击按钮时，按钮会发出信号并触发相应的事件。

使用自定义模型进行小部件插入

除了使用委托外，我们还可以通过使用自定义模型（Model）来在QTableView中插入小部件。自定义模型允许我们完全控制表格的显示和行为。

首先，我们需要定义一个继承自QAbstractTableModel的自定义模型类，并实现几个必要的方法（如rowCount()、columnCount()、data()等）。在data()方法中，我们可以根据需要返回不同的数据类型，包括小部件。

下面是一个示例，展示了如何使用自定义模型插入一个QLabel小部件，以替代特定单元格中的文本内容：

import sys
from PyQt5.QtWidgets import QApplication, QMainWindow, QTableView, QLabel, QAbstractTableModel

class CustomTableModel(QAbstractTableModel):
    def __init__(self, data, headers):
        super().__init__()
        self._data = data
        self._headers = headers

    def rowCount(self, parent):
        return len(self._data)

    def columnCount(self, parent):
        return len(self._data[0])

    def data(self, index, role):
        if role == QtCore.Qt.DisplayRole:
            row = index.row()
            col = index.column()
            value = self._data[row][col]
            return str(value)

        if role == QtCore.Qt.EditRole:
            row = index.row()
            col = index.column()
            value = self._data[row][col]
            return value

        if role == QtCore.Qt.DecorationRole:
            row = index.row()
            col = index.column()
            if row == 2 and col == 2:
                label = QLabel()
                label.setText("Custom Widget")
                return label

app = QApplication(sys.argv)
window = QMainWindow()
table_view = QTableView(window)

data = [
    ["A", "B", "C"],
    ["D", "E", "F"],
    ["G", "H", "I"]
]

headers = ["Column 1", "Column 2", "Column 3"]

model = CustomTableModel(data, headers)
table_view.setModel(model)

window.setCentralWidget(table_view)
window.show()
sys.exit(app.exec_())

在上面的示例中，我们创建了一个CustomTableModel类，并实现了rowCount()、columnCount()和data()方法。在data()方法中，我们首先返回文本数据作为DisplayRole和EditRole，然后在(2, 2)位置返回一个包含文本”Custom Widget”的QLabel小部件作为DecorationRole。

运行上述代码，我们将看到一个带有自定义小部件的表格。注意，我们可以使用data()方法的role参数来返回不同类型的数据，从而在QTableView的单元格中插入小部件。