fixing custom list widget items, datatypes for module/class to Any

Author: bhowiebkr

node_editor/gui/node_list.py

@@ -1,6 +1,7 @@
 from __future__ import annotations
 
 from types import ModuleType
+from typing import Any
 from typing import Dict
 from typing import Optional
 from typing import Union
@@ -11,8 +12,8 @@
 
 
 class CustomQListWidgetItem(QtWidgets.QListWidgetItem):  # type: ignore
-    module: str
-    class_name: str
+    module: Any
+    class_name: Any
 
 
 class CustomQMimeData(QtCore.QMimeData):  # type: ignore
@@ -21,8 +22,8 @@ class CustomQMimeData(QtCore.QMimeData):  # type: ignore
 
 class ImportData:
     def __init__(self, module: str, class_: str):
-        self.module: str = module
-        self.class_: str = class_
+        self.module: Any = module
+        self.class_: Any = class_
 
 
 # class NodeList(QtWidgets.QListWidget):  # type: ignore
@@ -41,14 +42,15 @@ def update_project(self, imports: Dict[str, Dict[str, Union[type, ModuleType]]])
         for name, data in imports.items():
             name = name.replace("_Node", "")
 
-            item = QtWidgets.QListWidgetItem(name)
+            item = CustomQListWidgetItem(name)
 
             item.module = data["module"]
             item.class_name = data["class"]
             self.addItem(item)
 
     def mousePressEvent(self, event: QtGui.QMouseEvent) -> None:
         item = self.itemAt(event.pos())
+
         if isinstance(item, CustomQListWidgetItem) and item.text():
             name = item.text()
 
@@ -64,4 +66,6 @@ def mousePressEvent(self, event: QtGui.QMouseEvent) -> None:
             drag.setPixmap(pixmap)
             drag.exec_()
 
+            print("Inside drag event from node list")
+
             super().mousePressEvent(event)