QTableView 切换折叠展开图标

QTableView 是 Qt 框架中的一个类，主要用于显示表格数据。如果想要在 QTableView 中实现折叠展开图标的切换，可以通过自定义 QStyledItemDelegate 类来实现。

具体实现步骤如下：

1. 创建自定义的 QStyledItemDelegate 类，重写 paint() 函数，在该函数中根据数据模型中的数据来绘制折叠展开图标。

2. 在 QTableView 中设置该自定义的 QStyledItemDelegate 类作为 itemDelegate。

3. 在数据模型中添加用于存储折叠展开状态的数据，通过该数据来控制折叠展开图标的显示状态。

4. 在 QTableView 中设置相应的信号与槽函数，当用户点击折叠展开图标时，修改数据模型中的折叠展开状态数据，并更新 QTableView 的显示。

下面是示例代码，仅供参考：

class MyDelegate(QStyledItemDelegate):
    def __init__(self, parent=None):
        super(MyDelegate, self).__init__(parent)

    def paint(self, painter, option, index):
        if index.data(Qt.UserRole) == "collapsed":
            icon = QIcon("collapsed.png")
        else:
            icon = QIcon("expanded.png")
        icon.paint(painter, option.rect, Qt.AlignLeft)

    def sizeHint(self, option, index):
        return QSize(20, 20)


class MyTableModel(QAbstractTableModel):
    def __init__(self, data, headers):
        super(MyTableModel, self).__init__()
        self._data = data
        self._headers = headers
        self._collapsed = [False] * len(data)

    def data(self, index, role):
        if role == Qt.DisplayRole:
            return self._data[index.row()][index.column()]
        elif role == Qt.UserRole:
            if self._collapsed[index.row()]:
                return "collapsed"
            else:
                return "expanded"
        return QVariant()

    def setData(self, index, value, role):
        if role == Qt.UserRole:
            if value == "collapsed":
                self._collapsed[index.row()] = True
            else:
                self._collapsed[index.row()] = False
            self.dataChanged.emit(index, index)
            return True
        return False

    def flags(self, index):
        return Qt.ItemIsEditable | Qt.ItemIsEnabled

    def headerData(self, section, orientation, role):
        if orientation == Qt.Horizontal and role == Qt.DisplayRole:
            return self._headers[section]
        return QVariant()

    def rowCount(self, index=QModelIndex()):
        if index.isValid():
            return 0
        return len(self._data)

    def columnCount(self, index=QModelIndex()):
        if index.isValid():
            return 0
        return len(self._data[0])


class MyTableView(QTableView):
    def __init__(self, parent=None):
        super(MyTableView, self).__init__(parent)
        self.horizontalHeader().setStretchLastSection(True)
        self.verticalHeader().setVisible(False)

        delegate = MyDelegate(self)
        self.setItemDelegate(delegate)

        self.clicked.connect(self.handleClicked)

    def handleClicked(self, index):
        if index.column() == 0:
            value = index.data(Qt.UserRole)
            if value == "collapsed":
                self.model().setData(index, "expanded", Qt.UserRole)
            else:
                self.model().setData(index, "collapsed", Qt.UserRole)

在上述代码中，MyDelegate 类用于绘制折叠展开图标，MyTableModel 类用于存储数据，并且添加用于存储折叠展开状态的数据，MyTableView 类用于设置 itemDelegate，以及处理用户点击操作。