第二十七篇 -- QTreeWidget总结
前言
之前写过几篇关于TreeWidget的文章，不过不方便查阅，特此重新整合作为总结。不过关于QtDesigner画图，还是不重新写了，看 第一篇 就OK。
准备工作
1. 用QtDesigner画一个QTreeWidget
2. 自己事先确定好层次与列的类型。
QtreeWidget的展开层次类定义。
# 节点类型的枚举类型 class TreeItemType(Enum): itGroupItem = 1001 # 群组 itMemberItem = 1002 # 成员
QTreeWidget列类型类定义。
class TreeColNum(Enum): # 目录树的列号的枚举类型 col_item_group = 0 # 组 col_item_name = 1 # 姓名 col_item_sex = 2 # 性别 col_item_score = 3 # 分数 col_item_excellent = 4 # 优秀
根据上述定义，此QTreeWidget有两层节点，有5列。
QTreeWidget样式的基本方法
1. 设置表头
# 设置表头 self.ui.treeWidget.setHeaderLabels(['Group', 'Name', 'Sex', 'Score', 'Excellent'])
2. 设置表头背景色
# 设置表头颜色 self.ui.treeWidget.setStyleSheet("QHeaderView::section{background:rgb(85, 181, 200);}")
3. 设置表头前景色
# 设置表头前景色 brush = QBrush(QColor(255, 240, 0)) brush.setStyle(Qt.SolidPattern) self.ui.treeWidget.headerItem().setForeground(0, brush)
4. 隐藏某列
# 隐藏某一列 self.ui.treeWidget.hideColumn(TreeColNum.col_item_excellent.value)
5. 设置列宽
# 设置列宽 self.ui.treeWidget.setColumnWidth(1, 150) # 第1列宽150
6. 清除QtreeWidget数据
self.ui.treeWidget.clear()
7. 添加根节点
# 定义一个节点类型 item = QTreeWidgetItem(TreeItemType.itGroupItem.value) # 设置节点的列对应的文本，此例为在group列写入数据group0,1,2,3... item.setText(TreeColNum.col_item_group.value, "Group" + str(i)) # 将节点添加进QTreeWidget self.ui.treeWidget.addTopLevelItem(item)
8. 添加子节点
# 定义节点类型 item_member = QTreeWidgetItem(TreeItemType.itMemberItem.value) # 设置节点列的文本 item_member.setText(TreeColNum.col_item_group.value, str(j)) # 将节点添加到父节点 item.addChild(item_member) # 设置节点展开 item.setExpanded(True)
9. 设置表头字体大小
font = QFont() font.setPointSize(14) # 表头第一列字体大小为14 self.ui.treeWidget.headerItem().setFont(1, font)
10. 设置节点字体大小
font = QFont() font.setPointSize(14) # 设置此item的第0列字体大小为14 item.setFont(0, font)
11. 设置节点字体颜色
# 设置节点颜色 brush = QBrush(QColor(210, 80, 234)) brush.setStyle(Qt.SolidPattern) item.setForeground(0, brush)
12. 设置节点字体背景色
# 设置背景色 item.setBackground(1, brush)
13. 定位设置背景色
# 第1个根节点的第二列设置为红色 brush = QBrush(QColor(255, 0, 0)) # 红色 brush.setStyle(Qt.SolidPattern) self.ui.treeWidget.topLevelItem(1).setBackground(2, brush) # 第0个根节点的的第1个子节点的第2列设置为黄色 brush = QBrush(QColor(255, 255, 0)) # 黄色 brush.setStyle(Qt.SolidPattern) self.ui.treeWidget.topLevelItem(0).child(1).setBackground(2, brush)
QTreeWidget操作的基本方法
1. 设置滚轮定位
# 设置滚轮到第5个节点的第二个子节点位置 child_item = self.ui.treeWidget.topLevelItem(5).child(2) self.ui.treeWidget.scrollToItem(child_item)
2. 设定绑定事件，当TreeWidget被单击时，执行绑定函数
绑定事件
# 设置绑定事件 self.ui.treeWidget.clicked.connect(self.onTreeClicked)
绑定函数
def onTreeClicked(self, qmodelindex): item = self.ui.treeWidget.currentItem() # 获取父节点 parent = item.parent() index_row = -1 if parent is None: # 获取当前节点的序号 index_top = self.ui.treeWidget.indexOfTopLevelItem(item) print("当前在根节点Group" + str(index_top)) else: # 获取父节点的序号 index_top = self.ui.treeWidget.indexOfTopLevelItem(parent) print("当前在父节点Group" + str(index_top), end="") # 获取当前节点的序号 index_row = parent.indexOfChild(item) print("下的" + str(index_row) + "号子节点", end="") # 打印当前行的内容 print("Group=%s ,Name=%s, Sex=%s, Score=%s" % (item.text(0), item.text(1), item.text(2), item.text(3)))
item = self.treeWidget.currentItem是获取当前选中行
这个绑定函数是针对有两层节点而言的。index_top是第一层节点的序号，index_row是当前节点的序号。
这部分获取当前行的树节点的层次，如果需要获取当前行的内容，item.text(0)表示当前行第0列的内容。
比如：
Group5的第0个子节点: item.text(0) = "0"，item.text(1) = "Alisa0", item.text(2) = "girl"
如果想要获取父节点的文本，同理，parent.text(0), parent.text(1),......
源码
#!/usr/bin/env python # _*_ coding: UTF-8 _*_ """================================================= @Project -> File : six-dialog_design -> myUI_treewidget.py @IDE : PyCharm @Author : zihan @Date : 2020/6/3 8:51 @Desc :QTreeWidget的基本样式方法 =================================================""" import sys from PyQt5.QtWidgets import QApplication, QWidget, QTreeWidgetItem from PyQt5.QtGui import QFont, QBrush, QColor from PyQt5.QtCore import Qt from ui_treewidget import Ui_Form from enum import Enum # 节点类型的枚举类型 class TreeItemType(Enum): itGroupItem = 1001 # 群组 itMemberItem = 1002 # 成员 class TreeColNum(Enum): # 目录树的列号的枚举类型 col_item_group = 0 # 组 col_item_name = 1 # 姓名 col_item_sex = 2 # 性别 col_item_score = 3 # 分数 col_item_excellent = 4 # 优秀 class QmyWidget(QWidget): def __init__(self, parent=None): super().__init__(parent) # 调用父类构造函数 self.ui = Ui_Form() # 创建UI对象 self.ui.setupUi(self) # 构造UI self.initial_tree() # 设置绑定事件 self.ui.treeWidget.clicked.connect(self.onTreeClicked) def onTreeClicked(self, qmodelindex): item = self.ui.treeWidget.currentItem() # 获取父节点 parent = item.parent() index_row = -1 if parent is None: # 获取当前节点的序号 index_top = self.ui.treeWidget.indexOfTopLevelItem(item) print("当前在根节点Group" + str(index_top)) else: # 获取父节点的序号 index_top = self.ui.treeWidget.indexOfTopLevelItem(parent) print("当前在父节点Group" + str(index_top), end="") # 获取当前节点的序号 index_row = parent.indexOfChild(item) print("下的" + str(index_row) + "号子节点", end="") # 打印当前行的内容 print("Group=%s ,Name=%s, Sex=%s, Score=%s" % (item.text(0), item.text(1), item.text(2), item.text(3))) def initial_tree(self): # 设置表头 self.ui.treeWidget.setHeaderLabels(['Group', 'Name', 'Sex', 'Score', 'Excellent']) # 设置表头背景色 self.ui.treeWidget.setStyleSheet("QHeaderView::section{background:rgb(85, 181, 200);}") # 设置表头前景色 brush = QBrush(QColor(255, 240, 0)) brush.setStyle(Qt.SolidPattern) self.ui.treeWidget.headerItem().setForeground(0, brush) # 隐藏某一列 self.ui.treeWidget.hideColumn(TreeColNum.col_item_excellent.value) # 设置列宽 self.ui.treeWidget.setColumnWidth(1, 150) # 第1列宽150 # 假设QTreeWidget第一层的节点有10个 for i in range(10): # 定义一个节点类型 item = QTreeWidgetItem(TreeItemType.itGroupItem.value) # 设置节点的列对应的文本，此例为在group列写入数据group0,1,2,3... item.setText(TreeColNum.col_item_group.value, "Group" + str(i)) # 设置根节点字体大小 font = QFont() font.setPointSize(14) item.setFont(0, font) # 设置节点颜色 brush = QBrush(QColor(210, 80, 234)) brush.setStyle(Qt.SolidPattern) # 设置前景色 item.setForeground(0, brush) # 设置背景色 item.setBackground(1, brush) # 将节点添加进QTreeWidget self.ui.treeWidget.addTopLevelItem(item) # 假设QTreeWidget第二层的节点有3个 for j in range(3): item_member = QTreeWidgetItem(TreeItemType.itMemberItem.value) item_member.setText(TreeColNum.col_item_group.value, str(j)) item_member.setText(TreeColNum.col_item_name.value, "Alisa" + str(j)) if j % 2 == 0: item_member.setText(TreeColNum.col_item_sex.value, "girl") item_member.setText(TreeColNum.col_item_score.value, "99") item_member.setText(TreeColNum.col_item_excellent.value, "True") else: item_member.setText(TreeColNum.col_item_sex.value, "boy") item_member.setText(TreeColNum.col_item_score.value, "50") item_member.setText(TreeColNum.col_item_excellent.value, "False") item.addChild(item_member) item.setExpanded(True) # 设置节点展开 font = QFont() font.setPointSize(14) self.ui.treeWidget.headerItem().setFont(0, font) # 第1个根节点的第二列设置为红色 brush = QBrush(QColor(255, 0, 0)) # 红色 brush.setStyle(Qt.SolidPattern) self.ui.treeWidget.topLevelItem(1).setBackground(2, brush) # 第0个根节点的的第1个子节点的第2列设置为黄色 brush = QBrush(QColor(255, 255, 0)) # 黄色 brush.setStyle(Qt.SolidPattern) self.ui.treeWidget.topLevelItem(0).child(1).setBackground(2, brush) # 设置右侧滚轮到指定位置 child_item = self.ui.treeWidget.topLevelItem(5).child(2) self.ui.treeWidget.scrollToItem(child_item) if __name__ == '__main__': app = QApplication(sys.argv) # 创建app form = QmyWidget() form.show() sys.exit(app.exec_())
只是为实现各种样式而写，并无要求美观实用。
