如何从 qtreeview 中的父行中获取特定列

Posted 2023-02-16

【中文标题】如何从 qtreeview 中的父行中获取特定列

【英文标题】：How to get a specific column out of parent row in a qtreeview

【发布时间】：2019-07-16 14:15:51

【问题描述】：

正如标题所述，我正在尝试从 qtreeview 中当前选定行的父行中的特定列中提取一个值。

我试过这样做：

index = self.view.currentIndex()
parent_index = index.parent()
parent_col1_index = self.model.index(parent_index.row(), 1)
text = self.model.data(parent_col1_index)

在这样的数据结构上，选择了 id3：

- id1                    id1
    - id1.id2            id2
        - id1.id2.id3    id3

当我使用 parent_index 变量从模型中提取数据时，我会从父行 (id2) 中获取我单击的任何列。但是，当我尝试使用 parent_col1_index 变量从该父级获取特定列时，我得到了正确的列，但来自顶行（id1）。即使对于更大的数据集，这也是一致的，如果我尝试从任何深度为 2 的行的父行中获取一列，我会返回树视图的第一行。

我认为这与我如何使用 model.index 创建列特定索引有关。但我不确定这还能怎么做。

有什么想法吗？

如果它有帮助，这里是一个工作示例。当您单击按钮时，如果您选择了 id3 行，它应该输出 id2

from PyQt5 import QtCore, QtGui, QtWidgets
from collections import deque

test_data = [
    "column2": "id1", "column1": "id1", "column3": "id1-value", "lvl": 0,
    "column2": "id2", "column1": "id1.id2", "column3": "id2-value", "lvl": 1,
    "column2": "id3", "column1": "id1.id2.id3", "column3": "id3-value", "lvl": 2,
]

class Ui_MainWindow(object):
    def setupUi(self, MainWindow):
        MainWindow.setObjectName("MainWindow")
        MainWindow.resize(640, 480)
        self.centralwidget = QtWidgets.QWidget(MainWindow)
        self.centralwidget.setObjectName("centralwidget")
        self.eventTree = QtWidgets.QTreeView(self.centralwidget)
        self.eventTree.setGeometry(QtCore.QRect(10, 40, 621, 391))
        self.eventTree.setObjectName("eventTree")
        self.pushButton = QtWidgets.QPushButton(self.centralwidget)
        self.pushButton.setGeometry(QtCore.QRect(20, 10, 89, 25))
        self.pushButton.setObjectName("pushButton")
        self.label = QtWidgets.QLabel(self.centralwidget)
        self.label.setGeometry(QtCore.QRect(170, 10, 391, 17))
        self.label.setObjectName("label")
        MainWindow.setCentralWidget(self.centralwidget)
        self.menubar = QtWidgets.QMenuBar(MainWindow)
        self.menubar.setGeometry(QtCore.QRect(0, 0, 640, 22))
        self.menubar.setObjectName("menubar")
        MainWindow.setMenuBar(self.menubar)
        self.statusbar = QtWidgets.QStatusBar(MainWindow)
        self.statusbar.setObjectName("statusbar")
        MainWindow.setStatusBar(self.statusbar)

        self.model = QtGui.QStandardItemModel()
        self.model.setHorizontalHeaderLabels(["Column1", "Column2", "Column3"])
        self.eventTree.header().setDefaultSectionSize(180)
        self.eventTree.setModel(self.model)
        self.pushButton.clicked.connect(self.clickHandle)

        self.retranslateUi(MainWindow)
        QtCore.QMetaObject.connectSlotsByName(MainWindow)

        self.importData(test_data)

    def retranslateUi(self, MainWindow):
        _translate = QtCore.QCoreApplication.translate
        MainWindow.setWindowTitle(_translate("MainWindow", "MainWindow"))
        self.pushButton.setText(_translate("MainWindow", "PushButton"))
        self.label.setText(_translate("MainWindow", "TextLabel"))

    def clickHandle(self):
        index = self.eventTree.currentIndex()
        parent_index = index.parent()
        parent_index = self.model.index(parent_index.row(), 1)
        text = parent_index.data()
        self.label.setText(text)

    def importData(self, data, root=None, extend=False):
        self.model.setRowCount(0)
        if root is None:
            root = self.model.invisibleRootItem()

        seen = 
        values = deque(data)
        while values:
            value = values.popleft()
            if value['lvl'] == 0:
                parent = root
            else:
                pid = ".".join(value['column1'].split(".")[:-1])
                if pid not in seen:
                    values.append(value)
                    continue
                parent = seen[pid]
            dbid = value['column1']
            parent.appendRow([
                QtGui.QStandardItem(value['column1']),
                QtGui.QStandardItem(value['column2']),
                QtGui.QStandardItem(value['column3']),
            ])
            seen[dbid] = parent.child(parent.rowCount() - 1)


if __name__ == "__main__":
    import sys
    app = QtWidgets.QApplication(sys.argv)
    MainWindow = QtWidgets.QMainWindow()
    ui = Ui_MainWindow()
    ui.setupUi(MainWindow)
    MainWindow.show()
    sys.exit(app.exec_())

【参考方案1】：

哎呀，想通了。只是我不应该调用 model.index 方法来为父级中的特定列创建索引。我需要的是调用 index.sibling 方法。像这样：

parent = index.parent()
parent_col1_index = parent.sibling(parent.row(), 1)