QTreeWidget子项插入及ComboBox切换TextEdit的实现问询

Let's walk through a solid implementation that matches your requirements—your idea of attaching parent item properties to ComboBoxes and using setItemWidget is spot-on, we just need to flesh out the details to make it work smoothly.

Step 1: Attach Parent Item Metadata to ComboBoxes #

When creating each ComboBox for your QTreeWidget cells, you need a way to link the widget back to its corresponding QTreeWidgetItem and column. Qt's QObject.setProperty() is perfect for this—it lets you store custom data directly on the ComboBox without messy global tracking:

from PyQt5.QtWidgets import (QApplication, QTreeWidget, QTreeWidgetItem, 
                             QComboBox, QTextEdit, QPushButton, QVBoxLayout, QWidget)

def create_combo_box(item, column):
    combo = QComboBox()
    combo.addItems(["Select Option", "Add Text Edit", "Keep ComboBox"])
    # Store references to the parent tree item and its column
    combo.setProperty("tree_item", item)
    combo.setProperty("column", column)
    # Connect selection change signal to our handler
    combo.currentIndexChanged.connect(on_combo_selection_changed)
    return combo

Step 2: Switch Between ComboBox and TextEdit on Selection #

We'll handle two common scenarios here: replacing the ComboBox with a TextEdit in the same cell, or adding the TextEdit to an adjacent column. Pick the one that fits your UI needs.

Option A: Replace ComboBox with TextEdit (Same Cell) #

This replaces the ComboBox entirely when the target option is selected, and restores it if the user switches back:

def on_combo_selection_changed(index):
    combo = sender()
    item = combo.property("tree_item")
    column = combo.property("column")
    
    if combo.currentText() == "Add Text Edit":
        # Create and configure the TextEdit
        text_edit = QTextEdit()
        text_edit.setPlaceholderText("Type your content here...")
        # Save reference to original ComboBox to restore later
        text_edit.setProperty("original_combo", combo)
        # Replace the ComboBox with TextEdit
        tree_widget.setItemWidget(item, column, text_edit)
    else:
        # Check if current widget is a TextEdit, then restore the ComboBox
        current_widget = tree_widget.itemWidget(item, column)
        if isinstance(current_widget, QTextEdit):
            original_combo = current_widget.property("original_combo")
            tree_widget.setItemWidget(item, column, original_combo)
            # Clean up the unused TextEdit to avoid memory leaks
            current_widget.deleteLater()

Option B: Add TextEdit Next to ComboBox (Adjacent Column) #

If you want to keep the ComboBox visible and add the TextEdit in the next column:

def on_combo_selection_changed(index):
    combo = sender()
    item = combo.property("tree_item")
    column = combo.property("column")
    
    if combo.currentText() == "Add Text Edit":
        # Create TextEdit and place it in the next column
        text_edit = QTextEdit()
        text_edit.setPlaceholderText("Type your content here...")
        tree_widget.setItemWidget(item, column + 1, text_edit)
    else:
        # Remove the TextEdit if it exists
        text_edit = tree_widget.itemWidget(item, column + 1)
        if isinstance(text_edit, QTextEdit):
            tree_widget.setItemWidget(item, column + 1, None)
            text_edit.deleteLater()

Step 3: Implement Child Item Insertion #

To add a child item to the selected row, we'll grab the current selected item, create a new child, and initialize it with a ComboBox using our existing setup function:

def add_child_item():
    selected_items = tree_widget.selectedItems()
    if not selected_items:
        return
    parent_item = selected_items[0]
    # Create new child item
    child_item = QTreeWidgetItem(parent_item)
    # Initialize child with a ComboBox (match parent's column setup)
    combo = create_combo_box(child_item, 0)
    tree_widget.setItemWidget(child_item, 0, combo)
    # Expand the parent to show the new child immediately
    parent_item.setExpanded(True)

Step 4: Full Working Example #

Here's how to tie everything together into a runnable UI:

if __name__ == "__main__":
    app = QApplication([])
    
    # Main window setup
    window = QWidget()
    layout = QVBoxLayout(window)
    
    tree_widget = QTreeWidget()
    tree_widget.setColumnCount(2)  # Use 2 columns if using the adjacent TextEdit option
    layout.addWidget(tree_widget)
    
    # Add sample parent items
    for i in range(3):
        parent_item = QTreeWidgetItem(tree_widget)
        combo = create_combo_box(parent_item, 0)
        tree_widget.setItemWidget(parent_item, 0, combo)
    
    # Button to trigger child item insertion
    add_child_btn = QPushButton("Add Child Item to Selected Row")
    add_child_btn.clicked.connect(add_child_item)
    layout.addWidget(add_child_btn)
    
    window.show()
    app.exec_()

Key Tips for Robustness #

• Signal Sender: Using sender() lets us handle all ComboBoxes with a single handler, no need for individual signal connections per widget.
• Widget Cleanup: Always call deleteLater() on widgets you remove to prevent memory leaks—Qt won't automatically clean up widgets removed from setItemWidget.
• Property Reusability: Storing references via setProperty() keeps your code clean and avoids complex class hierarchies or global variables.
• Consistency: Reusing create_combo_box for child items ensures they behave exactly like parent items, maintaining a uniform UI.

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