Remove django-admin2 from the dashboard if it's not been activated

This commit is contained in:
Daniel Greenfeld 2013-07-07 10:42:55 +02:00
parent ed23fbf09e
commit e66175ed21

View file

@ -1,7 +1,7 @@
from django.conf import settings
from django.contrib.auth.models import Group, User
from django.contrib.sites.models import Site
from rest_framework.relations import PrimaryKeyRelatedField
import djadmin2
@ -36,12 +36,16 @@ class UserAdmin2(djadmin2.ModelAdmin2):
api_serializer_class = UserSerializer
class SiteAdmin2(djadmin2.ModelAdmin2):
list_display = ('domain', 'name')
search_fields = ('domain', 'name')
# Register each model with the admin
djadmin2.default.register(User, UserAdmin2)
djadmin2.default.register(Group, GroupAdmin2)
djadmin2.default.register(Site, SiteAdmin2)
# Register the sites app if it's been activated in INSTALLED_APPS
if "django.contrib.sites" in settings.INSTALLED_APPS:
class SiteAdmin2(djadmin2.ModelAdmin2):
list_display = ('domain', 'name')
search_fields = ('domain', 'name')
djadmin2.default.register(Site, SiteAdmin2)