子类化QListWidgetItem的拖放功能

Question

我在实现自定义的QListWidgetItem的拖放功能时遇到了一些问题。这里有一段示例代码：

from PyQt4 import QtGui, QtCore
import sys, os


class MyListWidgetItem(QtGui.QListWidgetItem):      
    def __init__(self, label, data, parent=None):
        super(QtGui.QListWidgetItem, self).__init__(label, parent=parent)
        self.data = data

    def GetData(self):
        return self.data

class MyListWidget(QtGui.QListWidget):
    def __init__(self, type, parent=None):
        super(MyListWidget, self).__init__(parent)
        self.setDragDropMode(QtGui.QAbstractItemView.DragDrop)
        self.setSelectionMode(QtGui.QAbstractItemView.ExtendedSelection)
        self.setAcceptDrops(True)
        self.viewport().setAcceptDrops(True)
        self.setDropIndicatorShown(True)

    def startDrag(self, supportedActions):
        drag = QtGui.QDrag(self)
        t = [i.GetData() for i in self.selectedItems()]
        mimeData = self.model().mimeData(self.selectedIndexes())
        mimeData.setText(str(t))
        drag.setMimeData(mimeData)
        if drag.start(QtCore.Qt.MoveAction) == QtCore.Qt.MoveAction:
            for item in self.selectedItems():
                self.takeItem(self.row(item))

    def dragEnterEvent(self, event):
        if event.mimeData().hasUrls():
            event.ignore()
        else:
            event.accept()

    def dragMoveEvent(self, event):
        if event.mimeData().hasUrls():
            event.ignore()
        else:
            event.accept()

    def dropEvent(self, event):
        if event.mimeData().hasUrls():
            event.ignore()
        if isinstance(event.source(), MyListWidget):
            event.setDropAction(QtCore.Qt.MoveAction)
            super(MyListWidget, self).dropEvent(event)
        else:
            event.ignore()

    def dropMimeData(self, index, mimedata, action):
        super(MyListWidget, self).dropMimeData(index, mimedata, action)
        return True

class Test(QtGui.QMainWindow):
    def __init__(self):
        super(QtGui.QMainWindow,self).__init__()
        myQWidget = QtGui.QWidget()
        myBoxLayout = QtGui.QVBoxLayout()
        myQWidget.setLayout(myBoxLayout)
        self.setCentralWidget(myQWidget)

        self.listWidgetA = MyListWidget(self)
        self.listWidgetB = MyListWidget(self)

        for i in range(5):
            listItemAInstance = MyListWidgetItem(str(i), i, parent=self.listWidgetA)

        myBoxLayout.addWidget(self.listWidgetA)      
        myBoxLayout.addWidget(self.listWidgetB)   

if __name__ == '__main__':
    app = QtGui.QApplication(sys.argv)
    dialog_1 = Test()
    dialog_1.show()
    dialog_1.resize(480,320)
    sys.exit(app.exec_())

我的自定义类MyListWidgetItem有一个'数据'字段，这个字段在我的实际程序中存储与该项相关的一些信息。如果你运行这段代码，尝试把上面的列表中的项目拖到下面的列表，一切都很顺利。但是如果你再试着把它们拖回到上面的列表，就会出现这个错误：

Traceback (most recent call last):
  File "C:\Users\Massi\Desktop\t.py", line 23, in startDrag
    t = [i.GetData() for i in self.selectedItems()]
AttributeError: 'QListWidgetItem' object has no attribute 'GetData'

看起来很明显，默认的拖放行为忽略了列表项已经被子类化的事实，所以我想知道在这种情况下最好的解决办法是什么。非常感谢任何帮助！

Answer 1

如果你添加一些调试信息，比如下面这样，你会发现“i”已经是MyListWidgetItem类型了，所以这应该不是问题所在。

  def startDrag(self, supportedActions):
        drag = QtGui.QDrag(self)
        # add the following 2 line debug information
        for item in self.selectedItems():
            print item.__class__.__name__
        t = [i.GetData() for i in self.selectedItems() if hasattr(i, "GetData")]
        mimeData = self.model().mimeData(self.selectedIndexes())
        mimeData.setText(str(t))
        drag.setMimeData(mimeData)
        if drag.start(QtCore.Qt.MoveAction) == QtCore.Qt.MoveAction:
            for item in self.selectedItems():
                self.takeItem(self.row(item))

输出是

MyListWidgetItem

问题在于，你不能把这个项目的类型转换成MyListWidgetItem，因为Python不支持类型转换。所以你需要用反射机制来调用GetData()方法。我刚试过这个代码，运行得很好！

Answer 2

我会使用一个叫做 QListView 的组件，配合一个叫做 QStandardItemModel 的模型。然后我会用 QStandardItemModel.setItemPrototype(MyStandardItem()) 这个方法来设置一个模板项。接着，我会把 MyStandardItem 定义成类似下面这样的东西：

class MyStandardItem(QtGui.QStandardItem):
    def __init__(self, *__args):
        super().__init__(*__args)

    def clone(self):
        return MyStandardItem()