manager.py

from nodes.node import Node
from typing import List

class NodeManager:
    nodes: List[Node]

    def __init__(self, nodes: List[Node]):
        if not isinstance(nodes, list):
            raise ValueError("nodes must be a List of Nodes")
        
        self.nodes = nodes.copy()
    
    def __len__(self):
        return len(self.nodes)
    
    def __getitem__(self, index):
        return self.nodes[index]

    def remove(self, node: Node):
        if node not in self.nodes:
            raise ValueError("Node is not in the list of nodes")
        
        self.nodes.remove(node)
    
    def remove_cascade(self, node: Node):
        if node not in self.nodes:
            raise ValueError("Node is not in the list of nodes")
        
        children = [n for n in self.nodes if n.parent == node.id]
        
        for child in children:
            self.remove_cascade(child) 

        self.nodes.remove(node)