【pyqt6】qtreeview使用自定义model

在PyQt6中，可以使用自定义的模型来在QTreeView中显示数据。下面是一个使用自定义模型的示例代码。

from PyQt6.QtCore import QAbstractItemModel, QModelIndex, Qt
from PyQt6.QtWidgets import QApplication, QMainWindow, QTreeView

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

    def setupModelData(self, data, parent):
        for item in data:
            parent.appendChild(item)

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

        if not parent.isValid():
            parentItem = self.rootItem
        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.rootItem:
            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.rootItem
        else:
            parentItem = parent.internalPointer()

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

    def data(self, index, role=Qt.ItemDataRole.DisplayRole):
        if not index.isValid():
            return None

        item = index.internalPointer()

        if role == Qt.ItemDataRole.DisplayRole:
            return item.data(index.column())
        elif role == Qt.ItemDataRole.TextAlignmentRole:
            return Qt.AlignmentFlag.AlignVCenter + Qt.AlignmentFlag.AlignLeft

    def headerData(self, section, orientation, role=Qt.ItemDataRole.DisplayRole):
        if orientation == Qt.Orientation.Horizontal and role == Qt.ItemDataRole.DisplayRole:
            return self.rootItem.data(section)

    def flags(self, index):
        if not index.isValid():
            return Qt.ItemFlag.NoItemFlags

        return Qt.ItemFlag.ItemIsEnabled | Qt.ItemFlag.ItemIsSelectable

class CustomItem:
    def __init__(self, data, parent):
        self.parentItem = parent
        self.itemData = data
        self.childItems = []

    def appendChild(self, item):
        self.childItems.append(item)

    def child(self, row):
        return self.childItems[row]

    def childCount(self):
        return len(self.childItems)

    def columnCount(self):
        return len(self.itemData)

    def data(self, column):
        return self.itemData[column]

    def parent(self):
        return self.parentItem

if __name__ == "__main__":
    app = QApplication([])
    window = QMainWindow()
    treeView = QTreeView(window)

    data = [
        CustomItem(["Item 1"], None),
        CustomItem(["Item 2"], None)
    ]
    model = CustomModel(data)
    treeView.setModel(model)

    window.setCentralWidget(treeView)
    window.resize(500, 500)
    window.show()

    app.exec()

在这个示例中，定义了一个`CustomModel`类继承自`QAbstractItemModel`类，并重写了一些必要的方法。同时定义了一个`CustomItem`类作为模型的节点数据。

在`CustomModel`中，`index`方法用于获取给定行列的索引，`parent`方法用于获取给定索引的父索引，`rowCount`和`columnCount`方法用于获取给定索引的行数和列数，`data`方法返回指定索引的数据，`headerData`方法返回给定节的标题数据，`flags`方法定义每个索引的标志。

在主函数中，创建一个自定义模型`CustomModel`的实例，并将数据传递给模型。然后将该模型设置为`QTreeView`的模型，最后显示窗口并进入应用程序的事件循环。

这样，使用自定义模型的QTreeView就可以显示自定义数据了。您可以根据自己的需求扩展并修改此示例代码。