pytex/PyTeX/build/build/relative_path.py

from ctypes import Union
from pathlib import Path

from PyTeX.build.build.enums import PyTeXRootDirType


class RelativePath:
    """
    Represents a path that knows of its corresponding root directory
    """

    def __init__(
            self,
            root_dir: Path,
            *args,
            **kwargs
    ):
        pass
        self._path = Path(*args, **kwargs)
        self._root_dir = root_dir

    def __getattr__(self, attr):
        ret = getattr(self._path, attr)
        if isinstance(ret, Path):
            return RelativePath(self._pytex_root_dir_type, ret)
        else:
            return ret

    def __truediv__(self, other):
        if isinstance(other, RelativePath):
            path = self._path.__truediv__(other.path)
        else:
            path = self._path.__truediv__(other)
        return RelativePath(self._root_dir, path)

    def __rtruediv__(self, other):
        if isinstance(other, RelativePath):
            path = self._path.__rtruediv__(other.path)
        else:
            path = self._path.__rtruediv__(other)
        return RelativePath(self._root_dir, path)

    def __str__(self):
        return self._path.__str__()

    @property
    def path(self) -> Path:
        return self._path

    @property
    def root_dir_type(self) -> PyTeXRootDirType:
        return self._pytex_root_dir_type

    @property
    def root_dir(self) -> Path:
        return self._root_dir

    @property
    def relative_path(self) -> Path:
        try:
            return self.relative_to(self._root_dir)
        except ValueError as e:
            raise NotImplementedError

    def with_root(self, root: Path):
        return RelativePath(
            root,
            root / self.relative_path
        )