green_magic.data.encoding

Complexity

Total Complexity

5

Size/Duplication

Total Lines	31
Duplicated Lines	0 %

Importance

Changes

0

4 Methods

Rating	Name	Duplication	Size	Complexity
A	AbstractEncoder.encode()	0	3	1
A	Encoder.register_as_subclass()	0	6	1
A	EncoderInterface.encode()	0	3	1
A	Encoder.create()	0	5	2

import abc
from abc import ABC


class EncoderInterface(abc.ABC):
    @abc.abstractmethod
    def encode(self, *args, **kwargs):
        raise NotImplementedError

class AbstractEncoder(EncoderInterface):
    @abc.abstractmethod
    def encode(self, *args, **kwargs):
        raise NotImplementedError


class Encoder(AbstractEncoder, ABC):
    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('Bad "BinnerFactory Backend type" type \'{}\''.format(backend_type))
        return cls.subclasses[backend_type](*args, **kwargs)