Qt第六章：渲染树

叁歲伎倆 2024-04-07 15:03 101阅读 0赞

渲染树组件：

import sys
    from PySide6.QtWidgets import QApplication, QTreeWidget, QTreeWidgetItem
    
    # 数据
    data = {"Project A": ["file_a.py", "file_a.txt", "something.xls"],
            "Project B": ["file_b.csv", "photo.jpg"],
            "Project C": []}
    
    if __name__ == '__main__':
        app = QApplication()
    
        # 创建树组件
        tree = QTreeWidget()
        # 设置列数
        tree.setColumnCount(2)
        # 设置列头
        tree.setHeaderLabels(["Name", "Type"])
    
        items = []
        for key, values in data.items():
            item = QTreeWidgetItem([key])
            for value in values:
                ext = value.split(".")[-1].upper()
                child = QTreeWidgetItem([value, ext])
                item.addChild(child)
            items.append(item)
    
        # 挂载到树上
        tree.insertTopLevelItems(0, items)
        tree.show()
        
        sys.exit(app.exec())

封装函数：

from PySide6.QtWidgets import QApplication
    from PySide6.QtWidgets import QWidget, QTreeWidget, QTreeWidgetItem
    
    """
    data: 数据列表
    tree: 树组件句柄
    column: [(字段,显示名称),...]
    id_point: 数据主键（默认id）
    parent_pointer: 标注父子关系的字段（默认parent_id）
    """
    
    
    def list2QTree(data, tree: QTreeWidget, column: list, id_point='id', parent_pointer='parent_id') -> QTreeWidget:
        # 设置列数
        tree.setColumnCount(len(column))
    
        # 标题
        tree.setHeaderLabels([i[1] for i in column])
    
        # 顶层节点
        items = []
    
        # 转成QTreeWidgetItem列表
        data_item = [{id_point: i[id_point], parent_pointer: i[parent_pointer],
                      'item': QTreeWidgetItem([str(i[j[0]]) for j in column])} for i in data]
    
        # 转成ID为Key的字典
        mapping: dict = dict(zip([i[id_point] for i in data_item], data_item))
    
        # 组树
        for d in data_item:
            parent: dict = mapping.get(d[parent_pointer])
            if parent is None:
                items.append(d['item'])
            else:
                item: QTreeWidgetItem = parent.get('item')
                item.addChild(d.get('item'))
    
        # 挂载到树上
        tree.insertTopLevelItems(0, items)
        return tree

使用示例：

class Form(QWidget):
        def __init__(self, parent=None):
            data: list = [
                {'id': 1, 'parent_id': 0, 'name': '用户管理', 'url': ''},
                {'id': 2, 'parent_id': 0, 'name': '菜单管理', 'url': ''},
                {'id': 3, 'parent_id': 1, 'name': '新增用户', 'url': 'https://www.baidu.com'},
                {'id': 4, 'parent_id': 1, 'name': '删除用户', 'url': 'https://www.baidu.com'},
                {'id': 5, 'parent_id': 2, 'name': '新增菜单', 'url': 'https://www.baidu.com'},
                {'id': 6, 'parent_id': 2, 'name': '删除菜单', 'url': ''},
                {'id': 7, 'parent_id': 6, 'name': '单个删除', 'url': 'https://www.baidu.com'},
                {'id': 8, 'parent_id': 6, 'name': '批量删除', 'url': 'https://www.baidu.com'},
            ]
    
            super(Form, self).__init__(parent=parent)
            self.setWindowTitle("首次")
            self.resize(320, 180)
            tree = QTreeWidget(self)
            list2QTree(data=data, tree=tree, column=[('name', '姓名'), ('url', '地址')])
    
    
    if __name__ == '__main__':
        app = QApplication([])
        form = Form()
        form.show()
        app.exec()