PyQt5 QTreeWidget 树形结构递归遍历当前所有节点的实现

QTreeWidget类中的常用方法

QTreeWidgetItem类中的常用方法

准备一个小demo

常见小demo
简介：商品种类和商品的树形嵌套结构demo
功能：点击按键获取当前所有被选中的商品(夹带了一点私货〃‘▽'〃)
注意：篇幅有限，没有写子节点和父节点联动选中，所以在选择子节点时麻烦自行将父节点选上，不然会跳过。

代码块儿：

import sys
from PyQt5.QtWidgets import QTreeWidgetItem, QTreeWidget, QWidget, QVBoxLayout, QPushButton, QApplication
from PyQt5.QtCore import Qt


class Demo(QWidget):
    def __init__(self):
        super().__init__()
        # 实例化一个树形结构，隐藏了header
        self.tree = QTreeWidget()
        self.tree.setHeaderHidden(True)
        # 顶级分支
        self.tree_main = QTreeWidgetItem(self.tree)
        self.tree_main.setText(0, '商品种类')
        # 设置一些二级分支
        tree_second = ['电子产品', '水果', '日用品', '喜欢的人']
        self.gen_branch(self.tree_main, tree_second)
        # 设置一些三级分支
        tree_fruit = ['苹果', '香蕉', '梨']
        tree_daily_use = ['纸巾', '毛巾']
        tree_lovers = ['迪迪1号', '迪迪2号']
        # child(1) 意思是分支的第1个节点, 序号从0算起
        self.gen_branch(self.tree_main.child(1), tree_fruit)
        self.gen_branch(self.tree_main.child(2), tree_daily_use)
        self.gen_branch(self.tree_main.child(3), tree_lovers)
        # 一个按钮
        self.pushButton = QPushButton('选好了')
        # 显示出来
        self.qvl = QVBoxLayout()
        self.qvl.addWidget(self.tree)
        self.qvl.addWidget(self.pushButton)
        self.setLayout(self.qvl)

        # 绑定一下槽函数，传入主要的分支节点
        self.pushButton.clicked.connect(lambda: self.get_checked(self.tree_main))

    @staticmethod
    def gen_branch(node: QTreeWidgetItem, texts: list):
        """ 给定某个节点和列表 在该节点生成列表内分支"""
        for text in texts:
            item = QTreeWidgetItem()
            item.setText(0, text)
            item.setCheckState(0, Qt.Unchecked)
            node.addChild(item)

    def get_checked(self, node: QTreeWidgetItem)->list:
        """ 得到当前节点选中的所有分支， 返回一个 list """
        temp_list = []
        # 此处看下方注释 1
        for item in node.takeChildren():
            # 判断是否选中
            if item.checkState(0) == Qt.Checked:
                temp_list.append(item.text(0))
                # 判断是否还有子分支
                if item.childCount():
                    temp_list.extend(self.get_checked(item))
            node.addChild(item)
        print(temp_list)
        return temp_list


if __name__ == '__main__':
    app = QApplication(sys.argv)
    win = Demo()
    win.show()
    sys.exit(app.exec_())

注释01：在这个函数中，我传入了一个 node 节点，takeChildren() 这个方法会将该node节点的所有一级子分支拿出来(删除)，并返回节点的所有一级分支的列表，如下所示。该方法只能返回一级的节点信息，利用 childCount() 来判断是否有子分支，有则递归，一直到最底部的节点。因为在获取的时候 takeChildren() 删除了所有节点，所以在操作结束后重新加入到 node 节点中

[
<PyQt5.QtWidgets.QTreeWidgetItem object at 0x0000000008464708>, 
<PyQt5.QtWidgets.QTreeWidgetItem object at 0x0000000008464798>, 
]

这样的做法有什么好处和坏处？

最大的好处无疑是不需要创建额外的变量去存储子节点的信息，子节点的信息和顺序都是实时获取的而非前期就定好了的。坏处，我设想这个方法用多了可能会存在节点顺序改变的情况。比如“苹果、香蕉”变成了“香蕉、苹果”，目前未出现。

对比网上的方法

有一个关于 QTreeWidgetItemIterator 的办法，这是Qt中自带的遍历器，大概如下

item = QtWidgets.QTreeWidgetItemIterator(self.treeWidget)，

用 item.value() 来定位到一个节点，item.value() 的实例就是上文列表中的那种对象，个人感觉差不太多。

还有一种比较暴力做法。在生成子节点的时候将所有子节点放到当前类的作用域中，也就是作为属性存在。

self.item1 = QTreeWidgetItem()

或是生成的时候保存在一个定义在作用域的列表中，这么做有一个坏处，节点的信息都是提前定好了的。但实际上遇到的情况更多应该是未知的。

self.item_list = []
self.item_list.append([... ... ]) 

到此这篇关于PyQt5 QTreeWidget 树形结构递归遍历当前所有节点的实现的文章就介绍到这了,更多相关PyQt5 QTreeWidget 树形结构遍历内容请搜索猪先飞以前的文章或继续浏览下面的相关文章希望大家以后多多支持猪先飞！