Altered the registration naming so more than one field could be regis…

…tered for a model
jazzband · Feb 1, 2010 · 487a0a4 · 487a0a4
1 parent 0ec1fa7
commit 487a0a4
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/categories/__init__.py b/categories/__init__.py
@@ -18,10 +18,10 @@ def register_fk(model, field_name='category', extra_params={}):
     return _register(model, field_name, extra_params, fields.CategoryFKField)
 
 def _register(model, field_name, extra_params={}, field=fields.CategoryFKField):
-    if model in registry:
-        raise AlreadyRegistered('The model %s has already been registered.' % model.__name__)
-    registry.append(model)
-
+    registry_name = "%s.%s" % (model.__name__, field_name)
+    if registry_name in registry:
+        return
+    registry.append(registry_name)
     opts = model._meta
     try:
         opts.get_field(field_name)