59 lines
1.9 KiB
Python
59 lines
1.9 KiB
Python
from pathlib import Path
|
|
|
|
from ..config import GlobalPyTeXConfig
|
|
from PyTeX.build.enums.enums import PyTeXRootDirType
|
|
|
|
|
|
class RelativePath:
|
|
"""
|
|
Represents a path that knows of its corresponding root directory
|
|
"""
|
|
def __init__(self, pytex_root_dir_type: PyTeXRootDirType, *args, **kwargs):
|
|
self._path = Path(*args, **kwargs)
|
|
self._pytex_root_dir_type: PyTeXRootDirType = pytex_root_dir_type
|
|
self._root_dir: Path = {
|
|
PyTeXRootDirType.BUILD: GlobalPyTeXConfig.build_root(),
|
|
PyTeXRootDirType.PYTEX_SOURCE: GlobalPyTeXConfig.source_root(),
|
|
PyTeXRootDirType.DOC: GlobalPyTeXConfig.doc_root(),
|
|
PyTeXRootDirType.TEX_SOURCE: GlobalPyTeXConfig.tex_root()
|
|
}[self._pytex_root_dir_type]
|
|
|
|
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._pytex_root_dir_type, path)
|
|
|
|
def __rtruediv__(self, other):
|
|
if isinstance(other, RelativePath):
|
|
path = self._path.__rtruediv__(other.path)
|
|
else:
|
|
path = self._path.__rtruediv__(other)
|
|
return RelativePath(self._pytex_root_dir_type, path)
|
|
|
|
@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
|