PyQt 自动调整QTreeView大小

在本文中，我们将介绍如何使用PyQt来使QTreeView自动调整大小。QTreeView是一个常用的Qt小部件，用于展示层级数据。默认情况下，QTreeView的大小是固定的，但在某些情况下，我们希望它能够根据内容自动调整大小。

阅读更多：PyQt 教程

使用QTreeView自动调整大小

要使QTreeView自动调整大小，我们可以使用QHeaderView和QTreeModel。QHeaderView是QTreeView的标题栏，我们可以通过设置它的resizeContentsToContents属性来自动调整列的大小。QTreeModel是QTreeView的数据模型，我们可以通过设置它的layoutChanged信号来实时更新QTreeView的大小。

下面是一个简单的示例，展示了如何使用PyQt来自动调整QTreeView的大小：

from PyQt5.QtWidgets import QApplication, QMainWindow, QTreeView, QVBoxLayout, QWidget
from PyQt5.QtCore import QAbstractItemModel, QModelIndex


class TreeModel(QAbstractItemModel):
    def __init__(self, data, parent=None):
        super(TreeModel, self).__init__(parent)
        self.rootItem = TreeItem(("Title", "Description"))

        for itemData in data:
            self.rootItem.appendChild(TreeItem(itemData))

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

    def columnCount(self, parent=QModelIndex()):
        return self.rootItem.columnCount()

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

        if role != Qt.DisplayRole:
            return None

        item = index.internalPointer()

        return item.data(index.column())

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

        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)


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

        if parent:
            parent.appendChild(self)

    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

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

        return 0


if __name__ == "__main__":
    import sys
    from PyQt5.QtCore import Qt

    app = QApplication(sys.argv)

    data = [
        ("Item 1", "Description 1"),
        ("Item 2", "Description 2"),
        ("Item 3", "Description 3")
    ]

    model = TreeModel(data)
    treeView = QTreeView()
    treeView.setModel(model)
    treeView.header().setResizeContentsToContents(True)  # 设置自动调整列宽

    layout = QVBoxLayout()
    layout.addWidget(treeView)

    widget = QWidget()
    widget.setLayout(layout)

    mainWindow = QMainWindow()
    mainWindow.setCentralWidget(widget)
    mainWindow.show()

    sys.exit(app.exec_())

在上面的示例中，我们首先定义了一个自定义的数据模型TreeModel，它继承自QAbstractItemModel。在TreeModel中，我们通过创建TreeItem来组织数据，TreeItem表示模型中的一个项。然后，我们将数据模型设置为QTreeView的模型，并将QHeaderView的resizeContentsToContents属性设置为True，以实现自动调整列宽。