Hopiu/django-categories
1
0
Fork 0
mirror of https://github.com/jazzband/django-categories.git synced 2026-03-16 22:30:24 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 1

Added logic to skip adding categories that are already defined for a modeladmin

Browse source
This commit is contained in:
Erik Simmler 2011-01-17 10:55:16 -05:00
parent 44508e0606
commit 246c18d4e6
1 changed files with 15 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

21
categories/admin.py
View file

@ -103,9 +103,18 @@ admin.site.register(Category, CategoryAdmin)
for model,modeladmin in admin.site._registry.items():
if model in registry.values() and modeladmin.fieldsets:
admin.site.unregister(model)
admin.site.register(model, type('newadmin', (modeladmin.__class__,), {
'fieldsets': getattr(modeladmin, 'fieldsets', ()) + (('Categories',{
'fields': [cat.split('.')[1] for cat in registry]
}),)
}))
fieldsets = getattr(modeladmin, 'fieldsets', ())
fields = [cat.split('.')[1] for cat in registry]
# check each field to see if already defined
for cat in fields:
for k,v in fieldsets:
if cat in v['fields']:
fields.remove(cat)
# if there are any fields left, add them under the categories fieldset
if len(fields) > 0:
admin.site.unregister(model)
admin.site.register(model, type('newadmin', (modeladmin.__class__,), {
'fieldsets': fieldsets + (('Categories',{
'fields': fields
}),)
}))