A simple unordered tree implementation in Python for the weekly coding prompt on the Beginner.Codes Discord server. Join now https://beginner.codes/discord

unordered_tree.py

from typing import Generic, TypeVar


T = TypeVar("T")


class Node(Generic[T]):
    def __init__(self, value: T, parent: "Node[T] | None" = None):
        self._children = tuple()
        self._parent = parent
        self._value = value

    @property
    def children(self) -> "tuple[Node[T]]":
        return self._children

    @property
    def is_root(self):
        return self._parent is None

    @property
    def parent(self) -> "Node[T]":
        return self._parent or self

    @property
    def value(self) -> T:
        return self._value

    def add_child(self, child: "Node[T]"):
        self._children = (*self.children, child)