Move registry to separate file.

This commit is contained in:
Emil Stenström 2015-06-11 21:11:13 +02:00
parent ebf4ac1d20
commit df22a4049f
2 changed files with 22 additions and 21 deletions

View file

@ -0,0 +1,21 @@
class AlreadyRegistered(Exception):
pass
class NotRegistered(Exception):
pass
class ComponentRegistry(object):
def __init__(self):
self._registry = {} # component name -> component_class mapping
def register(self, name=None, component=None):
if name in self._registry:
raise AlreadyRegistered('The component "%s" is already registered' % name)
self._registry[name] = component
def unregister(self, name):
if name not in self._registry:
raise NotRegistered('The component "%s" is not registered' % name)
del self._registry[name]