PyQt5中如何集成QTreeView与QComboBox并获取选中项值？

问题说明 #

单独运行MyTreeView类时，点击"Select All"可选中所有项，功能正常；但将其集成到QComboBox后，下拉框中的项无法点击，同时需要实现获取已选中项的值的功能。

问题原因分析 #

1. QComboBox默认行为干扰：QComboBox在检测到视图中的项被点击时，会自动关闭下拉框，导致复选框的点击操作被中断，无法正常触发勾选状态变化。
2. 重复创建模型：MainWindow中初始化MyTreeView后，又调用了一次self.treeView.create_model()，导致模型被重复构建，引发交互异常。
3. 视图选择模式冲突：默认的视图选择逻辑会与复选框的勾选操作冲突，需要调整选择模式避免干扰。

修复后的完整代码 #

import sys
from PyQt5.QtWidgets import *
from PyQt5.QtGui import *
from PyQt5.QtCore import *

data = {"select all": {'Group 1': ['item11', 'item12'], 'Group 2': ['item21', 'item22'], 'Group 3': ['item31', 'item32']}}

class MyTreeView(QTreeView):
    def __init__(self, parent=None):
        super().__init__(parent)
        self.setModel(QStandardItemModel())
        self.setHeaderHidden(True)
        # 设置无选择模式，避免与复选框操作冲突
        self.setSelectionMode(QAbstractItemView.NoSelection)
        self.root_text, self.parent_text, self.child_text = [], [], []
        self.create_model()

    def create_model(self):
        model = self.model()
        model.clear()
        for root_key, root_value in data.items():
            if root_key not in self.root_text:
                self.root_text.append(root_key)
            root_item = QStandardItem(root_key)
            root_item.setCheckable(True)
            model.appendRow(root_item)

            for parent_key, parent_value in root_value.items():
                if parent_key not in self.parent_text:
                    self.parent_text.append(parent_key)
                parent_item = QStandardItem(parent_key)
                parent_item.setCheckable(True)
                root_item.appendRow(parent_item)

                for child_value in parent_value:
                    if child_value not in self.child_text:
                        self.child_text.append(child_value)
                    child_item = QStandardItem(child_value)
                    child_item.setCheckable(True)
                    parent_item.appendRow(child_item)

        model.itemChanged.connect(self.update_children)
        self.expandAll()

    def update_children(self, item):
        # 统一文本大小写判断，避免大小写敏感问题
        if item.text().lower() == "select all":
            for i in range(item.rowCount()):
                child = item.child(i)
                child.setCheckState(item.checkState())
                self.update_children(child)
        elif item.rowCount() > 0:
            for i in range(item.rowCount()):
                child = item.child(i)
                child.setCheckState(item.checkState())

    def get_checked_items(self):
        """递归获取所有选中的项文本"""
        checked_items = []
        def traverse(item):
            if item.checkState() == Qt.Checked:
                checked_items.append(item.text())
            for i in range(item.rowCount()):
                traverse(item.child(i))
        root_item = self.model().invisibleRootItem()
        for i in range(root_item.rowCount()):
            traverse(root_item.child(i))
        # 移除"select all"本身，只保留实际选项
        if "select all" in checked_items:
            checked_items.remove("select all")
        return checked_items

class CheckableComboBox(QComboBox):
    def __init__(self, parent=None):
        super().__init__(parent)
        self.setEditable(False)
        self._skip_hide = False

    def showPopup(self):
        self._skip_hide = False
        super().showPopup()

    def hidePopup(self):
        if not self._skip_hide:
            super().hidePopup()

    def mousePressEvent(self, event):
        # 判断点击区域是否为复选框
        index = self.view().indexAt(event.pos())
        if index.isValid():
            item = self.view().model().itemFromIndex(index)
            if item.isCheckable():
                self._skip_hide = True
                # 手动切换复选框状态，避免ComboBox干扰
                item.setCheckState(Qt.Checked if item.checkState() == Qt.Unchecked else Qt.Unchecked)
                return
        self._skip_hide = False
        super().mousePressEvent(event)

class MainWindow(QWidget):
    def __init__(self):
        super().__init__()
        self.setWindowTitle("带复选框的树形下拉框")
        self.resize(400, 300)

        self.comboBox = CheckableComboBox()
        self.treeView = MyTreeView()
        self.comboBox.setView(self.treeView)

        # 添加按钮用于获取选中项
        self.get_selected_btn = QPushButton("获取选中项")
        self.get_selected_btn.clicked.connect(self.print_selected_items)

        self.vbox = QVBoxLayout()
        self.setLayout(self.vbox)
        self.vbox.addWidget(self.comboBox)
        self.vbox.addWidget(self.get_selected_btn)

    def print_selected_items(self):
        selected = self.treeView.get_checked_items()
        print("已选中项:", selected)
        QMessageBox.information(self, "选中项", "\n".join(selected))

def main():
    app = QApplication(sys.argv)
    ex = MainWindow()
    ex.show()
    sys.exit(app.exec_())

if __name__ == '__main__':
    main()

关键修改点说明 #

1. 自定义CheckableComboBox：
   • 重写hidePopup和mousePressEvent方法，判断点击的是复选框时，阻止下拉框关闭，确保复选框状态能正常切换。
2. 调整MyTreeView选择模式：
   • 设置self.setSelectionMode(QAbstractItemView.NoSelection)，避免视图的选择行为干扰复选框操作。
3. 移除重复模型创建：
   • 删除MainWindow中多余的self.treeView.create_model()调用，因为MyTreeView初始化时已经创建了模型。
4. 添加获取选中项方法：
   • 在MyTreeView中实现get_checked_items递归方法，遍历所有项收集选中的文本，并排除"select all"本身。
5. 优化大小写判断：
   • 在update_children中使用item.text().lower()判断，避免大小写敏感问题。

使用说明 #

• 点击下拉框展开树形结构，勾选任意项，"Select All"可以一键选中所有子项。
• 点击"获取选中项"按钮，会弹出对话框显示所有已选中的项，同时控制台也会输出结果。

内容的提问来源于stack exchange，提问作者Rithik