树形控件QTreeWidget

 1 import sys
 2 from PyQt5.QtCore import Qt
 3 from PyQt5.QtWidgets import QApplication, QWidget, QTreeWidget, QTreeWidgetItem, QLabel, QHBoxLayout
 4 
 5 
 6 class Demo(QWidget):
 7     def __init__(self):
 8         super(Demo, self).__init__()
 9         self.resize(500, 300)
10         self.label = QLabel('No Click')
11 
12         self.tree = QTreeWidget(self)  # 实例化树形控件
13         self.tree.setColumnCount(1)  #将该树状控件的列数设为2(默认为1列)
14         self.tree.setHeaderLabels(['树形控件', '列'])
15         #设置每列的标题，如果只有一列的话，则应该通过setHeaderLabel(str)方法设置
16 
17         self.tree.itemClicked.connect(self.change_func)#单击控件时发出信号
18         #这个信号会自动传递两个参数
19         #参数1就是被单击的项目对象。<PyQt5.QtWidgets.QTreeWidgetItem object at 0x0000008269BE7EE8>
20         #参数2就是单击的列序号
21 
22         self.preview = QTreeWidgetItem(self.tree)  # 实例化一个项目。参数：指定父类
23         self.preview.setText(0, '顶层项目')  #给项目设置文本。参数1：列序号
24         self.preview.setText(1, '顶层项目-1列')
25 
26         self.preview1 = QTreeWidgetItem()
27         self.preview1.setText(0, '顶层项目1')
28         self.preview1.setText(1, '顶层项目1-1列')
29         self.tree.addTopLevelItem(self.preview1)  #把self.preview1项目设置为顶层项目
30 
31         self.qt5112 = QTreeWidgetItem()
32         self.qt5112.setText(0, 'Qt 5112 snapshot')
33         self.qt5112.setCheckState(0, Qt.Unchecked)  #设置复选框的位置和状态
34         #参数1 复选框的位置 列序号
35         #Qt.Unchecked   没选中
36         #Qt.Checked    选中
37         #Qt.PartiallyChecked   部分选中
38 
39         self.preview.addChild(self.qt5112)#给self.preview添加子项目self.qt5112
40         #给self.qt5112添加父类
41 
42         choice_list = ['macOS', 'Android x86', 'Android ARMv7', 'Sources', 'iOS']
43         self.item_list = []
44         for i, c in enumerate(choice_list):
45             item = QTreeWidgetItem(self.qt5112)
46             item.setText(0, c)
47             item.setCheckState(0, Qt.Unchecked)
48             self.item_list.append(item)
49 
50         self.test_item = QTreeWidgetItem(self.qt5112)
51         self.test_item.setText(0, 'test1')
52         self.test_item.setText(1, 'test2')
53 
54         self.tree.expandAll()
55         #让所有的项都是以打开状态显示的。注意必须要在所有项都已经实例化好之后再调用该方法，如果一开始就调用则会没有效果
56 
57         self.h_layout = QHBoxLayout()
58         self.h_layout.addWidget(self.tree)
59         self.h_layout.addWidget(self.label)
60         self.setLayout(self.h_layout)
61 
62     def change_func(self, item, column):
63         self.label.setText(item.text(column))
64         #item.text(column)  返回项目指定列的文本
65 
66         if item == self.qt5112:
67             if self.qt5112.checkState(0) == Qt.Checked:#判断self.qt5112复选框的状态。参数复选框位置：列序号
68                 #self.qt5112.checkState(0)   返回复选框的状态
69                 [x.setCheckState(0, Qt.Checked) for x in self.item_list]
70             else:
71                 [x.setCheckState(0, Qt.Unchecked) for x in self.item_list]
72         else:
73             check_count = 0
74             for x in self.item_list:
75                 if x.checkState(0) == Qt.Checked:
76                     check_count += 1
77 
78             if check_count == 5:
79                 self.qt5112.setCheckState(0, Qt.Checked)
80             elif 0 < check_count < 5:
81                 self.qt5112.setCheckState(0, Qt.PartiallyChecked)  #设置复选框为半选中状态
82             else:
83                 self.qt5112.setCheckState(0, Qt.Unchecked)
84 
85 
86 if __name__ == '__main__':
87     app = QApplication(sys.argv)
88     demo = Demo()
89     demo.show()
90     sys.exit(app.exec_())