✨ add graph.remove_all()

ehne · Aug 14, 2021 · 91d324f · 91d324f
1 parent ff02d99
commit 91d324f
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/pynode_next/graph.py b/pynode_next/graph.py
@@ -147,6 +147,15 @@ def add_all(self, elements: Iterable[Union[Node, Edge]]):
             elif isinstance(i, Edge):
                 self.add_edge(i)
             pause(20)
+
+    def remove_all(self, elements: Iterable[Union[Node, Edge]]):
+        """Removes all node and edge objects from an iterable. all elements need to be of the type `Node` or `Edge`"""
+        for i in elements:
+            if isinstance(i, Node):
+                self.remove_node(i)
+            elif isinstance(i, Edge):
+                self.remove_edge(i)
+            pause(20)
 
     def has_edge(self, edge):
         """Checks if an edge exists in the graph."""