如何在QTreeView中显示自定义小部件？

在Qt框架中，QTreeView 是一个用于显示层次数据的视图组件。如果你想在 QTreeView 中显示自定义的小部件，你需要通过自定义模型和委托来实现。以下是实现这一功能的基础概念和相关步骤：

基础概念

模型（Model）：在Qt中，模型负责管理数据。你需要创建一个自定义模型，继承自 QAbstractItemModel 或者其子类，如 QStandardItemModel。
视图（View）：QTreeView 是一个视图，它显示模型的数据。
委托（Delegate）：委托负责绘制和编辑视图中的项。你可以通过自定义委托来绘制自定义小部件。

实现步骤

自定义模型

首先，你需要创建一个自定义模型来管理你的数据。

from PyQt5.QtCore import QAbstractItemModel, QModelIndex

class CustomModel(QAbstractItemModel):
    def __init__(self, data, parent=None):
        super().__init__(parent)
        self._data = data

    def index(self, row, column, parent=QModelIndex()):
        if not self.hasIndex(row, column, parent):
            return QModelIndex()

        if not parent.isValid():
            parentItem = self._data
        else:
            parentItem = parent.internalPointer()

        childItem = parentItem.child(row)
        if childItem:
            return self.createIndex(row, column, childItem)
        else:
            return QModelIndex()

    def parent(self, index):
        if not index.isValid():
            return QModelIndex()

        childItem = index.internalPointer()
        parentItem = childItem.parent()

        if parentItem == self._data:
            return QModelIndex()

        return self.createIndex(parentItem.row(), 0, parentItem)

    def rowCount(self, parent=QModelIndex()):
        if parent.column() > 0:
            return 0

        if not parent.isValid():
            parentItem = self._data
        else:
            parentItem = parent.internalPointer()

        return parentItem.childCount()

    def columnCount(self, parent=QModelIndex()):
        if parent.isValid():
            return parent.internalPointer().columnCount()
        else:
            return self._data.columnCount()

    def data(self, index, role):
        if not index.isValid():
            return None

        item = index.internalPointer()

        if role == Qt.DisplayRole:
            return item.data(index.column())

        return None

自定义委托

接下来，你需要创建一个自定义委托来绘制你的自定义小部件。

from PyQt5.QtWidgets import QStyledItemDelegate, QWidget, QVBoxLayout, QLabel
from PyQt5.QtCore import Qt

class CustomDelegate(QStyledItemDelegate):
    def createEditor(self, parent, option, index):
        editor = QWidget(parent)
        layout = QVBoxLayout(editor)
        label = QLabel(editor)
        layout.addWidget(label)
        return editor

    def setEditorData(self, editor, index):
        value = index.model().data(index, Qt.DisplayRole)
        label = editor.layout().itemAt(0).widget()
        label.setText(value)

    def setModelData(self, editor, model, index):
        label = editor.layout().itemAt(0).widget()
        model.setData(index, label.text(), Qt.DisplayRole)

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

使用自定义模型和委托

最后，你需要在你的 QTreeView 中使用这些自定义组件。

from PyQt5.QtWidgets import QApplication, QTreeView

app = QApplication([])

tree_view = QTreeView()
model = CustomModel(data)  # 假设你已经有了一个合适的数据结构
delegate = CustomDelegate()

tree_view.setModel(model)
tree_view.setItemDelegate(delegate)
tree_view.show()

app.exec_()