green_magic.utils.registry

Complexity

Total Complexity

11

Size/Duplication

Total Lines	42
Duplicated Lines	0 %

Importance

Changes

0

6 Methods

Rating	Name	Duplication	Size	Complexity
A	ObjectRegistry.get()	0	4	2
A	ObjectRegistry.__contains__()	0	2	1
A	ObjectRegistry.add()	0	5	2
A	ObjectRegistry.__new__()	0	7	2
A	ObjectRegistry.pop()	0	4	2
A	ObjectRegistry.remove()	0	4	2

from abc import ABC

__all__ = ['ObjectRegistry', 'ObjectRegistryError']


class ObjectRegistry(ABC):
    """Simple dict-like retrieval/inserting "store" facility."""

    def __new__(cls, *args, **kwargs):
        x = super().__new__(cls)
        if args:
            x.objects = args[0]
        else:
            x.objects = {}
        return x

    def add(self, key, value):
        if self.objects.get(key, None):
            raise ObjectRegistryError(f"Requested to insert value {value} in already existing key {key}."
                                      f"All keys are [{', '.join(_ for _ in self.objects)}]")
        self.objects[key] = value

    def remove(self, key):
        if key not in self.objects:
            raise ObjectRegistryError(f"Requested to remove item with key {key}, which does not exist.")
        self.objects.pop(key)

    def pop(self, key):
        if key not in self.objects:
            raise ObjectRegistryError(f"Requested to remove item with key {key}, which does not exist.")
        return self.objects.pop(key)

    def get(self, key):
        if key not in self.objects:
            raise ObjectRegistryError(f"Requested to get item with key {key}, which does not exist.")
        return self.objects[key]

    def __contains__(self, item):
        return item in self.objects

class ObjectRegistryError(Exception): pass