Don't override UserAdmin for custom auth model

maroux · Aug 13, 2014 · b2da26c · b2da26c
1 parent f15d0a1
commit b2da26c
Showing 1 changed file with 10 additions and 3 deletions.
diff --git a/relationships/admin.py b/relationships/admin.py
@@ -1,5 +1,6 @@
 from django.contrib import admin
 from django.contrib.auth.admin import UserAdmin
+from django.contrib.auth.models import User as DefaultUserModel
 
 from .compat import User
 from .forms import RelationshipStatusAdminForm
@@ -13,13 +14,19 @@ class RelationshipInline(admin.TabularInline):
     fk_name = 'from_user'
 
 
-class UserRelationshipAdmin(UserAdmin):
+class UserRelationshipAdminMixin(object):
     inlines = (RelationshipInline,)
 
 
 class RelationshipStatusAdmin(admin.ModelAdmin):
     form = RelationshipStatusAdminForm
 
-admin.site.unregister(User)
-admin.site.register(User, UserRelationshipAdmin)
+
+if User == DefaultUserModel:
+    class UserRelationshipAdmin(UserRelationshipAdminMixin, UserAdmin):
+        pass
+
+    admin.site.unregister(User)
+    admin.site.register(User, UserRelationshipAdmin)
+
 admin.site.register(RelationshipStatus, RelationshipStatusAdmin)