extracted pieces of selection logic

Author: alexej

editor.py

@@ -112,7 +112,8 @@ def handle_mouse_down(self, event):
                 if node.contains_point(world_x, world_y) and clicked_node is None:
                     node.dragging = True
                     node.drag_offset = (world_x - node.x, world_y - node.y)
-                    node.selected = True
+                    # Delegate selection logic:
+                    self.selection.select_node(node, self.nodes)
                     clicked_node = node
                     # --- Selected node should be always on top ---
                     self.nodes.remove(node)
@@ -121,8 +122,9 @@ def handle_mouse_down(self, event):
                     if not self._node_drag_in_progress:
                         self.undo_stack.push(copy.deepcopy(self.nx_graph))
                         self._node_drag_in_progress = True
-                else:
-                    node.selected = False
+            # Deselect all if no node was clicked
+            if clicked_node is None:
+                self.selection.clear_selection(self.nodes)
 
         elif event.button == pygame.BUTTON_MIDDLE:
             # Check if a node was clicked

selection.py

@@ -3,6 +3,7 @@ def __init__(self):
         self.rect_start = None
         self.rect_end = None
         self.nodes = []
+        self.selected_nodes = []
 
     def begin(self, start_pos):
         self.rect_start = start_pos
@@ -36,3 +37,15 @@ def clear(self):
 
     def is_active(self):
         return self.rect_start is not None and self.rect_end is not None
+
+    def select_node(self, node, all_nodes):
+        # Deselect all others, select only this node
+        for n in all_nodes:
+            n.selected = False
+        node.selected = True
+        self.selected_nodes = [node]
+
+    def clear_selection(self, all_nodes):
+        for n in all_nodes:
+            n.selected = False
+        self.selected_nodes = []