From 7ce6d5d153d0a50e810f90411b1198440a6d88d3 Mon Sep 17 00:00:00 2001 From: Sergio Oliveira Date: Wed, 25 Sep 2013 10:17:40 -0300 Subject: [PATCH] Adding admin classes for custom user --- src/accounts/admin.py | 54 ++++++++++++++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 54 insertions(+), 0 deletions(-) create mode 100644 src/accounts/admin.py diff --git a/src/accounts/admin.py b/src/accounts/admin.py new file mode 100644 index 0000000..866e92e --- /dev/null +++ b/src/accounts/admin.py @@ -0,0 +1,54 @@ + +from django import forms +from django.contrib import admin +from django.contrib.auth.admin import UserAdmin + +from .models import User + + +class UserCreationForm(forms.ModelForm): + class Meta: + model = User + fields = ('username', 'email') + + def __init__(self, *args, **kwargs): + super(UserCreationForm, self).__init__(*args, **kwargs) + self.fields['email'].required = True + + +class UserChangeForm(forms.ModelForm): + class Meta: + model = User + fields = ('username', 'first_name', 'last_name', 'email', 'is_active', + 'is_staff', 'is_superuser', 'groups', 'last_login', + 'date_joined', 'twitter', 'facebook', 'google_talk', + 'webpage') + + def __init__(self, *args, **kwargs): + super(UserChangeForm, self).__init__(*args, **kwargs) + self.fields['email'].required = True + + + +class MyUserAdmin(UserAdmin): + form = UserChangeForm + add_form = UserCreationForm + + fieldsets = ( + (None, {'fields': ('username',)}), + ('Personal info', {'fields': ('first_name', 'last_name', 'email', + 'twitter', 'facebook', 'google_talk', + 'webpage')}), + ('Permissions', {'fields': ('is_active', 'is_staff', 'is_superuser', 'groups')}), + ('Important Dates', {'fields': ('last_login', 'date_joined')}) + ) + + add_fieldsets = ( + (None, { + 'classes': ('wide',), + 'fields': ('username', 'email')} + ), + ) + + +admin.site.register(User, MyUserAdmin) -- libgit2 0.21.2