so_magic.data.backend.backend

Complexity

Total Complexity

6

Size/Duplication

Total Lines	37
Duplicated Lines	0 %

Importance

Changes

0

4 Methods

Rating	Name	Duplication	Size	Complexity
A	Backend.create()	0	5	2
A	Backend.register_as_subclass()	0	6	1
A	Backend.engine()	0	3	1
A	Backend.__init__()	0	4	1

from so_magic.data.dataset import BroadcastingDatapointsFactory, DatapointsManager
from so_magic.utils import Subject


class Backend:
    """
        Args:
            engine (DataEngine): a data engine represented as a class object (eg class MyClass: pass)
    """
    def __init__(self, engine):
        self._engine = engine
        self.datapoints_manager = DatapointsManager()
        self.datapoints_factory = BroadcastingDatapointsFactory(Subject([]))

    @property
    def engine(self):
        return self._engine

    @engine.setter
    def engine(self, engine):
        self._engine = engine

    subclasses = {}

    @classmethod
    def register_as_subclass(cls, backend_type):
        def wrapper(subclass):
            cls.subclasses[backend_type] = subclass
            return subclass
        return wrapper

    @classmethod
    def create(cls, backend_type, *args, **kwargs):
        if backend_type not in cls.subclasses:
            raise ValueError(f"Request Backend of type '{backend_type}'; supported are [{', '.join(sorted(cls.subclasses.keys()))}]")
        return cls.subclasses[backend_type](*args, **kwargs)