Python QTreeWidget之间的Qt拖放

在我的QT应用程序中，我有两个QTreeWidget A和B

我想为这两种行为重写拖放函数：

• 从A到A将项目与所有子项一起移动
• 从B到A复制所有子项

我认为在dropEvent函数中，我有用于获取源的event.source（）

但如何改变功能和标志的移动或复制

有一个例子吗

---

谢谢

为了执行此任务，我们必须区分该操作是来自他自己还是其他小部件，因为它被

dropEvent

方法覆盖，这将返回类型为

qdropent

的事件，该事件具有返回执行该操作的对象的方法

source（）

：

def dropEvent(self, event):
    if event.source() == self:
        # move
    else:
        # copy

• 从A到A将项目与所有子项一起移动

每个

QTreeWidgetItem

都与其父项关联，因此如果它移动到父项，子项也将移动

if event.source() == self:
    event.setDropAction(Qt.MoveAction)
    QTreeWidget.dropEvent(self, event)

• 从B到A复制所有子项

这是最复杂的任务，因为当

QTreeWidget

使用拖放时，它使用mimetype

x-qabstractitemmodeldatalist

存储与项的QModelIndex关联的行和列的信息，以及一个包含角色及其值的字典：

row_1 | column_1 | number of valid roles | role_1_1 | value_1_1 | ... | role_n_1 | value_n_1 | row_2 | column_2 | ...

因此，了解行和列不可能获得项目，因此也不可能获得您的子项，因为您必须拥有父项。因此，您应该寻找另一种策略，解决方案是创建一个新的mimetype，在其中保存必要的信息以再次获取该项，唯一的方法是保存该项的行，然后保存父项的行，然后像这样保存父项的父项的行，直到到达某个

topLevelItem

由于您已经拥有该项目，因此可以访问您的孩子及其包含的信息。全面实施情况如下所示：

class TreeWidget(QTreeWidget):
    customMimeType = "application/x-customTreeWidgetdata"

    def mimeTypes(self):
        mimetypes = QTreeWidget.mimeTypes(self)
        mimetypes.append(TreeWidget.customMimeType)
        return mimetypes

    def startDrag(self, supportedActions):
        drag = QDrag(self)
        mimedata = self.model().mimeData(self.selectedIndexes())

        encoded = QByteArray()
        stream = QDataStream(encoded, QIODevice.WriteOnly)
        self.encodeData(self.selectedItems(), stream)
        mimedata.setData(TreeWidget.customMimeType, encoded)

        drag.setMimeData(mimedata)
        drag.exec_(supportedActions)

    def dropEvent(self, event):
        if event.source() == self:
            event.setDropAction(Qt.MoveAction)
            QTreeWidget.dropEvent(self, event)
        elif isinstance(event.source(), QTreeWidget):
            if event.mimeData().hasFormat(TreeWidget.customMimeType):
                encoded = event.mimeData().data(TreeWidget.customMimeType)
                parent = self.itemAt(event.pos())
                items = self.decodeData(encoded, event.source())
                for it in items:
                    item = QTreeWidgetItem(parent)
                    self.fillItem(it, item)
                    self.fillItems(it, item)
                event.acceptProposedAction()

    def fillItem(self, inItem, outItem):
        for col in range(inItem.columnCount()):
            for key in range(Qt.UserRole):
                role = Qt.ItemDataRole(key)
                outItem.setData(col, role, inItem.data(col, role))

    def fillItems(self, itFrom, itTo):
        for ix in range(itFrom.childCount()):
            it = QTreeWidgetItem(itTo)
            ch = itFrom.child(ix)
            self.fillItem(ch, it)
            self.fillItems(ch, it)

    def encodeData(self, items, stream):
        stream.writeInt32(len(items))
        for item in items:
            p = item
            rows = []
            while p is not None:
                rows.append(self.indexFromItem(p).row())
                p = p.parent()
            stream.writeInt32(len(rows))
            for row in reversed(rows):
                stream.writeInt32(row)
        return stream

    def decodeData(self, encoded, tree):
        items = []
        rows = []
        stream = QDataStream(encoded, QIODevice.ReadOnly)
        while not stream.atEnd():
            nItems = stream.readInt32()
            for i in range(nItems):
                path = stream.readInt32()
                row = []
                for j in range(path):
                    row.append(stream.readInt32())
                rows.append(row)

        for row in rows:
            it = tree.topLevelItem(row[0])
            for ix in row[1:]:
                it = it.child(ix)
            items.append(it)
        return items

在下面，您将看到一个示例