From 9891314b8e77d82edf7eafe3bd237da0ac585da5 Mon Sep 17 00:00:00 2001 From: Matthew Tretter Date: Fri, 5 Apr 2013 12:25:02 -0400 Subject: [PATCH] Don't error if same generator is registered --- imagekit/registry.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/imagekit/registry.py b/imagekit/registry.py index a28738f..620f9f8 100644 --- a/imagekit/registry.py +++ b/imagekit/registry.py @@ -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