from django.contrib import admin

from django.contrib.auth.admin import UserAdmin, GroupAdmin

from django.utils.translation import ugettext as _

from ldt import settings

from copy import deepcopy

from forms import ldtForm, IriGroupForm

from models import ldt, IriGroup

class ldtAdmin(UserAdmin):    

    list_display = ('username', 'email', 'first_name', 'last_name')

    fieldsets = [

        (None, {'fields': ('username', ('password1', 'password2'))}),

        (_('User details'), {'fields': (('first_name', 'last_name'), 'email')}),

        (_('Groups'), {'fields': ('groups',)}),

        (_('Permissions'), {'fields': ('is_staff', 'user_permissions')}),        

    ]    

    form = ldtForm

    model = ldt

    filter_horizontal = ('user_permissions',)

    def get_fieldsets(self, request, obj=None): 

        fieldsets = deepcopy(self.fieldsets)

        if not '/add' in request.path:

            fieldsets[0] = (None, {'fields': ('username',)})

            fieldsets.append((_('Password'), {'fields': ('password1', 'password2'), 'classes': ('collapse',)}))

        return fieldsets

    def add_view(self, request):

        return super(UserAdmin, self).add_view(request)               

admin.site.unregister(ldt)

admin.site.register(ldt, ldtAdmin)

class IriGroupAdmin(admin.ModelAdmin):

    form = IriGroupForm

    model = IriGroup

    filter_horizontal = ('permissions',)

admin.site.unregister(IriGroup)

admin.site.register(IriGroup, IriGroupAdmin)