自定义带TreeView的QComboBox无法自动选中子项求助

问题根源 #

1. QComboBox默认基于平面模型设计，setCurrentIndex(int)仅接受顶层节点的行号。子项的QModelIndex包含父节点上下文，直接取index.row()得到的是子项在父节点内的行号，而非顶层行号，导致误选父项。
2. 原生findText仅搜索顶层节点，无法匹配树形结构中的子项文本。
3. 手动选中子项时，QComboBox无法正确识别树形索引，因此currentIndexChanged返回父项的行号（通常为0）。

修复后的完整代码 #

class CustomComboBox(QtWidgets.QComboBox):

    def __init__(self, parent=None):
        super(CustomComboBox, self).__init__(parent=parent)

        self.view = QtWidgets.QTreeView()
        self.view.setHeaderHidden(True)
        self.view.setSelectionMode(QtWidgets.QAbstractItemView.SingleSelection)

        self.model = QtGui.QStandardItemModel()

        self.setModel(self.model)
        self.setView(self.view)

        # 绑定视图选中信号，同步ComboBox状态
        self.view.selectionModel().currentChanged.connect(self._sync_combo_from_view)

    def addUserItems(self, header, items):
        userItem = QtGui.QStandardItem(header)
        userItem.setSelectable(False)
        userItem.setEnabled(False)  # 彻底禁用父项选中
        for item in sorted(items):
            childItem = QtGui.QStandardItem(item)
            userItem.appendRow(childItem)
        self.model.appendRow(userItem)

        self.view.expandAll()
        self.view.setItemsExpandable(False)

    def selectChildItem(self, itemText):
        # 递归遍历所有节点查找目标子项
        def traverse_item(item):
            if item.text() == itemText:
                return item
            for row in range(item.rowCount()):
                match = traverse_item(item.child(row))
                if match:
                    return match
            return None

        target_item = None
        for row in range(self.model.rowCount()):
            target_item = traverse_item(self.model.item(row))
            if target_item:
                break

        if target_item:
            index = self.model.indexFromItem(target_item)
            # 1. 设置视图选中目标子项
            self.view.selectionModel().setCurrentIndex(index, QtWidgets.QItemSelectionModel.SelectCurrent)
            # 2. 同步ComboBox显示文本
            self.setCurrentText(target_item.text())

    def _sync_combo_from_view(self, current, previous):
        # 视图选中子项时，同步更新ComboBox状态
        if current.isValid() and self.model.itemFromIndex(current).isSelectable():
            self.setCurrentText(current.data())

    def findText(self, text, flags=QtCore.Qt.MatchExactly):
        # 重写findText，支持搜索所有子项
        def search_all_items(item):
            if item.text() == text:
                return True
            for row in range(item.rowCount()):
                if search_all_items(item.child(row)):
                    return True
            return False

        for row in range(self.model.rowCount()):
            if search_all_items(self.model.item(row)):
                return text
        return -1

核心修复点 #

• 重写selectChildItem：通过递归遍历找到目标子项，直接操作视图的选择模型完成选中，同时同步ComboBox的显示文本。
• 绑定视图currentChanged信号：手动选中子项时强制同步ComboBox状态，避免默认信号返回错误索引。
• 重写findText：支持遍历树形结构中的所有节点进行文本匹配。
• 父项添加setEnabled(False)：彻底杜绝父项被选中的可能。

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