so_magic.data.datapoints.datapoints_manager

Complexity

Total Complexity

6

Size/Duplication

Total Lines	30
Duplicated Lines	0 %

Importance

Changes

0

3 Methods

Rating	Name	Duplication	Size	Complexity
A	DatapointsManager.update()	0	9	3
A	DatapointsManager.datapoints()	0	6	2
A	DatapointsManager.state()	0	3	1

import attr
from so_magic.utils import notification as n


@attr.s
class DatapointsManager(n.Observer):
    datapoints_objects = attr.ib(init=True, default={})
    _last_key = attr.ib(init=False, default='')

    def update(self, subject: n.Subject):
        datapoints_object = subject.state
        key = getattr(subject, 'name', '')
        if key == '':
            raise RuntimeError(f"Subject {subject} with state {str(subject.state)} resulted in an empty string as key (to use in dict/hash).")
        if key in self.datapoints_objects:
            raise RuntimeError(f"Attempted to register a new Datapoints object at the existing key '{key}'.")
        self.datapoints_objects[key] = datapoints_object
        self._last_key = key

    @property
    def state(self):
        return self._last_key

    @property
    def datapoints(self):
        try:
            return self.datapoints_objects[self._last_key]
        except KeyError as e:
            print(f"{e}. Requested datapoints with id '{self._last_key}', but was not found in registered [{', '.join(_ for _ in self.datapoints_objects.keys())}]")