Refactored Command to add owner

src/myfasthtml/controls/TreeView.py

@@ -66,74 +66,67 @@ class Commands(BaseCommands):
   
   def toggle_node(self, node_id: str):
     """Create command to expand/collapse a node."""
-    return Command(
-      "ToggleNode",
-      f"Toggle node {node_id}",
-      self._owner._toggle_node,
-      node_id
-    ).htmx(target=f"#{self._owner.get_id()}")
+    return Command("ToggleNode",
+                   f"Toggle node {node_id}",
+                   self._owner,
+                   self._owner._toggle_node,
+                   node_id).htmx(target=f"#{self._owner.get_id()}")
   
   def add_child(self, parent_id: str):
     """Create command to add a child node."""
-    return Command(
-      "AddChild",
-      f"Add child to {parent_id}",
-      self._owner._add_child,
-      parent_id
-    ).htmx(target=f"#{self._owner.get_id()}")
+    return Command("AddChild",
+                   f"Add child to {parent_id}",
+                   self._owner,
+                   self._owner._add_child,
+                   parent_id).htmx(target=f"#{self._owner.get_id()}")
   
   def add_sibling(self, node_id: str):
     """Create command to add a sibling node."""
-    return Command(
-      "AddSibling",
-      f"Add sibling to {node_id}",
-      self._owner._add_sibling,
-      node_id
-    ).htmx(target=f"#{self._owner.get_id()}")
+    return Command("AddSibling",
+                   f"Add sibling to {node_id}",
+                   self._owner,
+                   self._owner._add_sibling,
+                   node_id
+                   ).htmx(target=f"#{self._owner.get_id()}")
   
   def start_rename(self, node_id: str):
     """Create command to start renaming a node."""
-    return Command(
-      "StartRename",
-      f"Start renaming {node_id}",
-      self._owner._start_rename,
-      node_id
-    ).htmx(target=f"#{self._owner.get_id()}")
+    return Command("StartRename",
+                   f"Start renaming {node_id}",
+                   self._owner,
+                   self._owner._start_rename,
+                   node_id).htmx(target=f"#{self._owner.get_id()}")
   
   def save_rename(self, node_id: str):
     """Create command to save renamed node."""
-    return Command(
-      "SaveRename",
-      f"Save rename for {node_id}",
-      self._owner._save_rename,
-      node_id
-    ).htmx(target=f"#{self._owner.get_id()}")
+    return Command("SaveRename",
+                   f"Save rename for {node_id}",
+                   self._owner,
+                   self._owner._save_rename,
+                   node_id).htmx(target=f"#{self._owner.get_id()}")
   
   def cancel_rename(self):
     """Create command to cancel renaming."""
-    return Command(
-      "CancelRename",
-      "Cancel rename",
-      self._owner._cancel_rename
-    ).htmx(target=f"#{self._owner.get_id()}")
+    return Command("CancelRename",
+                   "Cancel rename",
+                   self._owner,
+                   self._owner._cancel_rename).htmx(target=f"#{self._owner.get_id()}")
   
   def delete_node(self, node_id: str):
     """Create command to delete a node."""
-    return Command(
-      "DeleteNode",
-      f"Delete node {node_id}",
-      self._owner._delete_node,
-      node_id
-    ).htmx(target=f"#{self._owner.get_id()}")
+    return Command("DeleteNode",
+                   f"Delete node {node_id}",
+                   self._owner,
+                   self._owner._delete_node,
+                   node_id).htmx(target=f"#{self._owner.get_id()}")
   
   def select_node(self, node_id: str):
     """Create command to select a node."""
-    return Command(
-      "SelectNode",
-      f"Select node {node_id}",
-      self._owner._select_node,
-      node_id
-    ).htmx(target=f"#{self._owner.get_id()}")
+    return Command("SelectNode",
+                   f"Select node {node_id}",
+                   self._owner,
+                   self._owner._select_node,
+                   node_id).htmx(target=f"#{self._owner.get_id()}")
 
 
 class TreeView(MultipleInstance):
@@ -187,7 +180,7 @@ class TreeView(MultipleInstance):
     self._state.items[node.id] = node
     if parent_id is None and node.parent is not None:
       parent_id = node.parent
-      
+    
     node.parent = parent_id
     
     if parent_id and parent_id in self._state.items: