updated node list widget.

Author: jchanvfx

NodeGraphQt/base/graph.py

@@ -1,4 +1,5 @@
 #!/usr/bin/python
+# -*- coding: utf-8 -*-
 import json
 import os
 import re
@@ -13,6 +14,7 @@
 from NodeGraphQt.base.model import NodeGraphModel
 from NodeGraphQt.base.node import NodeObject
 from NodeGraphQt.base.port import Port
+from NodeGraphQt.constants import DRAG_DROP_ID
 from NodeGraphQt.widgets.node_list import NodeListWidget
 from NodeGraphQt.widgets.properties_bin import PropertiesBinWidget
 from NodeGraphQt.widgets.viewer import NodeViewer
@@ -122,6 +124,18 @@ def _on_node_data_dropped(self, data, pos):
             data (QtCore.QMimeData): mime data.
             pos (QtCore.QPoint): scene position relative to the drop.
         """
+
+        # don't emit signal for internal widget drops.
+        if data.hasFormat('text/plain'):
+            if data.text().startswith('<${}>:'.format(DRAG_DROP_ID)):
+                node_ids = data.text()[len('<${}>:'.format(DRAG_DROP_ID)):]
+                x, y = pos.x(), pos.y()
+                for node_id in node_ids.split(','):
+                    self.create_node(node_id, pos=[x, y])
+                x += 20
+                y += 20
+                return
+
         self.data_dropped.emit(data, pos)
 
     def _on_nodes_moved(self, node_data):

NodeGraphQt/constants.py

@@ -65,6 +65,8 @@
 
 SCENE_AREA = 8000.0
 
+DRAG_DROP_ID = 'n0deGraphQT'
+
 # === PATHS ===
 
 BASE_PATH = os.path.dirname(os.path.abspath(__file__))

NodeGraphQt/widgets/node_list.py

@@ -1,4 +1,11 @@
+#!/usr/bin/python
+# -*- coding: utf-8 -*-
 from NodeGraphQt import QtWidgets, QtCore
+from NodeGraphQt.constants import DRAG_DROP_ID
+
+
+TYPE_NODE = QtWidgets.QTreeWidgetItem.UserType + 1
+TYPE_CATEGORY = QtWidgets.QTreeWidgetItem.UserType + 2
 
 
 class BaseListWidgetItem(QtWidgets.QListWidgetItem):
@@ -11,10 +18,17 @@ class NodeListWidget(QtWidgets.QTreeWidget):
 
     def __init__(self, parent=None):
         super(NodeListWidget, self).__init__(parent)
+        self.setDragDropMode(QtWidgets.QAbstractItemView.DragOnly)
         self.setHeaderHidden(True)
         self._factory = None
         self._custom_labels = {}
 
+    def mimeData(self, items):
+        node_ids = ','.join(i.toolTip(0) for i in items)
+        mime_data = super(NodeListWidget, self).mimeData(items)
+        mime_data.setText('<${}>:{}'.format(DRAG_DROP_ID, node_ids))
+        return mime_data
+
     def _build_tree(self):
         """
         Populate the node tree.
@@ -33,22 +47,27 @@ def _build_tree(self):
                 label = self._custom_labels[category]
             else:
                 label = '- {}'.format(category)
-            item = QtWidgets.QTreeWidgetItem(self, [label])
-            item.setFirstColumnSpanned(True)
-            item.setFlags(QtCore.Qt.ItemIsEnabled)
-            self.addTopLevelItem(item)
-            item.setExpanded(True)
-            category_items[category] = item
+            cat_item = QtWidgets.QTreeWidgetItem(
+                self, [label], type=TYPE_CATEGORY
+            )
+            cat_item.setFirstColumnSpanned(True)
+            cat_item.setFlags(QtCore.Qt.ItemIsEnabled)
+            self.addTopLevelItem(cat_item)
+            cat_item.setExpanded(True)
+            category_items[category] = cat_item
 
         for node_id, node_name in node_types.items():
             category = '.'.join(node_id.split('.')[:-1])
             category_item = category_items[category]
 
-            item = QtWidgets.QTreeWidgetItem(category_item, [node_name])
+            item = QtWidgets.QTreeWidgetItem(
+                category_item, [node_name], type=TYPE_NODE
+            )
             item.setToolTip(0, node_id)
 
             category_item.addChild(item)
 
+
     def set_node_factory(self, factory):
         """
         Set current node factory.

NodeGraphQt/widgets/viewer.py

@@ -1,4 +1,5 @@
 #!/usr/bin/python
+# -*- coding: utf-8 -*-
 import os
 
 from NodeGraphQt import QtGui, QtCore, QtWidgets

requirements.txt

@@ -1,3 +1,3 @@
 PySide2>=5.12
-
+Qt.py>=1.2.0.b2
 python>=3.6