如何从QListWidget拖放到ComboBox

时间:2014-03-31 23:32:23

标签: python drag-and-drop pyqt qlistwidget qcombobox

目标是能够将ListWidget项拖放到comboBox上。丢弃的项目应添加到组合框中。理想情况下,我们希望避免阅读listWidget的.currentItem()或.selectedItems()等等的任何技巧......想法?

enter image description here

from PyQt4 import QtGui, QtCore
import sys, os

class MyClass(object):
    def __init__(self):
        super(MyClass, self).__init__()
        self.name=None     
    def setName(self, arg):
        self.name=arg
    def getName(self):
        return self.name       

class DropableComboBox(QtGui.QComboBox):
    def __init__(self):
        self.model_mime_type = 'application/x-qabstractitemmodeldatalist'
        super(DropableComboBox, self).__init__()
        self.setAcceptDrops(True)

    def dragEnterEvent(self, event):
        if event.mimeData().hasFormat(self.model_mime_type) or event.mimeData().hasFormat('text/plain'):
            event.accept()
        else:
            event.ignore()

    def dropEvent(self, event):
        if event.mimeData().hasUrls():
            event.setDropAction(QtCore.Qt.CopyAction)
            event.accept()
            links = []
            for url in event.mimeData().urls():
                links.append(str(url.toLocalFile()))
            self.emit(QtCore.SIGNAL("dropped"), links)
        else:
            super(DropableComboBox, self).dropEvent(event)
            self.emit(QtCore.SIGNAL("dropped"))

class Dialog_01(QtGui.QMainWindow):
    def __init__(self):
        super(QtGui.QMainWindow,self).__init__()

        myQWidget = QtGui.QWidget()
        myBoxLayout = QtGui.QVBoxLayout()
        myQWidget.setLayout(myBoxLayout)
        self.setCentralWidget(myQWidget)

        self.listWidget = QtGui.QListWidget()
        self.listWidget.setDragDropMode(QtGui.QAbstractItemView.DragDrop)
        self.listWidget.setSelectionMode(QtGui.QAbstractItemView.ExtendedSelection)
        self.listWidget.currentItemChanged.connect(self.item_clicked)

        for i in range(3):
            my_item=QtGui.QListWidgetItem()
            name='ListWidget Item '+str(i)
            my_item.setText(name)
            self.listWidget.addItem(my_item)

            myObject=MyClass()
            myObject.setName(name) 

            my_item.setData(QtCore.Qt.UserRole, myObject)

        myBoxLayout.addWidget(self.listWidget)

        self.ComboBox = DropableComboBox()
        for i in range(3):
            self.ComboBox.addItem("Combobox Item " + str(i))

        self.ComboBox.currentIndexChanged.connect(self.combobox_selected)
        self.connect(self.ComboBox, QtCore.SIGNAL("dropped"), self.droppedOnCombobox)

        myBoxLayout.addWidget(self.ComboBox)

    def item_clicked(self, arg=None):
        print arg.data(QtCore.Qt.UserRole).toPyObject().getName()

    def combobox_selected(self, index):
        myObject=self.ComboBox.itemData(index).toPyObject()
        if hasattr(myObject, 'getName'): print myObject.getName()

    def droppedOnCombobox(self):
        print "Drop!"


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

1 个答案:

答案 0 :(得分:3)

"适当"这样做的方法是使用QDataStream解压缩mimedata。但是,这似乎需要使用QMap,这在PyQt中是不可用的。相反,它可以通过让代理模型为我们的肮脏工作获得一点点hacky(或者应该是"棘手的"?)方式来完成:

class DropableComboBox(QtGui.QComboBox):
    def __init__(self):
        super(DropableComboBox, self).__init__()
        self.model_mime_type = 'application/x-qabstractitemmodeldatalist'
        self.setAcceptDrops(True)
        self._proxymodel = QtGui.QStandardItemModel(self)

    def dropEvent(self, event):
        if event.mimeData().hasUrls():
            event.setDropAction(QtCore.Qt.CopyAction)
            event.accept()
            links = []
            for url in event.mimeData().urls():
                links.append(str(url.toLocalFile()))
            self.emit(QtCore.SIGNAL("dropped"), links)
        elif event.mimeData().hasFormat(self.model_mime_type):
            self._proxymodel.setRowCount(0)
            self._proxymodel.dropMimeData(
                event.mimeData(), QtCore.Qt.CopyAction,
                0, 0, QtCore.QModelIndex())
            for index in range(self._proxymodel.rowCount()):
                item = self._proxymodel.item(index, 0)                  
                self.addItem(item.text())
            # no point calling the base-class dropEvent here,
            # because it's a NO-OP in QComboBox
            self.emit(QtCore.SIGNAL("dropped"))

NB:

这将复制列表小部件中的项目,而不是移动它们(您没有要求)。此外,如果要防止添加重复项,请使用setDuplicatesEnabled。如果您想更改项目的添加方式,请使用setInsertPolicy