PyQt6中如何实现树形嵌套QComboBox及解决显示布局问题？

需求概述 #

• 闭合状态：QComboBox显示单个值
• 展开状态：显示树形结构列表，部分/全部项支持嵌套选择子项
• 选择完成：闭合后显示选中的子项，并通过pyqtSignal返回结果

现有尝试的问题 #

1. 使用QTreeView作为QComboBox视图的方式：

wh = QComboBox()
tv = QTreeView()
wh.setView(tv)
wh.setModel(QStandardItemModel())

存在行为不一致、列表截断的问题。

2. 基于QPushButton的递归控件：
   功能符合需求，但外观完全不符合QComboBox样式，且嵌套列表会弹出独立窗口，设置父控件后出现显示截断。

核心疑问：如何让嵌套列表在自身位置显示并自适应内容尺寸？

简洁实现方案 #

思路 #

利用QComboBox的下拉菜单本质是QMenu的特性，递归构建带子菜单的树形菜单结构，既保留原生QComboBox的外观和交互逻辑，又实现嵌套选择功能，无需重写大量控件成员。

代码实现 #

from PyQt6.QtCore import pyqtSignal, Qt
from PyQt6.QtWidgets import QComboBox, QMenu, QWidget, QApplication, QVBoxLayout

class TreeComboBox(QComboBox):
    selected = pyqtSignal(str)  # 返回选中的叶子项文本

    def __init__(self, parent=None):
        super().__init__(parent)
        self.setEditable(False)
        self._root_menu = QMenu(self)
        self.setMenu(self._root_menu)
        # 点击菜单外区域时关闭菜单，同时重置显示文本
        self._root_menu.aboutToHide.connect(self._reset_display_if_no_selection)
        self._current_selected = ""

    def set_tree_data(self, data_dict: dict, parent_menu: QMenu = None):
        """递归构建树形菜单"""
        target_menu = parent_menu if parent_menu else self._root_menu
        for key, value in data_dict.items():
            if isinstance(value, dict):
                # 非叶子节点，创建子菜单
                sub_menu = target_menu.addMenu(key)
                self.set_tree_data(value, sub_menu)
            else:
                # 叶子节点，创建菜单项，点击后触发选择
                action = target_menu.addAction(key)
                action.triggered.connect(lambda checked, txt=key: self._on_leaf_selected(txt))

    def _on_leaf_selected(self, text: str):
        """叶子项选中后的处理"""
        self._current_selected = text
        self.setCurrentText(text)
        self.selected.emit(text)
        self.hidePopup()  # 关闭下拉菜单

    def _reset_display_if_no_selection(self):
        """如果没有选中叶子项，重置显示为上次选中的值"""
        if self.currentText() != self._current_selected:
            self.setCurrentText(self._current_selected)

if __name__ == '__main__':
    import sys
    main_course = {
        'meat': {
            'steak': 'steak',
            'meatballs': 'meatballs',
        },
        'fish': {
            'cod': 'cod',
            'mullet': 'mullet',
            'scallops': 'scallops'
        },
        'vegetarian': {
            'salad': {
                'mixed': 'mixed',
                'cesar': 'cesar'
            },
            'flan': 'flan'
        }
    }

    app = QApplication(sys.argv)
    mw = QWidget()
    lo = QVBoxLayout(mw)
    combo = TreeComboBox()
    combo.set_tree_data(main_course)
    combo.setCurrentText("请选择主菜")
    combo.selected.connect(lambda s: print(f"选中项：{s}"))
    lo.addWidget(combo)
    mw.show()
    sys.exit(app.exec())

方案优势 #

• 原生外观：完全保留QComboBox的样式和交互（下拉箭头、焦点状态、系统主题适配）
• 自适应显示：下拉菜单自动根据内容尺寸调整，嵌套子菜单在父菜单项右侧弹出，无截断问题
• 逻辑简洁：利用QMenu原生递归结构实现树形选择，无需重写控件事件处理
• 符合需求：闭合状态显示选中值，展开显示树形列表，选中后自动闭合并触发信号

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