119 lines
3.7 KiB
Python
119 lines
3.7 KiB
Python
from __future__ import annotations
|
|
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, str]] = None):
|
|
# TODO: what if paths are not absolute? Have a root available?
|
|
if isinstance(content, list):
|
|
self._content: Optional[List[str]] = content
|
|
self._path = None
|
|
self._initialized = True
|
|
elif isinstance(content, Path) or isinstance(content, str):
|
|
self._content = None
|
|
self._path = 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:
|
|
if self._path is None:
|
|
raise NotImplementedError # Programming error
|
|
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:
|
|
padding = True
|
|
if 'padding' in kwargs.keys():
|
|
padding = kwargs['padding']
|
|
kwargs.pop('padding', None)
|
|
lines = []
|
|
for line in self.text:
|
|
try:
|
|
line = '% ' + line.rstrip().format(**kwargs)
|
|
if padding:
|
|
line = line.ljust(79) + '%'
|
|
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: Union[None, GenericText, List[str]]) -> GenericText:
|
|
if not self.has_value():
|
|
return other
|
|
if isinstance(other, GenericText):
|
|
if not other.has_value():
|
|
return self
|
|
return GenericText(self.text + other.text)
|
|
else:
|
|
return GenericText(self.text + other)
|
|
|
|
def __iadd__(self, other: Union[None, GenericText, List[str]]) -> GenericText:
|
|
if not self.has_value():
|
|
if isinstance(other, GenericText):
|
|
self.text = other.text
|
|
else:
|
|
self.text = other
|
|
elif isinstance(other, GenericText):
|
|
if other.has_value():
|
|
self.text += other.text
|
|
else:
|
|
self.text += other
|
|
return self
|
|
|
|
def __radd__(self, other):
|
|
if other is None:
|
|
return self
|
|
return other + self
|