Class CodeChangeCommand

src/commands/node/property_changes.py:82–126

Command for tracking code changes in nodes.

class CodeChangeCommand(CommandBase):
    """Command for tracking code changes in nodes."""
    
    def __init__(self, node_graph, node, old_code: str, new_code: str):
        """
        Initialize code change command.
        
        Args:
            node_graph: The NodeGraph instance
            node: Node whose code is changing
            old_code: Original code
            new_code: New code
        """
        super().__init__(f"Change code in '{node.title}'")
        self.node_graph = node_graph
        self.node = node
        self.old_code = old_code
        self.new_code = new_code
    
    def execute(self) -> bool:
        """Apply the code change."""
        try:
            self.node.set_code(self.new_code)
            self._mark_executed()
            return True
        except Exception as e:
            print(f"Failed to change code: {e}")
            return False
    
    def undo(self) -> bool:
        """Revert the code change."""
        try:
            self.node.set_code(self.old_code)
            self._mark_undone()
            return True
        except Exception as e:
            print(f"Failed to undo code change: {e}")
            return False
    
    def get_memory_usage(self) -> int:
        """Estimate memory usage for code changes."""
        base_size = 512
        old_code_size = len(self.old_code) * 2
        new_code_size = len(self.new_code) * 2
        return base_size + old_code_size + new_code_size