QTreeView的使用（五）

QTreeView的系类文章如下，可根据熟练程度自行食用
QTreeView的使用（一）
QTreeView的使用（二）
QTreeView的使用（三）
QTreeView的使用（四）
本章主要讲述实现一种节点拖拽的方法，会用到之前提到过的QMimeData实现的拖拽
QMimeData拖拽的系类文章如下，可根据熟练程度自行食用
PyQt拖拽事件——mimeData的使用（一）
PyQt拖拽事件——mimeData的使用（二）

写作思路

本篇博客主要实现在QTreeView中实现节点的拖拽，QTreeView似乎本身也是有这个功能的（感兴趣的可以去研究一下QTreeView和他的父类，然后和博主探讨！）

实现效果如上，具体的删除添加方式可以根据需要去定制，已经在下面的代码中标注

# -*- coding: utf-8 -*-
from PyQt5.QtCore import Qt, QMimeData
from PyQt5.QtGui import QDrag
from PyQt5.QtWidgets import QTreeView, QApplication, QAbstractItemView
from PyQT import BluePrint


class HierarchyTreeView(QTreeView):
    def __init__(self):
        super(HierarchyTreeView, self).__init__()
        self.setSelectionMode(QAbstractItemView.ExtendedSelection)
        self.setAcceptDrops(True)
        self.startPos = None
        self.isDragItem = False
        self.pressIndex = None
        self.moveSuccess = False

    def currentChanged(self, current, previous):
        model = self.model()
        if model.data(current, Qt.UserRole) is not None:
            if model.data(current, Qt.UserRole):
                pass
        return super().currentChanged(current, previous)

    def mousePressEvent(self, event):
        if event.button() == Qt.LeftButton:
            self.startPos = event.pos()
        self.pressIndex = self.indexAt(event.pos())
        if self.pressIndex.isValid() is False:
            self.isDragItem = False
            self.setCurrentIndex(self.rootIndex())
        else:
            self.isDragItem = True
        super(HierarchyTreeView, self).mousePressEvent(event)

    def mouseMoveEvent(self, event):
        distance = (event.pos() - self.startPos).manhattanLength()
        if distance >= QApplication.startDragDistance() and self.isDragItem:
            self.moveSuccess = False
            mimeData = QMimeData()
            drag = QDrag(self)
            drag.setMimeData(mimeData)
            print("删除self.pressIndex对应的item")
            # TODO 删除index对应的item
            drag.exec(Qt.MoveAction)
            if self.moveSuccess is False:
                print("对于点击了键盘ESC 或者drop没有落在QTreeView上的 则还原删除操作")
                # TODO 撤销之前的操作

    def dragMoveEvent(self, event):
        index = self.indexAt(event.pos())
        if index.isValid() is False:
            self.setCurrentIndex(self.rootIndex())
        else:
            self.setCurrentIndex(index)
        source = event.source()
        if source and source == self:
            event.setDropAction(Qt.MoveAction)

    def dragEnterEvent(self, event):
        if event.source() == self:
            event.setDropAction(Qt.MoveAction)
            event.accept()

    def dropEvent(self, event):
        source = event.source()
        if source and source == self:
            event.setDropAction(Qt.MoveAction)
            print("在drop的地方将index对应的item加上")
            # TODO 将 index对应的item添加进来
            self.moveSuccess = True

因为上述所讲的在开篇的文章中都有介绍，因此此处就不再赘述，另外文中写了self.setSelectionMode(QAbstractItemView.ExtendedSelection)，可以实现ctrl和shift的相关多选操作