Djangocon coding

[tranquillity.git] / customerportal / admin.py

diff --git a/customerportal/admin.py b/customerportal/admin.py
index 8c38f3f3dad51e4585f3984282c2a4bec5349c1e..e2bdf1661dc71bcd76ce140c7bed6641dbb8e8fd 100644 (file)
--- 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',)