python - PyQt5:如何从字典项列表生成 QTreeView？

Question

我有这样一个数据集:

[{'level': 0, 'dbID': 77, 'parent_ID': 6, 'short_name': '0:0:0:<new> to 6', 'long_name': '', 'order': 1, 'pos': 0} ,
{'level': 1, 'dbID': 88, 'parent_ID': 77, 'short_name': '1:1:1:Store13', 'long_name': '', 'order': 2, 'pos': 1} ,
{'level': 0, 'dbID': 442, 'parent_ID': 6, 'short_name': '2:<new>', 'long_name': '', 'order': 1, 'pos': 2} ,
{'level': 1, 'dbID': 522, 'parent_ID': 442, 'short_name': '3:<new>', 'long_name': '', 'order': 2, 'pos': 3} ,
{'level': 2, 'dbID': 171, 'parent_ID': 522, 'short_name': '3:<new>', 'long_name': '', 'order': 1, 'pos': 3} ,
{'level': 0, 'dbID': 456, 'parent_ID': 6, 'short_name': '4:<new>', 'long_name': '', 'order': 1, 'pos': 4} ,
{'level': 1, 'dbID': 523, 'parent_ID': 456, 'short_name': '5:<new>', 'long_name': '', 'order': 3, 'pos': 5}]

这里的“级别”表示数据将位于树的哪个级别。级别 0 表示父节点，1 表示 0 的子节点，2 表示 1 的子节点，依此类推。我必须从中生成一个 QTreeView。我一直在尝试许多不同的方法来做到这一点，但我失败了。这是我到目前为止编写的代码:

函数:

def populateTree(self):
    tree=self.structureData
    root_model=MoDaTreeModel()
    self.treeViewStructure.setModel(root_model)
    self.recursiveAdd(tree,0,root_model)

def recursiveAdd(self,tree,n,parent):
    for i in range(n,len(tree)):            
        if(tree[i]['level']<=tree[i+1]['level']):
            child_item=QStandardItem()
            child_item.setText(str(tree[i]['short_name']))
            parent.appendRow(child_item)
        else:
            self.recursiveAdd(tree,i+1) #I want to write the child node, but don't know how

有人可以帮助我如何从这里开始吗？

Answer 1

由于您的数据结构不是分层的，因此迭代解决方案更为合适。使用平面数据结构，每个项目的父级并不总是提前可用，因此必须将某些项目推回到堆栈中，直到它们的父级被处理。

这是一个基于您问题中的数据的简单演示:

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

class Window(QtWidgets.QWidget):
    def __init__(self, data):
        super(Window, self).__init__()
        self.tree = QtWidgets.QTreeView(self)
        layout = QtWidgets.QVBoxLayout(self)
        layout.addWidget(self.tree)
        self.model = QtGui.QStandardItemModel()
        self.model.setHorizontalHeaderLabels(['Name', 'dbID'])
        self.tree.header().setDefaultSectionSize(180)
        self.tree.setModel(self.model)
        self.importData(data)
        self.tree.expandAll()

    def importData(self, data, root=None):
        self.model.setRowCount(0)
        if root is None:
            root = self.model.invisibleRootItem()
        seen = {}
        values = deque(data)
        while values:
            value = values.popleft()
            if value['level'] == 0:
                parent = root
            else:
                pid = value['parent_ID']
                if pid not in seen:
                    values.append(value)
                    continue
                parent = seen[pid]
            dbid = value['dbID']
            parent.appendRow([
                QtGui.QStandardItem(value['short_name']),
                QtGui.QStandardItem(str(dbid)),
                ])
            seen[dbid] = parent.child(parent.rowCount() - 1)

if __name__ == '__main__':

    data = [
        {'level': 0, 'dbID': 77, 'parent_ID': 6, 'short_name': '0:0:0:<new> to 6', 'long_name': '', 'order': 1, 'pos': 0} ,
        {'level': 1, 'dbID': 88, 'parent_ID': 77, 'short_name': '1:1:1:Store13', 'long_name': '', 'order': 2, 'pos': 1} ,
        {'level': 0, 'dbID': 442, 'parent_ID': 6, 'short_name': '2:<new>', 'long_name': '', 'order': 1, 'pos': 2} ,
        {'level': 1, 'dbID': 522, 'parent_ID': 442, 'short_name': '3:<new>', 'long_name': '', 'order': 2, 'pos': 3} ,
        {'level': 2, 'dbID': 171, 'parent_ID': 522, 'short_name': '3:<new>', 'long_name': '', 'order': 1, 'pos': 3} ,
        {'level': 0, 'dbID': 456, 'parent_ID': 6, 'short_name': '4:<new>', 'long_name': '', 'order': 1, 'pos': 4} ,
        {'level': 1, 'dbID': 523, 'parent_ID': 456, 'short_name': '5:<new>', 'long_name': '', 'order': 3, 'pos': 5}
        ]

    app = QtWidgets.QApplication(sys.argv)
    window = Window(data)
    window.setGeometry(600, 50, 400, 250)
    window.show()
    sys.exit(app.exec_())