如何在QTreeView小部件中更改图标

在QTreeView小部件中更改图标，您可以按照以下步骤操作：

创建一个自定义的QAbstractItemModel子类。该模型将负责提供数据给QTreeView小部件，并处理图标的更改。您可以根据需要实现data()和setData()方法来返回和设置图标数据。
在QTreeView中设置模型。通过调用QTreeView的setModel()方法，并将自定义的模型对象作为参数传递给该方法，将模型与QTreeView关联起来。
在自定义模型中，可以通过重写data()方法来为需要更改图标的项提供自定义的图标。在该方法中，根据需要判断当前项的索引和角色，并根据不同的情况返回相应的图标数据。
当需要更改图标时，可以通过调用setData()方法来更新项的图标。在该方法中，可以根据需要将新的图标数据设置给相应的项。

下面是一个示例代码，演示如何在QTreeView小部件中更改图标：

from PyQt5.QtCore import Qt, QModelIndex, QAbstractItemModel, QVariant
from PyQt5.QtGui import QIcon

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

    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.parentItem()

        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.DisplayRole):
        if not index.isValid():
            return QVariant()

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

        if role == Qt.DecorationRole:
            if index.column() == 0:
                item = index.internalPointer()
                icon = item.icon()
                if icon:
                    return QIcon(icon)
        
        return QVariant()

    def setData(self, index, value, role=Qt.EditRole):
        if role == Qt.EditRole:
            item = index.internalPointer()
            if index.column() == 0:
                item.setIcon(value)
            
            self.dataChanged.emit(index, index)
            return True
        
        return False

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

    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):
        if column < len(self.itemData):
            return self.itemData[column]

        return None

    def parentItem(self):
        return self.parentItem

    def row(self):
        if self.parentItem:
            return self.parentItem.childItems.index(self)

        return 0

    def icon(self):
        return self.itemIcon

    def setIcon(self, icon):
        self.itemIcon = icon

if __name__ == "__main__":
    import sys
    from PyQt5.QtWidgets import QApplication, QTreeView, QVBoxLayout, QWidget

    app = QApplication(sys.argv)
    window = QWidget()
    layout = QVBoxLayout(window)

    # 创建自定义模型
    rootItem = TreeItem(["Name", "Size"])
    childItem1 = TreeItem(["File1.txt", "10 KB"], rootItem)
    childItem2 = TreeItem(["File2.txt", "20 KB"], rootItem)
    rootItem.appendChild(childItem1)
    rootItem.appendChild(childItem2)
    model = CustomModel(rootItem)

    # 创建QTreeView小部件
    treeView = QTreeView()
    treeView.setModel(model)
    
    # 更改图标
    model.setData(model.index(0, 0), QIcon("folder.png"), Qt.EditRole)

    layout.addWidget(treeView)
    window.show()

    sys.exit(app.exec_())

在上面的示例中，我们创建了一个自定义的模型（CustomModel），并在其中存储了文件的名称和大小数据。我们通过重写data()和setData()方法来返回和设置图标数据。在主程序中，我们创建了一个QTreeView小部件，并将自定义模型设置给它。我们通过调用setData()方法来更改第一个项的图标。

请注意，此示例仅演示了如何更改一个项的图标。根据您的实际需求，您可以根据需要修改模型和视图，并为不同的项提供不同的图标。