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

from .models import Competence, CustomUser, StatutCategory, Worker
from .models import MissionCategory, Structure, Mission, Motif, ProfilRecherche
from .models import Service, Metier

class CustomUserAdmin(UserAdmin):
    # pass
    list_display = (
        'username', 'email', 'first_name', 'last_name', 'is_staff',
        'is_admin', 'is_manager', 'is_worker'
    )

    fieldsets = (
        (None, {
            'fields': ('username', 'password')
        }),
        ('Personal info', {
            'fields': ('first_name', 'last_name', 'email')
        }),
        ('Permissions', {
            'fields': (
                'is_active', 'is_staff', 'is_superuser',
                'groups', 'user_permissions'
            )
        }),
        ('Important dates', {
            'fields': ('last_login', 'date_joined')
        }),
        ('Additional info', {
            'fields': ('is_admin', 'is_manager', 'is_worker')
        })
    )

    add_fieldsets = (
        (None, {
            'fields': ('username', 'password1', 'password2')
        }),
        ('Personal info', {
            'fields': ('first_name', 'last_name', 'email')
        }),
        ('Permissions', {
            'fields': (
                'is_active', 'is_staff', 'is_superuser',
                'groups', 'user_permissions'
            )
        }),
        ('Important dates', {
            'fields': ('last_login', 'date_joined')
        }),
        ('Additional info', {
            'fields': ('is_admin', 'is_manager', 'is_worker')
        })
    )


admin.site.register(CustomUser, CustomUserAdmin)

admin.site.register(Structure)
admin.site.register(MissionCategory)
admin.site.register(Mission)
admin.site.register(Motif)
admin.site.register(ProfilRecherche)
admin.site.register(Worker)
admin.site.register(StatutCategory)
admin.site.register(Competence)
admin.site.register(Service)
admin.site.register(Metier)