Pyqt -在添加到"IconMode“中的QTreeViewItem时，在自定义小部件上丢失选择高亮显示

在PyQt中，当你在IconMode下的QTreeView中添加自定义小部件时，可能会遇到选择高亮显示丢失的问题。这是因为QTreeView默认情况下不知道如何处理自定义小部件的选择状态。以下是一些基础概念和相关解决方案。

基础概念

QTreeView: PyQt中的一个视图类，用于显示层次数据模型。
IconMode: QTreeView的一种显示模式，通常用于显示图标和文本。
自定义小部件: 用户定义的小部件，可能需要特殊处理以与QTreeView集成。

问题原因

当你在QTreeView中使用自定义小部件时，QTreeView默认的选择机制可能无法正确地应用到这些小部件上，导致选择高亮显示丢失。

解决方案

为了确保自定义小部件在选择时能够正确高亮显示，你需要手动处理选择状态。以下是一个示例代码，展示了如何实现这一点：

import sys
from PyQt5.QtWidgets import QApplication, QTreeView, QWidget, QVBoxLayout, QLabel, QStyledItemDelegate, QStyleOptionViewItem
from PyQt5.QtCore import Qt, QModelIndex
from PyQt5.QtGui import QStandardItemModel, QStandardItem

class CustomWidget(QWidget):
    def __init__(self, text):
        super().__init__()
        layout = QVBoxLayout(self)
        label = QLabel(text)
        layout.addWidget(label)

class CustomDelegate(QStyledItemDelegate):
    def createEditor(self, parent, option, index):
        widget = CustomWidget("Custom Item")
        return widget

    def paint(self, painter, option, index):
        if option.state & QStyle.State_Selected:
            painter.fillRect(option.rect, option.palette.highlight())
        super().paint(painter, option, index)

class TreeViewExample(QWidget):
    def __init__(self):
        super().__init__()
        self.initUI()

    def initUI(self):
        self.treeView = QTreeView()
        model = QStandardItemModel()
        rootItem = model.invisibleRootItem()

        item1 = QStandardItem("Item 1")
        item2 = QStandardItem("Item 2")
        rootItem.appendRow(item1)
        rootItem.appendRow(item2)

        self.treeView.setModel(model)
        self.treeView.setItemDelegate(CustomDelegate())

        layout = QVBoxLayout()
        layout.addWidget(self.treeView)
        self.setLayout(layout)

if __name__ == '__main__':
    app = QApplication(sys.argv)
    ex = TreeViewExample()
    ex.show()
    sys.exit(app.exec_())

解释

CustomWidget: 这是一个简单的自定义小部件，包含一个标签。
CustomDelegate: 这是一个自定义委托，用于处理自定义小部件的绘制和编辑。
- createEditor: 创建并返回自定义小部件。
- paint: 重写绘制方法，确保在选择状态下正确应用高亮显示。