Skip to content

Commit

Permalink
Fixed a problem in the new admin creation where it wouldn't properly …
Browse files Browse the repository at this point in the history
…filter out the category fields by model
  • Loading branch information
Corey Oordt committed May 16, 2011
1 parent 97c45e7 commit d90dccd
Showing 1 changed file with 4 additions and 3 deletions.
7 changes: 4 additions & 3 deletions categories/admin.py
Original file line number Diff line number Diff line change
Expand Up @@ -101,20 +101,21 @@ class Media:

admin.site.register(Category, CategoryAdmin)

for model,modeladmin in admin.site._registry.items():
for model, modeladmin in admin.site._registry.items():
if model in registry.values() and modeladmin.fieldsets:
fieldsets = getattr(modeladmin, 'fieldsets', ())
fields = [cat.split('.')[1] for cat in registry]
fields = [cat.split('.')[1] for cat in registry if registry[cat] == model]
# 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:
print fields
admin.site.unregister(model)
admin.site.register(model, type('newadmin', (modeladmin.__class__,), {
'fieldsets': fieldsets + (('Categories',{
'fieldsets': fieldsets + (('Categories', {
'fields': fields
}),)
}))

0 comments on commit d90dccd

Please sign in to comment.