attempt at hacking a through table (field errors)

categories/__init__.py

@@ -1,5 +1,4 @@
 import fields
-import models
 
 from django.db.models import FieldDoesNotExist
 
@@ -9,7 +8,7 @@ class AlreadyRegistered(Exception):
     """
     pass
 
-registry = []
+registry = {}
 
 def register_m2m(model, field_name='categories', extra_params={}):
     return _register(model, field_name, extra_params, fields.CategoryM2MField)
@@ -20,8 +19,8 @@ def register_fk(model, field_name='category', extra_params={}):
 def _register(model, field_name, extra_params={}, field=fields.CategoryFKField):
     registry_name = "%s.%s" % (model.__name__, field_name)
     if registry_name in registry:
-        return
-    registry.append(registry_name)
+        raise AlreadyRegistered
+    registry[registry_name] = model
     opts = model._meta
     try:
         opts.get_field(field_name)

categories/admin.py

@@ -61,19 +61,19 @@ def clean(self):
 
 
 class CategoryAdmin(TreeEditor, admin.ModelAdmin):
-    form=CategoryAdminForm
+    form = CategoryAdminForm
     list_display = ('__unicode__',)
     search_fields = (('name',))
     prepopulated_fields = {'slug': ('name',)}
 
 
 admin.site.register(Category, CategoryAdmin)
 
-
-class CategorizedAdmin(admin.ModelAdmin):
-    change_form_template = 'admin/categories/change_form.html'
-
 for model,modeladmin in admin.site._registry.items():
-    if model in registry:
+    if model in registry.values():
         admin.site.unregister(model)
-        admin.site.register(model, type('newadmin', (CategorizedAdmin, modeladmin.__class__,), {}))
+        admin.site.register(model, type('newadmin', (modeladmin.__class__,), {
+            'fieldsets': modeladmin.fieldsets + (('Categories',{
+                'fields': ('category','categories'),
+            }),)
+        }))

categories/fields.py

@@ -1,11 +1,12 @@
 from django.db.models import ForeignKey, ManyToManyField
-from categories.models import Category
 
 class CategoryM2MField(ManyToManyField):
     def __init__(self, **kwargs):
+        from categories.models import Category
         super(CategoryM2MField, self).__init__(to=Category, **kwargs)
 
 
 class CategoryFKField(ForeignKey):
     def __init__(self, **kwargs):
+        from categories.models import Category
         super(CategoryFKField, self).__init__(to=Category, **kwargs)

categories/models.py

@@ -33,3 +33,25 @@ def __unicode__(self):
         return ' > '.join([force_unicode(i.name) for i in ancestors]+[self.name,])
 
 mptt.register(Category, order_insertion_by=['name'])
+
+
+class CategoryIntermediary(models.Model):
+    category = models.ForeignKey(Category)
+    is_primary = models.BooleanField(default=False)
+
+    class Meta:
+        abstract = True
+
+from categories import registry
+from django.contrib import admin
+
+for model in registry.values():
+    field_name = model._meta.verbose_name.lower()
+    class_name = str('%sCategories' % model._meta.verbose_name.title())
+    attrs = {
+        'category': models.ForeignKey(Category),
+        'is_primary': models.BooleanField(default=False),
+        '__module__': model.__dict__['__module__'],
+        field_name: models.ForeignKey(model),
+    }
+    admin.site.register(type(class_name, (models.Model,), attrs))