decorator_classes.py

Go to the documentation of this file.

 
 
from abc import ABC, abstractmethod
 
 
 
 
 
class IRenderElement(ABC):
    
    @abstractmethod
    def Render(self) -> str:
        pass
 
 
 
 
 
 
 
class Decorator(IRenderElement):
    
    
    def __init__(self, element : IRenderElement) -> None:
        self._wrappedElement = element
        if not self._wrappedElement:
            raise ValueError("The element being decorated cannot be null.")
 
    
 
 
    
    def Render(self) -> str:
        return self._wrappedElement.Render()
 
 
 
 
 
 
class WhiteBackgroundDecorator(Decorator):
    
    def __init__(self, element: IRenderElement) -> None:
        super().__init__(element)
 
 
    # IRenderElement Members
 
    
    def Render(self) -> str:
        return "\x1b[47m{0}\x1b[49m".format(super().Render())
 
 
 
 
 
 
class UnderlineDecorator(Decorator):
    
    def __init__(self, element : IRenderElement) -> None:
        super().__init__(element)
 
    # IRenderElement Members
 
    
    def Render(self) -> str:
        return "\x1b[4m{0}\x1b[24m".format(super().Render())
 
 
 
 
 
 
 
class RedForegroundDecorator(Decorator):
    
    def __init__(self, element : IRenderElement) -> None:
        super().__init__(element)
 
    # IRenderElement Members
 
    
    def Render(self) -> str:
        return "\x1b[31m{0}\x1b[39m".format(super().Render())
 
 
 
 
 
 
class TextElement(IRenderElement):
 
    
    def __init__(self, element : str) -> None:
        self._elementText = element;
    
    
 
    # IRenderElement Members
 
    
    def Render(self) -> str:
        return self._elementText;