make sure we remove deleted nodes from new_nodes and simplify code

zplorer52 · Feb 1, 2019 · 10f54e7 · 10f54e7
1 parent 3f2c1fa
commit 10f54e7
Showing 1 changed file with 4 additions and 5 deletions.
diff --git a/uamodeler/model_manager.py b/uamodeler/model_manager.py
@@ -50,11 +50,10 @@ def __init__(self, modeler):
 
     def delete_node(self, node):
         if node:
-            nodes = get_node_children(node)
-            for n in nodes:
-                n.delete(delete_references=True, recursive=True)
-                if n in self.new_nodes:
-                    self.new_nodes.remove(n)
+            deleted_nodes = node.delete(delete_references=True, recursive=True)
+            for dn in deleted_nodes:
+                if dn in self.new_nodes:
+                    self.new_nodes.remove(dn)
             self.modeler.tree_ui.remove_current_item()
 
     def paste_node(self, node):