pytex/PyTeX/format/generic_text.py
2022-02-07 00:42:03 +01:00

93 lines
2.8 KiB
Python

from pathlib import Path
from typing import Union, List, Optional
from ..logger import logger
class GenericText:
def __init__(self, content: Optional[Union[List[str], Path]] = None):
if isinstance(content, list):
self._content: Optional[List[str]] = content
self._path = None
self._initialized = True
elif isinstance(content, Path):
self._content: Optional[List[str]] = None
self._path = content
self._initialized = True
else:
self._content = None
self._path = None
self._initialized = False
@property
def text(self) -> List[str]:
if self._initialized:
if self._content is None:
try:
with open(self._path, 'r') as file:
self._content = file.readlines()
except FileNotFoundError:
raise NotImplementedError
except:
raise NotImplementedError
return self._content
else:
return []
@text.setter
def text(self, content: Union[List[str], Path, None]) -> None:
if isinstance(content, List):
self._content = content
self._path = None
self._initialized = True
elif isinstance(content, Path):
self._content = None
self._path = content
self._initialized = True
else:
self._content = None
self._path = None
self._initialized = False
@property
def path(self) -> Optional[Path]:
return self._path
@property
def pathname(self) -> Optional[str]:
return str(self._path) if self._path else None
def format(self, **kwargs) -> str:
lines = []
for line in self.text:
try:
line = '% ' + line.format(**kwargs).rjust(77) + '%'
if len(line) > 80:
logger.warning(
'Line too long') # TODO
lines.append(line)
except ValueError:
raise NotImplementedError
return '\n'.join(lines)
def has_value(self) -> bool:
return self._initialized
@property
def real_text(self) -> Optional[List[str]]:
if self.has_value():
return self._content
else:
return None
def __add__(self, other):
if not self.has_value():
return other
if not other.has_value():
return self
if isinstance(other, GenericText):
return GenericText(self.text + other.text)
else:
return GenericText(self.text + other)
def __iadd__(self, other):
self.text = self + other