FileSystem.py

from __future__ import annotations
from dataclasses import dataclass, field
from pathlib import Path
from typing import List, Optional
import uuid

@dataclass
class Node:
    name: str
    type: str # folder, archive, file
    path: Path = Path()

    id: Optional[uuid.UUID] = None
    parent: Optional["Node"] = None
    children: List["Node"] = field(default_factory=list)

    def get_path(self, path: Path) -> "Node":  # pyright: ignore[reportReturnType] # Not working
        """Gets the Node at the passed path

        Args:
            path (Path): path to the node

        Returns:
            Node: the node at passed path
        """
        if len(path.parts) == 1:
            return self

        for child in self.children:
            if child.name == path.parts[1]:
                return child.get_path(Path(*path.parts[1:]))

    def remove(self, path: Optional[Path] = None):
        """Removes the node at passed path

        Args:
            path (Path): path to the node

        Raises:
            NotImplementedError
        """
        if path is None:
            node: "Node" = self
        else:
            node: "Node" = self.get_path(path)

        node.parent._remove_child(node) # pyright: ignore[reportOptionalMemberAccess]

    def _remove_child(self, node: "Node"):
        """Removes the child node

        Args:
            node (Node): node to remove
        """
        self.children.remove(node)

    def add(self, child: "Node", path: Optional[Path] = None):
        """Adds the passed Node at the passed path

        Args:
            path (Path): the new node parent directory
            child (Node): the node to be added

        Raises:
            NotImplementedError: _description_
        """
        
        if path is None:
            parent = self
        else:
            parent: "Node" = self.get_path(path)

        if parent.type == "file":
            raise Exception("Cannot add child node to a type `file`")
        
        if child.path == Path():
            child.path = self.path / child.name
        
        child.parent = self

        parent._add_child(child)

        return child

    def _add_child(self, node: "Node"):
        """Adds the passed Node to the children

        Args:
            node (Node): the node to be added
        """
        self.children.append(node)

    def __str__(self) -> str:
        """Returns the path of the node as a string"""
        return f"{self.path}"