在QTreeView中通过QAbstractItemModel实现用户可编辑复选框的自定义

from PyQt5.QtCore import Qt, QAbstractItemModel, QModelIndex
from PyQt5.QtGui import QStandardItem, QStandardItemModel
from PyQt5.QtWidgets import QApplication, QTreeView

class CustomItemModel(QAbstractItemModel):
    def __init__(self, parent=None):
        super().__init__(parent)
        self.rootItem = QStandardItem("Root")
        self.setupModelData()

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

    def columnCount(self, parent=QModelIndex()):
        return 1

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

        if role == Qt.DisplayRole or role == Qt.EditRole:
            item = index.internalPointer()
            return item.text()

        if role == Qt.CheckStateRole:
            item = index.internalPointer()
            if item.isCheckable():
                return item.checkState()

        return None

    def setData(self, index, value, role=Qt.EditRole):
        if index.isValid() and role == Qt.CheckStateRole:
            item = index.internalPointer()
            if item.isCheckable():
                item.setCheckState(value)
                return True
        return False

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

        item = index.internalPointer()
        flags = Qt.ItemIsEnabled | Qt.ItemIsSelectable

        if item.isCheckable():
            flags |= Qt.ItemIsUserCheckable

        return flags

    def headerData(self, section, orientation, role=Qt.DisplayRole):
        if orientation == Qt.Horizontal and role == Qt.DisplayRole:
            return "Header"

        return None

    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 setupModelData(self):
        parentItem = self.rootItem

        for i in range(3):
            item = QStandardItem(f"Item {i}")
            item.setCheckable(True)
            parentItem.appendRow(item)

            for j in range(2):
                childItem = QStandardItem(f"Child {i}-{j}")
                childItem.setCheckable(True)
                item.appendRow(childItem)

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

    model = CustomItemModel()
    treeView.setModel(model)

    treeView.show()
    app.exec_()