from copy import deepcopy #@UnresolvedImport

from django.contrib import admin

from django.contrib.auth.admin import UserAdmin

from django.contrib.auth.models import Group, User

from django.utils.translation import ugettext as _

from forms import LdtForm

from guardian.admin import GuardedModelAdmin

from models import Ldt, UserProfile

class GroupAdmin(GuardedModelAdmin):

    pass

class UserProfileInline(admin.StackedInline):

    model = UserProfile

class UserProfileAdmin(UserAdmin):

    inlines = [UserProfileInline, ]

class LdtAdmin(UserProfileAdmin):    

    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',)

    inlines = [UserProfileInline, ]

    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(Group)

admin.site.register(Group, GroupAdmin)

admin.site.unregister(Ldt) 

admin.site.register(Ldt, LdtAdmin)

admin.site.unregister(User)

admin.site.register(User, UserProfileAdmin)