from django.contrib import admin

from django.contrib.auth.admin import UserAdmin as BaseUserAdmin, GroupAdmin as BaseGroupAdmin

from django.contrib.auth.models import Group

from user.forms import UserChangeForm, UserCreationForm

from user.models import User, BlockedUser

class UserAdmin(BaseUserAdmin):

permission_field_name = "user_permissions"

# The forms to add and change user instances

form = UserChangeForm

add_form = UserCreationForm

# The fields to be used in displaying the User model.

# These override the definitions on the base UserAdmin

# that reference specific fields on auth.User.

list_display = ('email', 'is_staff')

list_filter = ('is_staff', 'is_superuser', 'groups')

fieldsets = (

(None, {'fields': ('email', 'password')}),

('Personal info', {'fields': ('first_name', 'last_name', 'phone_number', 'diet', 'other_diet', 'gender',

'other_gender', 'under_age', 'tshirt_size', 'qr_code')}),

('Permissions', {'fields': ('email_verified', 'is_staff', 'is_superuser', 'groups', 'user_permissions',

'is_active')}),

)

# add_fieldsets is not a standard ModelAdmin attribute. UserAdmin

# overrides get_fieldsets to use this attribute when creating a user.

add_fieldsets = (

(None, {

'classes': ('wide',),

'fields': ('first_name', 'last_name', 'email', 'password1', 'password2'),

}),

)

search_fields = ('email', 'first_name', 'last_name')

ordering = ('email', 'first_name', 'last_name')

filter_horizontal = ()

class GroupAdmin(BaseGroupAdmin):

permission_field_name = 'permissions'

# Now register the new UserAdmin...

admin.site.register(User, UserAdmin)

admin.site.unregister(Group)

admin.site.register(Group, GroupAdmin)

admin.site.register(BlockedUser)