🐟🐟🐟 实现TreeWidget中的复选框及多级联动

实现方法

  • 添加复选框

QtDesigner绘制出的TreeWidget控件是不具有复选框的功能，需要通过setCheckState(index, state)来对TreeWidget中的Item进行复选框的添加，index指的是列的序号，state则为添加时复选框的三种状态（Qt.Checked，Qt.Unchecked，Qt.PartiallyChecked）。例子中通过setTreeCheckBox()来实现复选框的添加。

  1. # 给树添加复选框
  2. def setTreeCheckBox(self):
  3.     root_node = self.treeWidget.topLevelItem(0)
  4.     root_node.setCheckState(0, Qt.Unchecked)
  5.     for i in range(root_node.childCount()):
  6.         parent_node = root_node.child(i)
  7.         parent_node.setCheckState(0, Qt.Unchecked)
  8.         if parent_node.childCount() > 0:
  9.             for j in range(parent_node.childCount()):
  10.                 child_node = parent_node.child(j)
  11.                 child_node.setCheckState(0, Qt.Unchecked)
  12.     self.treeWidget.expandAll()  # 展开树
  • 实现复选框多级联动

复选框的多级联动实现主要是通过TreeWidget中自带的信号（itemChanged (QTreeWidgetItem * item , int column )）来实现。通过接收信号传输的Item对Item进行从上往下的遍历用来联动子节点的状态，以及通过自下而上的遍历来联动父节点的状态。由于在进行节点联动时，所有因为联动而改变状态的节点也会触发信号传输，所以在槽函数中不需要做递归操作。

  1. # 设置节点联动事件
  2. def onItemChange(self, item):
  3.     # 联动子节点状态
  4.     if item.checkState(0) != Qt.PartiallyChecked:
  5.         state = item.checkState(0) # 存储节点状态防止联动时导致状态进行改变
  6.         for i in range(item.childCount()):
  7.             item.child(i).setCheckState(0, state)
  8.     # 联动父节点状态
  9.     if item.parent() is not None:
  10.         check = 0
  11.         uncheck = 0
  12.         for j in range(item.parent().childCount()):
  13.             state = item.parent().child(j).checkState(0)
  14.             if state == Qt.Unchecked:
  15.                 uncheck += 1
  16.             else:
  17.                 check += 1
  18.         if check > 0 and uncheck == 0:
  19.             item.parent().setCheckState(0, Qt.Checked) # 设置父节点为全选
  20.         elif check == 0 and uncheck > 0:
  21.             item.parent().setCheckState(0, Qt.Unchecked) # 设置父节点为未选中
  22.         else:
  23.             item.parent().setCheckState(0, Qt.PartiallyChecked) # 设置父节点为部分选中

效果展示

🍙 TreeWidget 实现复选框及多级联动 - 图1

