Don't error if same generator is registered

This commit is contained in:
Matthew Tretter 2013-04-05 12:25:02 -04:00
parent 681527fa9d
commit 9891314b8e

View file

@ -15,7 +15,8 @@ class GeneratorRegistry(object):
before_access.connect(self.before_access_receiver)
def register(self, id, generator):
if id in self._generators:
registered_generator = self._generators.get(id)
if registered_generator and generator != self._generators[id]:
raise AlreadyRegistered('The generator with id %s is'
' already registered' % id)
self._generators[id] = generator