PyQt6实现QTreeView同层级元素重排的问题求助

要实现仅允许元素在原父节点的同级范围内重排，核心是在dragMoveEvent里精准判断拖拽目标的合法性，同时拦截会改变层级的拖拽动作。以下是具体实现方案：

关键判断逻辑 #

需要同时满足三个条件才允许拖拽：

• 拖拽源和目标的父索引完全相同（确保在同一父文件夹下）
• 目标位置是同级插入（AboveItem/BelowItem），而非作为子节点插入（OnItem）
• 拖拽动作是移动（MoveAction），避免误处理复制等其他操作

完整代码示例 #

from PyQt6.QtWidgets import QTreeView, QStandardItemModel, QStandardItem, QApplication
from PyQt6.QtCore import Qt, QModelIndex

class RestrictedTreeView(QTreeView):
    def __init__(self, parent=None):
        super().__init__(parent)
        # 设置内部拖拽模式
        self.setDragDropMode(QTreeView.DragDropMode.InternalMove)
        self.setDefaultDropAction(Qt.DropAction.MoveAction)

    def dragMoveEvent(self, event):
        # 获取源索引和目标索引
        source_index = self.model().index(event.source().currentIndex().row(), 
                                         event.source().currentIndex().column(),
                                         event.source().currentIndex().parent())
        target_index = self.indexAt(event.pos())

        # 情况1：目标是根节点或无有效索引，直接拒绝
        if not target_index.isValid() or not source_index.isValid():
            event.ignore()
            return

        # 情况2：源和目标父节点不同，拒绝跨父节点拖拽
        if source_index.parent() != target_index.parent():
            event.ignore()
            return

        # 情况3：目标位置是要将源作为子节点插入（OnItem），拒绝
        if self.dropIndicatorPosition() == QTreeView.DropIndicatorPosition.OnItem:
            event.ignore()
            return

        # 情况4：拖拽到自身位置，无意义，拒绝
        if source_index == target_index:
            event.ignore()
            return

        # 所有条件满足，允许拖拽
        event.accept()

if __name__ == "__main__":
    app = QApplication([])
    # 构建测试模型
    model = QStandardItemModel()
    root = model.invisibleRootItem()

    # 添加父节点AAA
    parent_aaa = QStandardItem("AAA")
    root.appendRow(parent_aaa)
    parent_aaa.appendRow(QStandardItem("A1"))
    parent_aaa.appendRow(QStandardItem("A2"))
    parent_aaa.appendRow(QStandardItem("A3"))

    # 添加父节点BBB
    parent_bbb = QStandardItem("BBB")
    root.appendRow(parent_bbb)
    parent_bbb.appendRow(QStandardItem("B1"))
    parent_bbb.appendRow(QStandardItem("B2"))

    # 配置TreeView
    tree = RestrictedTreeView()
    tree.setModel(model)
    tree.expandAll()
    tree.show()
    app.exec()

核心代码解释 #

1. 拖拽模式设置：setDragDropMode(InternalMove)确保只能在树内部拖拽移动，setDefaultDropAction(MoveAction)默认使用移动动作。
2. dragMoveEvent拦截：
   • 先校验源和目标索引的有效性
   • 对比父索引，确保在同一父节点下
   • 通过dropIndicatorPosition()判断拖拽位置：OnItem表示要将源作为目标的子节点，直接拦截；只允许AboveItem或BelowItem的同级插入
   • 排除拖拽到自身的无意义操作

额外注意事项 #

• 确保模型中的项开启了拖拽和允许放置的flags：默认QStandardItem已经包含ItemIsDragEnabled和ItemIsDropEnabled，如果是自定义模型，需要在flags()方法中返回这些标识。
• 如果需要支持多选拖拽，需要额外处理多个源索引的父节点判断，确保所有选中项的父节点都和目标父节点一致。

内容的提问来源于stack exchange，提问作者Lukas Fürst