from pathlib import Path, PurePath, PurePosixPath, PureWindowsPath from PyTeX.build.build import PyTeXBuilder from PyTeX.build.enums.enums import PyTeXRootDirType import os class PyTeXPurePath: pass class PyTeXPurePath(PurePath): def __new__(cls, pytex_root_dir_type: PyTeXRootDirType, *args): if cls is PyTeXPurePath: cls = PyTeXPureWindowsPath if os.name == 'nt' else PyTeXPurePosixPath self = cls._from_parts(args) self.__init(pytex_root_dir_type) return self def __init(self, pytex_root_dir_type: PyTeXRootDirType): self._pytex_root_dir_type: PyTeXRootDirType = pytex_root_dir_type self._root_dir: Path = { PyTeXRootDirType.BUILD: PyTeXBuilder.build_root(), PyTeXRootDirType.PYTEX_SOURCE: PyTeXBuilder.source_root(), PyTeXRootDirType.DOC: PyTeXBuilder.doc_root(), PyTeXRootDirType.TEX_SOURCE: PyTeXBuilder.tex_root() }[self._pytex_root_dir_type] try: if self._pytex_root_dir_type == PyTeXRootDirType.PYTEX_SOURCE: self._relative_path = self.relative_to(self._root_dir) else: self._relative_path = self.relative_to( self._root_dir / PyTeXBuilder.wrapper_dir() ) except ValueError as e: raise NotImplementedError def root_dir_type(self) -> PyTeXRootDirType: return self._pytex_root_dir_type @property def relative_path(self) -> PyTeXPurePath: return self._relative_path def __truediv__(self, other): if self._pytex_root_dir_type == PyTeXRootDirType.PYTEX_SOURCE: return super().__truediv__(other) else: return super().__truediv__(PyTeXBuilder.wrapper_dir()).__truediv__(other) class PyTeXPurePosixPath(PurePosixPath, PyTeXPurePath): pass class PyTeXPureWindowsPath(PureWindowsPath, PyTeXPurePath): pass class PyTeXPath(Path, PyTeXPurePath): def __new__(cls, pytex_root_dir_type: PyTeXRootDirType, *args, **kwargs): if cls is PyTeXPath: cls = PyTeXWindowsPath if os.name == 'nt' else PyTeXPosixPath return super().__new__(cls, *args, **kwargs) class PyTeXPosixPath(PyTeXPath, PyTeXPurePosixPath): pass class PyTeXWindowsPath(PyTeXPath, PyTeXPureWindowsPath): pass