pytex/PyTeX/format/generic_text.py

from pathlib import Path
from typing import Union, List, Optional
from ..logger import  logger


class GenericText:
    def __init__(self, content: Union[List[str], Path]):
        if isinstance(content, List):
            self._content: Optional[List[str]] = content
            self._path = None
        else:
            self._content: Optional[List[str]] = None
            self._path = content

    @property
    def text(self) -> List[str]:
        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

    @text.setter
    def text(self, content: Union[List[str], Path]) -> None:
        if isinstance(content, List):
            self._content = content
            self._path = None
        else:
            self._content = None
            self._path = content

    def format(self, **kwargs) -> str:
        lines = []
        for line in self._content:
            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 __add__(self, other):
        if isinstance(other, GenericText):
            return GenericText(self.text + other.text)
        else:
            return GenericText(self.text + other)

    def __iadd__(self, other):
        self.text = self + other