PyQt5 QTreeWidget同层级节点对等拖拽实现异常排查

现有代码逻辑存在三个核心缺陷，导致非第二层节点拖拽不符合要求：

• 第一层（根节点）判断逻辑缺失父节点校验：代码对type==0的根节点直接放行所有拖拽事件，没有判断目标位置是否属于根节点层，导致根节点可以被拖入其他根节点内部。
• 未处理拖放指示位置判断：拖拽到节点上方/下方的同层排序场景、拖拽到空白区域的场景没有做统一的父节点匹配，硬编码按type分层判断的逻辑无法覆盖第三层节点跨父节点判断、同层边缘放置的场景，比如node1-1-1和node1-2-2父节点不同，逻辑里没有正确识别父节点差异，同时同属一个父节点的三级节点拖拽时也会因为判断条件不匹配被拦截。
• 硬编码节点type做层级判断扩展性极差，不管节点有多少层，核心校验规则不需要按层级单独写判断。

放弃按节点type分层判断的逻辑，抽离统一的拖拽合法性校验方法，所有节点共用一套规则：

1. 拖拽源节点和目标位置所属的父节点必须完全一致：根节点的父节点为树控件本身，子节点的父节点为其直接上级QTreeWidgetItem
2. 禁止所有节点接收内部拖放（即拖到节点正上方的OnItem位置全部拦截），只允许拖到同层节点的上方/下方做排序，从根源上避免节点被拖入其他节点内部修改层级
3. 拖拽到树控件空白区域时，仅允许根节点放置，其他层级节点拖到空白区域直接拦截。

from PyQt5.QtWidgets import QTreeWidget, QTreeWidgetItem, QMainWindow, QApplication
from PyQt5.QtCore import Qt
import sys

class TreeWidget(QTreeWidget):
    def __init__(self):
        super().__init__()
        self.setDragDropMode(QTreeWidget.InternalMove)
        self.setExpandsOnDrag(False) # 关闭拖拽时自动展开节点的默认行为

    def _check_drop_allowed(self, event):
        # 统一校验拖拽是否合法，dragMoveEvent和dropEvent复用逻辑
        source_item = self.currentItem()
        if not source_item:
            return False
        
        target_item = self.itemAt(event.pos())
        drop_pos = self.dropIndicatorPosition()
        # 统一计算源节点的父节点：根节点父节点为树控件本身，其余为直接上级item
        source_parent = source_item.parent() if source_item.parent() else self

        if not target_item:
            # 拖到空白区域，仅根节点允许放置
            return source_parent == self
        
        if drop_pos == QTreeWidget.OnItem:
            # 禁止拖到任何节点内部，避免修改父子层级
            return False
        else:
            # 拖到目标节点上/下方，校验双方父节点是否一致
            target_parent = target_item.parent() if target_item.parent() else self
            return source_parent == target_parent

    def dragMoveEvent(self, event):
        if self._check_drop_allowed(event):
            super().dragMoveEvent(event)
        else:
            event.ignore()

    def dropEvent(self, event):
        if self._check_drop_allowed(event):
            super().dropEvent(event)
        else:
            event.ignore()


class BasicTreeTest1(QMainWindow):
    def __init__(self):
        super().__init__()
        self.setWindowTitle('QTreeWidget')
        self.tree = TreeWidget()

        # 初始化第一层节点
        root1 = QTreeWidgetItem(self.tree)
        root1.setText(0, 'node1')
        root1.setFlags(root1.flags() & ~Qt.ItemIsDropEnabled)

        root2 = QTreeWidgetItem(self.tree)
        root2.setText(0, 'node2')
        root2.setFlags(root2.flags() & ~Qt.ItemIsDropEnabled)

        root3 = QTreeWidgetItem(self.tree)
        root3.setText(0, 'node3')
        root3.setFlags(root3.flags() & ~Qt.ItemIsDropEnabled)

        # 初始化第二层节点
        child1_1 = QTreeWidgetItem(root1)
        child1_1.setText(0, 'node1-1')
        child1_1.setFlags(child1_1.flags() & ~Qt.ItemIsDropEnabled)

        child1_2 = QTreeWidgetItem(root1)
        child1_2.setText(0, 'node1-2')
        child1_2.setFlags(child1_2.flags() & ~Qt.ItemIsDropEnabled)

        child2_1 = QTreeWidgetItem(root2)
        child2_1.setText(0, 'node2-1')
        child2_1.setFlags(child2_1.flags() & ~Qt.ItemIsDropEnabled)

        child2_2 = QTreeWidgetItem(root2)
        child2_2.setText(0, 'node2-2')
        child2_2.setFlags(child2_2.flags() & ~Qt.ItemIsDropEnabled)

        child3_1 = QTreeWidgetItem(root3)
        child3_1.setText(0, 'node3_1')
        child3_1.setFlags(child3_1.flags() & ~Qt.ItemIsDropEnabled)

        child3_2 = QTreeWidgetItem(root3)
        child3_2.setText(0, 'node3_2')
        child3_2.setFlags(child3_2.flags() & ~Qt.ItemIsDropEnabled)

        # 初始化第三层节点
        child1_1_1 = QTreeWidgetItem(child1_1)
        child1_1_1.setText(0, 'node1-1-1')
        child1_1_1.setFlags(child1_1_1.flags() & ~Qt.ItemIsDropEnabled)

        child1_2_1 = QTreeWidgetItem(child1_1)
        child1_2_1.setText(0, 'node1-2-1')
        child1_2_1.setFlags(child1_2_1.flags() & ~Qt.ItemIsDropEnabled)

        child1_2_2 = QTreeWidgetItem(child1_2)
        child1_2_2.setText(0, 'node1-2-2')
        child1_2_2.setFlags(child1_2_2.flags() & ~Qt.ItemIsDropEnabled)

        child3_2_1 = QTreeWidgetItem(child3_2)
        child3_2_1.setText(0, 'node3-2-1')
        child3_2_1.setFlags(child3_2_1.flags() & ~Qt.ItemIsDropEnabled)

        child3_2_2 = QTreeWidgetItem(child3_2)
        child3_2_2.setText(0, 'node3-2-2')
        child3_2_2.setFlags(child3_2_2.flags() & ~Qt.ItemIsDropEnabled)

        self.setCentralWidget(self.tree)
        self.tree.expandAll()

if __name__ == '__main__':
    app = QApplication(sys.argv)
    w = BasicTreeTest1()
    w.show()
    sys.exit(app.exec_())

• 所有层级节点仅允许在同属一个父节点的范围内拖拽排序，跨父节点拖拽会被直接拦截
• 任何节点都无法被拖入其他节点内部，不会修改原有树层级结构
• 逻辑自动适配任意层级深度的节点，后续新增四级、五级节点不需要修改拖拽校验代码。

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