pytex/PyTeX/build/paths/relative_path.py

from pathlib import Path

from PyTeX.build.enums.enums import PyTeXRootDirType


class RelativePath:
    """
    Represents a path that knows of its corresponding root directory
    """
    def __init__(
            self,
            root_dir: Path,
            *args,
            **kwargs
    ):
        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):
        try:
            return self.relative_to(self._root_dir)
        except ValueError as e:
            raise NotImplementedError