drop event update

Author: jchanvfx

NodeGraphQt/base/graph.py

@@ -34,7 +34,7 @@ class NodeGraph(QtCore.QObject):
     #: signal for when a node has been connected.
     port_connected = QtCore.Signal(Port, Port)
     #: signal for when drop data has been added to the graph.
-    data_dropped = QtCore.Signal(str, tuple)
+    data_dropped = QtCore.Signal(QtCore.QMimeData, QtCore.QPoint)
 
     def __init__(self, parent=None, default_actions=True):
         super(NodeGraph, self).__init__(parent)
@@ -91,8 +91,8 @@ def _on_node_data_dropped(self, data, pos):
         (emits the node type and the x,y position where the data was dropped)
 
         Args:
-            data (str): text data.
-            pos (tuple): x, y scene position relative to the cursor.
+            data (QtCore.QMimeData): text data.
+            pos (QtCore.QPoint): x, y scene position relative to the cursor.
         """
         self.data_dropped.emit(data, pos)
 

NodeGraphQt/widgets/viewer.py

@@ -28,7 +28,7 @@ class NodeViewer(QtWidgets.QGraphicsView):
 
     # pass through signals
     node_selected = QtCore.Signal(str)
-    data_dropped = QtCore.Signal(str, tuple)
+    data_dropped = QtCore.Signal(QtCore.QMimeData, QtCore.QPoint)
 
     def __init__(self, parent=None):
         super(NodeViewer, self).__init__(parent)
@@ -233,15 +233,9 @@ def wheelEvent(self, event):
         self._set_viewer_zoom(adjust)
 
     def dropEvent(self, event):
-        if event.mimeData().hasFormat('text/plain'):
-            pos = self.mapToScene(event.pos())
-            drop_pos = pos.x(), pos.y()
-            drop_str = event.mimeData().text()
-            event.setDropAction(QtCore.Qt.MoveAction)
-            event.accept()
-            self.data_dropped.emit(drop_str, drop_pos)
-        else:
-            event.ignore()
+        pos = self.mapToScene(event.pos())
+        event.setDropAction(QtCore.Qt.MoveAction)
+        self.data_dropped.emit(event.mimeData(), pos)
 
     def dragEnterEvent(self, event):
         if event.mimeData().hasFormat('text/plain'):