From e66175ed210c0d668d41890c20d667d4b6029548 Mon Sep 17 00:00:00 2001 From: Daniel Greenfeld Date: Sun, 7 Jul 2013 10:42:55 +0200 Subject: [PATCH] Remove django-admin2 from the dashboard if it's not been activated --- djadmin2/admin2.py | 18 +++++++++++------- 1 file changed, 11 insertions(+), 7 deletions(-) diff --git a/djadmin2/admin2.py b/djadmin2/admin2.py index 4cb7f69..947987a 100644 --- a/djadmin2/admin2.py +++ b/djadmin2/admin2.py @@ -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)