From 487a0a4f2388611b053749ed7e45283eb5f30fd7 Mon Sep 17 00:00:00 2001 From: Corey Oordt Date: Mon, 1 Feb 2010 10:28:31 -0500 Subject: [PATCH] Altered the registration naming so more than one field could be registered for a model --- categories/__init__.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/categories/__init__.py b/categories/__init__.py index 602831e..151ed00 100644 --- 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)