green_magic.data.magic_encoder

Complexity

Total Complexity

1

Size/Duplication

Total Lines	16
Duplicated Lines	0 %

Importance

Changes

0

1 Method

Rating	Name	Duplication	Size	Complexity
A	MagicEncoder.encode()	0	6	1

import attr
from green_magic.utils import Subject


@attr.s
class MagicEncoder:
    encoders = attr.ib(init=True, default={})
    subject = attr.ib(init=True, default=Subject([]))

    def encode(self, *args, **kwargs):
        encoder = self.encoders[args[0]]

        self.subject.state = encoder.encode(*args[1:], **kwargs)
        self.subject.columns = encoder.columns
        self.subject.notify()