X-Git-Url: https://git.frykholm.com/tranquillity.git/blobdiff_plain/93be18536faaa33c9c2d3a4e5625fa7bfca8eb20..40e726c7920217dc583a6f23b9ff0d5c8e1c801e:/customerportal/admin.py

diff --git a/customerportal/admin.py b/customerportal/admin.py
index 8c38f3f..e2bdf16 100644
--- a/customerportal/admin.py
+++ b/customerportal/admin.py
@@ -1,3 +1,29 @@
+"""Integrate with admin module."""
+
 from django.contrib import admin
+from django.contrib.auth.admin import UserAdmin as DjangoUserAdmin
+from django.utils.translation import ugettext_lazy as _
+
+from .models import User
+
+
+@admin.register(User)
+class UserAdmin(DjangoUserAdmin):
+    """Define admin model for custom User model with no email field."""
 
-# Register your models here.
+    fieldsets = (
+        (None, {'fields': ('email', 'password')}),
+        (_('Personal info'), {'fields': ('first_name', 'last_name','fortnox_external_id')}),
+        (_('Permissions'), {'fields': ('is_active', 'is_staff', 'is_superuser',
+                                       'groups', 'user_permissions')}),
+        (_('Important dates'), {'fields': ('last_login', 'date_joined')}),
+    )
+    add_fieldsets = (
+        (None, {
+            'classes': ('wide',),
+            'fields': ('email', 'password1', 'password2'),
+        }),
+    )
+    list_display = ('email', 'first_name', 'last_name', 'is_staff')
+    search_fields = ('email', 'first_name', 'last_name')
+    ordering = ('email',)