helper method to get items by type (Node, Connection, etc)

Author: bhowiebkr

node_editor/gui/view.py

@@ -1,5 +1,7 @@
 from __future__ import annotations
 
+from typing import Any
+from typing import List
 from typing import Optional
 
 from PySide6 import QtCore
@@ -158,25 +160,12 @@ def contextMenuEvent(self, event: QtGui.QContextMenuEvent) -> None:
         This method is called when a context menu event is triggered in the view. It finds the item at the
         event position and shows a context menu if the item is a Node.
         """
-        # cursor = QtGui.QCursor()
-        # origin = self.mapFromGlobal(cursor.pos())
-        # pos = self.mapFromGlobal(cursor.pos())
         item = self.itemAt(event.pos())
 
         if item:
             if isinstance(item, Node):
                 print("Found Node", item)
 
-                # menu = QtWidgets.QMenu(self)
-
-                # hello_action = QtWidgets.QAction("Hello", self)
-
-                # menu.addAction(hello_action)
-                # action = menu.exec_(self.mapToGlobal(pos))
-
-                # if action == hello_action:
-                #    print("Hello")
-
     def dragEnterEvent(self, e: QtGui.QDragEnterEvent) -> None:
         """
         This method is called when a drag and drop event enters the view. It checks if the mime data format
@@ -233,3 +222,11 @@ def mouseMoveEvent(self, event: QtGui.QMouseEvent) -> None:
             self._pan_start_y = event.y()
 
         super().mouseMoveEvent(event)
+
+    def get_items_by_type(self, item_class: type) -> List[Any]:
+        items = []
+        for item in self.scene().items():
+            print(f"current item: {item}, class: {item_class}")
+            if isinstance(item, item_class):
+                items.append(item)
+        return items