PyQt QTreeView的拖放支持

在本文中，我们将介绍如何在PyQt中使用QTreeView控件，并添加拖放支持。QTreeView是Qt框架中常用的树状结构控件，可用于展示层级结构的数据。通过添加拖放支持，我们可以实现树状结构数据的拖动和放置操作。

阅读更多：PyQt 教程

1. 创建一个简单的QTreeView

首先，我们需要创建一个简单的QTreeView示例，以便进行后续的拖放支持添加。

import sys
from PyQt5.QtWidgets import QApplication, QMainWindow, QTreeView, QFileSystemModel

class MainWindow(QMainWindow):
    def __init__(self):
        super().__init__()

        self.treeView = QTreeView(self)
        self.model = QFileSystemModel()
        self.model.setRootPath("")
        self.treeView.setModel(self.model)
        self.treeView.setRootIndex(self.model.index("/"))

        self.setCentralWidget(self.treeView)

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

在上述代码中，我们创建了一个MainWindow类，继承自QMainWindow。在MainWindow的构造函数中，我们创建了一个QTreeView实例，并将其设置为主窗口的中央部件。为了使QTreeView能够显示文件系统的树状结构，我们使用QFileSystemModel作为数据模型，并将其设置为QTreeView的模型。最后，我们将QTreeView的根索引设置为文件系统模型的根索引。

运行以上代码，我们将看到一个简单的树状结构视图，显示了文件系统的内容。

2. 添加拖放支持

接下来，我们将为我们的QTreeView添加拖放支持。具体来说，我们将添加拖动和放置的功能，使树状结构的节点能够在视图中移动和重新排列。

from PyQt5.QtCore import Qt, QMimeData
from PyQt5.QtGui import QDrag

class MainWindow(QMainWindow):
    # ...

    def mousePressEvent(self, event):
        index = self.treeView.indexAt(event.pos())
        if not index.isValid():
            return

        itemData = self.model.itemData(index)
        drag = QDrag(self)
        mimeData = QMimeData()
        mimeData.setText(itemData[0])
        drag.setMimeData(mimeData)
        dropAction = drag.exec_()

    def dragEnterEvent(self, event):
        event.acceptProposedAction()

    def dragMoveEvent(self, event):
        event.acceptProposedAction()

    def dropEvent(self, event):
        index = self.treeView.indexAt(event.pos())
        if not index.isValid():
            return

        itemData = self.model.itemData(index)
        print("Dropped data: ", itemData[0])

# ...

在上述代码中，我们重写了MainWindow类的鼠标按下事件处理函数mousePressEvent，并在其中实现了对鼠标按下的节点的拖动操作。首先，我们通过所点击位置获取到的index判断其是否有效。然后，我们获取到该节点的数据，创建了一个QDrag实例并将其设置为拖动操作的来源。为了传递节点数据，我们创建了一个QMimeData实例，并将节点数据作为文本设置到QMimeData中。最后，我们调用drag.exec_()开始拖动操作，并将拖动操作的结果保存在dropAction变量中。

接下来，我们重写了dragEnterEvent、dragMoveEvent和dropEvent函数，以实现拖动进入、拖动移动和放置节点的操作。在这些函数中，我们使用event.acceptProposedAction()接受拖放操作，并在dropEvent函数中获取放置节点的数据并进行处理，这里仅简单地打印出来。

运行以上代码，我们可以看到当我们点击树状结构中的节点并拖动后，在控制台中会显示出被拖动的节点的数据。这表明拖放支持已成功添加到我们的QTreeView中。

3. 自定义拖放操作

在上一节中，我们实现了基本的拖放操作，但是默认情况下，QTreeView的拖放操作是利用节点的文本进行的。在本节中，我们将介绍如何自定义QTreeView的拖放操作，以实现更复杂的拖放需求。

class MainWindow(QMainWindow):
    # ...

    def mousePressEvent(self, event):
        index = self.treeView.indexAt(event.pos())
        if not index.isValid():
            return

        drag = QDrag(self)
        mimeData = QMimeData()
        mimeData.setData("customType", b"customData")
        drag.setMimeData(mimeData)
        drag.exec_()

    def dropEvent(self, event):
        index = self.treeView.indexAt(event.pos())
        if not index.isValid():
            return

        itemData = self.model.itemData(index)
        if event.mimeData().hasFormat("customType"):
            print("Custom data dropped!")
        elif event.mimeData().hasText():
            print("Text data dropped: ", event.mimeData().text())

# ...

在上述代码中，我们重写了mousePressEvent函数，并在其中实现了自定义的拖动操作。与之前不同的是，我们使用mimeData.setData()方法设置了一个自定义的MIME类型，并为其设置了一个自定义的数据。这里我们将自定义类型设置为”customType”，自定义数据设置为”customData”。

然后，在dropEvent函数中，我们通过event.mimeData().hasFormat()方法判断拖放的数据类型。如果数据类型是我们自定义的类型，我们将打印出”Custom data dropped!”；如果数据类型是文本类型，我们将打印出”Text data dropped: “和实际的文本数据。

通过这种方式，我们可以根据不同的数据类型来执行不同的操作，以满足我们的拖放需求。