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

2021-11-28 10:55:49 作者：庸了个白

Qt中实现树形结构可以使用QTreeWidget类，也可以使用QTreeView类，本文主要介绍了PyQt5 QTreeWidget 树形结构递归遍历当前所有节点的实现，文中通过示例代码介绍的非常详细，具有一定的参考价值，感兴趣的小伙伴们可以参考一下

QTreeWidget类中的常用方法

简介：商品种类和商品的树形嵌套结构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>,