Qt 在 QTreeWidget 之间拖放

Cat*_*aro 3 python pyqt qtreewidget pyqt4 pyqt5

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

我想为这两种行为重写拖放功能:

  • 从 A 到 A 移动所有孩子的物品。
  • 从 B 到 A 与所有孩子一起复制项目。

我认为在 dropEvent 函数中我有 event.source() 作为获取源。

但是如何使用标志更改功能以进行移动或动态复制?

有例子吗?

谢谢

eyl*_*esc 5

为了执行这个任务,我们必须区分动作是来自他自己还是另一个小部件,因为它被dropEvent方法覆盖,这将返回一个类型的事件,该事件QDropEvent具有source()返回执行该动作的对象的方法:

def dropEvent(self, event):
    if event.source() == self:
        # move
    else:
        # copy
Run Code Online (Sandbox Code Playgroud)
  • 从 A 到 A 移动所有孩子的物品。

每个QTreeWidgetItem都与其父级相关联,因此如果它移动到父级,子级也将移动。

if event.source() == self:
    event.setDropAction(Qt.MoveAction)
    QTreeWidget.dropEvent(self, event)
Run Code Online (Sandbox Code Playgroud)
  • 从 B 到 A 与所有孩子一起复制项目。

这是最复杂的任务,因为当QTreeWidget使用拖放时,它使用x-qabstractitemmodeldatalist存储与项目的 QModelIndex 关联的行和列的信息的 mimetype以及带有角色及其值的字典:

row_1 | column_1 | number of valid roles | role_1_1 | value_1_1 | ... | role_n_1 | value_n_1 | row_2 | column_2 | ...
Run Code Online (Sandbox Code Playgroud)

所以知道行和列是不可能获得项目的,因此也无法获得你的孩子,因为你必须有父母。所以你应该寻找另一种策略,解决方案是创建一个新的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
Run Code Online (Sandbox Code Playgroud)

在以下链接中,您将看到一个示例