UI部分

  1. from PySide2.QtCore import *
  2. from PySide2.QtGui import *
  3. from PySide2.QtWidgets import *
  6. class Ui_Dialog(object):
  7.     def setupUi(self, Dialog):
  8.         if not Dialog.objectName():
  9.             Dialog.setObjectName(u"Dialog")
  10.         Dialog.resize(400, 300)
  11.         self.horizontalLayout = QHBoxLayout(Dialog)
  12.         self.horizontalLayout.setObjectName(u"horizontalLayout")
  13.         self.treeWidget = QTreeWidget(Dialog)
  14.         __qtreewidgetitem = QTreeWidgetItem(self.treeWidget)
  15.         QTreeWidgetItem(__qtreewidgetitem)
  16.         __qtreewidgetitem1 = QTreeWidgetItem(__qtreewidgetitem)
  17.         QTreeWidgetItem(__qtreewidgetitem1)
  18.         QTreeWidgetItem(__qtreewidgetitem1)
  19.         __qtreewidgetitem2 = QTreeWidgetItem(__qtreewidgetitem)
  20.         QTreeWidgetItem(__qtreewidgetitem2)
  21.         QTreeWidgetItem(__qtreewidgetitem2)
  22.         self.treeWidget.setObjectName(u"treeWidget")
  24.         self.horizontalLayout.addWidget(self.treeWidget)
  27.         self.retranslateUi(Dialog)
  29.         QMetaObject.connectSlotsByName(Dialog)
  30.     # setupUi
  32.     def retranslateUi(self, Dialog):
  33.         Dialog.setWindowTitle(QCoreApplication.translate("Dialog", u"Dialog", None))
  34.         ___qtreewidgetitem = self.treeWidget.headerItem()
  35.         ___qtreewidgetitem.setText(0, QCoreApplication.translate("Dialog", u"Test", None));
  37.         __sortingEnabled = self.treeWidget.isSortingEnabled()
  38.         self.treeWidget.setSortingEnabled(False)
  39.         ___qtreewidgetitem1 = self.treeWidget.topLevelItem(0)
  40.         ___qtreewidgetitem1.setText(0, QCoreApplication.translate("Dialog", u"\u5168\u90e8\u9879\u76ee", None));
  41.         ___qtreewidgetitem2 = ___qtreewidgetitem1.child(0)
  42.         ___qtreewidgetitem2.setText(0, QCoreApplication.translate("Dialog", u"\u9879\u76ee1", None));
  43.         ___qtreewidgetitem3 = ___qtreewidgetitem1.child(1)
  44.         ___qtreewidgetitem3.setText(0, QCoreApplication.translate("Dialog", u"\u9879\u76ee2", None));
  45.         ___qtreewidgetitem4 = ___qtreewidgetitem3.child(0)
  46.         ___qtreewidgetitem4.setText(0, QCoreApplication.translate("Dialog", u"hello", None));
  47.         ___qtreewidgetitem5 = ___qtreewidgetitem3.child(1)
  48.         ___qtreewidgetitem5.setText(0, QCoreApplication.translate("Dialog", u"world", None));
  49.         ___qtreewidgetitem6 = ___qtreewidgetitem1.child(2)
  50.         ___qtreewidgetitem6.setText(0, QCoreApplication.translate("Dialog", u"\u9879\u76ee3", None));
  51.         ___qtreewidgetitem7 = ___qtreewidgetitem6.child(0)
  52.         ___qtreewidgetitem7.setText(0, QCoreApplication.translate("Dialog", u"bye", None));
  53.         ___qtreewidgetitem8 = ___qtreewidgetitem6.child(1)
  54.         ___qtreewidgetitem8.setText(0, QCoreApplication.translate("Dialog", u"good", None));
  55.         self.treeWidget.setSortingEnabled(__sortingEnabled)

逻辑代码

  1. from PySide2.QtCore import *
  2. from PySide2.QtGui import *
  3. from PySide2.QtWidgets import *
  5. from Test import Ui_Dialog
  7. class TestDialog(QDialog, Ui_Dialog):
  9.     def __init__(self):
  10.         super().__init__()
  11.         self.setupUi(self)
  12.         self.callback()
  14.     def setupUi(self, Dialog):
  15.         super().setupUi(Dialog)
  16.         self.setTreeCheckBox()
  17.         Dialog.setWindowFlags(Qt.FramelessWindowHint)
  19.     # 回调函数
  20.     def callback(self):
  21.         self.treeWidget.itemChanged.connect(self.onItemChange)
  23.     # 给树添加复选框
  24.     def setTreeCheckBox(self):
  25.         root_node = self.treeWidget.topLevelItem(0)
  26.         root_node.setCheckState(0, Qt.Unchecked)
  27.         for i in range(root_node.childCount()):
  28.             parent_node = root_node.child(i)
  29.             parent_node.setCheckState(0, Qt.Unchecked)
  30.             if parent_node.childCount() > 0:
  31.                 for j in range(parent_node.childCount()):
  32.                     child_node = parent_node.child(j)
  33.                     child_node.setCheckState(0, Qt.Unchecked)
  34.         self.treeWidget.expandAll()  # 展开树
  36.     # 设置节点联动事件
  37.     def onItemChange(self, item):
  38.         if item.checkState(0) != Qt.PartiallyChecked:
  39.             state = item.checkState(0)
  40.             for i in range(item.childCount()):
  41.                 item.child(i).setCheckState(0, state)
  43.         if item.parent() is not None:
  44.             check = 0
  45.             uncheck = 0
  46.             for j in range(item.parent().childCount()):
  47.                 state = item.parent().child(j).checkState(0)
  48.                 if state == Qt.Unchecked:
  49.                     uncheck += 1
  50.                 else:
  51.                     check += 1
  52.             if check > 0 and uncheck == 0:
  53.                 item.parent().setCheckState(0, Qt.Checked)
  54.             elif check == 0 and uncheck > 0:
  55.                 item.parent().setCheckState(0, Qt.Unchecked)
  56.             else:
  57.                 item.parent().setCheckState(0, Qt.PartiallyChecked)
  60. if __name__ == '__main__':
  61.     import sys
  62.     app = QApplication(sys.argv)
  63.     window = TestDialog()
  64.     window.show()
  65.     sys.exit(app.exec_())