boromir674/so-magic

from abc import ABC, abstractmethod, ABCMeta

__all__ = ['GenericMediator', 'BaseComponent']


class Mediator(ABC):
    """
    The Mediator interface declares a method used by components to notify the
    mediator about various events. The Mediator may react to these events and
    pass the execution to other components.
    """
    @abstractmethod
    def notify(self, sender: object, event: str) -> None:
        """[summary]

        Args:
            sender (object): [description]
            event (str): [description]
        """
        raise NotImplementedError


class GenericMediator(Mediator, metaclass=ABCMeta):
    r"""Abstract Mediator class that automatically configures components received as \*args through the constructor."""
    def __new__(cls, *components, **kwargs):
        generic_mediator = super().__new__(cls)
        for i, component in enumerate(components):
            setattr(generic_mediator, f'_component{i+1}', component)
            getattr(generic_mediator, f'_component{i+1}').mediator = generic_mediator
        return generic_mediator


class BaseComponent:
    """
    The Base Component provides the basic functionality of storing a mediator's
    instance inside component objects.
    """

    def __init__(self, mediator: Mediator = None) -> None:
        self._mediator = mediator

    @property
    def mediator(self) -> Mediator:
        return self._mediator

    @mediator.setter
    def mediator(self, mediator: Mediator) -> None:
        self._mediator = mediator