[auf_rh_dae.git] / project / rh / admin.py

# -*- encoding: utf-8 -*-

from collections import defaultdict
import datetime

from django.db import models
from django import forms
from django.core.urlresolvers import reverse
from django.contrib import admin
from django.conf import settings
from django.db.models import Q
from ajax_select import make_ajax_form
from auf.django.metadata.admin import AUFMetadataAdminMixin, AUFMetadataInlineAdminMixin, AUF_METADATA_READONLY_FIELDS
from forms import ContratForm, AyantDroitForm, EmployeAdminForm, AjaxSelect
from dae.utils import get_employe_from_user
import models as rh

# Override of the InlineModelAdmin to support the link in the tabular inline
class LinkedInline(admin.options.InlineModelAdmin):
    template = "admin/linked.html"
    admin_model_path = None

    def __init__(self, *args):
        super(LinkedInline, self).__init__(*args)
        if self.admin_model_path is None:
            self.admin_model_path = self.model.__name__.lower()


class ProtectRegionMixin(object):

    def queryset(self, request):
        from dae.workflow import grp_drh, grp_correspondants_rh
        qs = super(ProtectRegionMixin, self).queryset(request)

        if request.user.is_superuser:
            return qs

        user_groups = request.user.groups.all()

        if grp_drh in user_groups:
            return qs

        if grp_correspondants_rh in user_groups:
            employe = get_employe_from_user(request.user)
            q = Q(**{self.model.prefix_implantation: employe.implantation.region})
            qs = qs.filter(q).distinct()
            return qs
        return qs.none()

    def has_change_permission(self, request, obj=None):
        if obj is None:
            return True
        ids = [o.id for o in self.queryset(request)]
        return obj.id in ids


# Inlines

class ReadOnlyInlineMixin(object):
    def get_readonly_fields(self, request, obj=None):
        return [f.name for f in self.model._meta.fields if f.name not in AUF_METADATA_READONLY_FIELDS]


class AyantDroitInline(AUFMetadataInlineAdminMixin, admin.StackedInline):
    model = rh.AyantDroit
    form = AyantDroitForm
    extra = 0

    fieldsets = (
        (None, {
            'fields': (('nom', 'prenom'), ('nom_affichage', 'genre'), 'nationalite', 'date_naissance', 'lien_parente', )
        }),
    )


class AyantDroitCommentaireInline(AUFMetadataInlineAdminMixin, admin.TabularInline):
    readonly_fields = ('owner', )
    model = rh.AyantDroitCommentaire
    extra = 1


class ContratInline(AUFMetadataInlineAdminMixin, admin.TabularInline):
    form = ContratForm
    model = rh.Contrat
    extra = 1


class DossierROInline(ReadOnlyInlineMixin, LinkedInline):
    template = "admin/rh/dossier/linked.html"
    exclude = AUF_METADATA_READONLY_FIELDS
    model = rh.Dossier
    extra = 0
    can_delete = False

    def has_add_permission(self, request=None):
        return False

    def has_change_permission(self, request, obj=None):
        return False

    def has_delete_permission(self, request, obj=None):
        return False


class DossierCommentaireInline(AUFMetadataInlineAdminMixin, admin.TabularInline):
    readonly_fields = ('owner', )
    model = rh.DossierCommentaire
    extra = 1


class DossierPieceInline(admin.TabularInline):
    model = rh.DossierPiece
    extra = 4


class EmployeInline(admin.TabularInline):
    model = rh.Employe

class EmployeCommentaireInline(AUFMetadataInlineAdminMixin, admin.TabularInline):
    readonly_fields = ('owner', )
    model = rh.EmployeCommentaire
    extra = 1


class EmployePieceInline(admin.TabularInline):
    model = rh.EmployePiece
    extra = 4


class PosteCommentaireInline(AUFMetadataInlineAdminMixin, admin.TabularInline):
    readonly_fields = ('owner', )
    model = rh.PosteCommentaire
    extra = 1


class PosteFinancementInline(admin.TabularInline):
    model = rh.PosteFinancement


class PostePieceInline(admin.TabularInline):
    model = rh.PostePiece


class RemunerationInline(AUFMetadataInlineAdminMixin, admin.TabularInline):
    model = rh.Remuneration
    extra = 1


class RemunerationROInline(ReadOnlyInlineMixin, RemunerationInline):
    pass


class TypePosteInline(AUFMetadataInlineAdminMixin, admin.TabularInline):
    model = rh.TypePoste


class AyantDroitAdmin(AUFMetadataAdminMixin, ProtectRegionMixin, admin.ModelAdmin):
    """
    L'ajout d'un nouvel ayantdroit se fait dans l'admin de l'employé.
    """
    alphabet_filter = 'nom'
    search_fields = ('nom', 'prenom', 'employe__nom', 'employe__prenom', )
    list_display = ('_employe', 'lien_parente', '_ayantdroit', )
    inlines = (AyantDroitCommentaireInline,)
    readonly_fields = AUFMetadataAdminMixin.readonly_fields + ('employe',)
    fieldsets = AUFMetadataAdminMixin.fieldsets + (
        ("Lien avec l'employé", {
            'fields': (('employe', 'lien_parente'), )
        }),

        ('Identification', {
            'fields': (('nom', 'prenom'), ('nom_affichage', 'genre'), 'nationalite', 'date_naissance', )
        }),
    )

    def save_formset(self, request, form, formset, change):
        instances = formset.save(commit=False)
        for instance in instances:
            if instance.__class__ == rh.AyantDroitCommentaire:
                instance.owner = request.user
            instance.save()

    def _ayantdroit(self, obj):
        return unicode(obj)
    _ayantdroit.short_description = u'Ayant droit'

    def _employe(self, obj):
        return unicode(obj.employe)
    _employe.short_description = u'Employé'

    def has_add_permission(self, request):
        return False

class AyantDroitCommentaireAdmin(admin.ModelAdmin):
    pass


class ClassementAdmin(AUFMetadataAdminMixin, admin.ModelAdmin):
    list_display = ('_classement', 'date_modification', 'user_modification', )
    fieldsets = AUFMetadataAdminMixin.fieldsets + (
        (None, {
            'fields': ('type', 'echelon', 'degre', 'coefficient',  )
        }),
    )

    def _classement(self, obj):
        return unicode(obj)
    _classement.short_description = u"Classement"

class CommentaireAdmin(admin.ModelAdmin):
    pass


class DeviseAdmin(AUFMetadataAdminMixin, admin.ModelAdmin):
    list_display = ('code', 'nom', 'date_modification', 'user_modification', 'actif', )
    fieldsets = AUFMetadataAdminMixin.fieldsets + (
        (None, {
            'fields': ('code', 'nom', ),
        }),
    )


class DossierAdmin(AUFMetadataAdminMixin, ProtectRegionMixin, admin.ModelAdmin, AjaxSelect):
    alphabet_filter = 'employe__nom'
    search_fields = ('employe__nom', 'employe__prenom', 'poste__nom', 'poste__nom_feminin')
    list_display = (
        '_id',
        '_poste',
        '_employe',
        '_date_debut',
        '_date_fin',
        'date_modification',
        'user_modification',
        'actif',
    )
    list_filter = (
        'poste__implantation__region',
        'poste__implantation',
        'poste__type_poste',
        'poste__type_poste__famille_emploi',
        'rh_contrats__type_contrat',
        'actif',
    )
    inlines = (DossierPieceInline, ContratInline,
               RemunerationInline,
               DossierCommentaireInline,
              )
    fieldsets = AUFMetadataAdminMixin.fieldsets + (
        (None, {
            'fields': ('employe', 'poste', 'statut', 'organisme_bstg',)
        }),
        ('Recrutement', {
            'fields': ('statut_residence', 'remplacement', 'remplacement_de', )
        }),
        ('Rémunération', {
            'fields': ('classement', ('regime_travail', 'regime_travail_nb_heure_semaine'),)
        }),
        ('Occupation du Poste par cet Employe', {
            'fields': (('date_debut', 'date_fin'), )
        }),
    )
    form =  make_ajax_form(rh.Dossier, {
        'employe' : 'employes',
        'poste' : 'postes',
        'remplacement_de' : 'dossiers',
    })

    def lookup_allowed(self, key, value):
        if key in (
            'employe__nom__istartswith',
            'actif__exact',
            'poste__implantation__region__id__exact',
            'poste__implantation__id__exact',
            'poste__type_poste__id__exact',
            'poste__type_poste__famille_emploi__id__exact',
            'rh_contrats__type_contrat__id__exact',
            ):
            return True

    def _id(self, d):
        link = u"""<a onclick="return showAddAnotherPopup(this);" href='%s'>%s</a> <a href="%s" title="Modifier le dossier"><img src="%simg/page_edit.png" /></a>""" % \
            (reverse('dossier_apercu', args=(d.id,)),
             d.id,
             reverse('admin:rh_dossier_change', args=(d.id,)),
             settings.MEDIA_URL,
             )
        return link
    _id.allow_tags = True
    _id.short_description = u'#'
    _id.admin_order_field = 'id'
    

    def _actif(self, dossier):
        if dossier.employe.actif:
          html = """<img alt="True" src="%simg/admin/icon-yes.gif">"""
        else:
          html = """<img alt="False" src="%simg/admin/icon-no.gif">"""
        return html % settings.ADMIN_MEDIA_PREFIX 
    _actif.allow_tags = True
    _actif.short_description = u'Employé actif'
    _actif.admin_order_field = 'employe__actif'

    def _date_debut(self, obj):
        return obj.date_debut
    _date_debut.short_description = u'Occupation début'
    _date_debut.admin_order_field = 'date_debut'

    def _date_fin(self, obj):
        return obj.date_fin
    _date_fin.short_description = u'Occupation fin'
    _date_fin.admin_order_field = 'date_fin'

    def _poste(self, dossier):
        link = u"""<a onclick="return showAddAnotherPopup(this);" href='%s'>%s</a> <a href="%s" title="Modifier le poste"><img src="%simg/page_edit.png" /></a>""" % \
                (reverse('poste_apercu', args=(dossier.poste.id,)),
                 dossier.poste,
                 reverse('admin:rh_poste_change', args=(dossier.poste.id,)),
                 settings.MEDIA_URL,
                 )
        return link
    _poste.allow_tags = True
    _poste.short_description = u'Poste'
    _poste.admin_order_field = 'poste__nom'

    def _employe(self, obj):
        employe = obj.employe
        view_link = reverse('employe_apercu', args=(employe.id,))
        edit_link = reverse('admin:rh_employe_change', args=(employe.id,))

        if employe.actif == False:
            style = "color: grey";
            edit = ""
        else:
            style = ""
            edit = u"""<a href="%s" title="Modifier l'employé"><img src="%simg/user_edit.png" /></a>""" % (edit_link, settings.MEDIA_URL,)
        return u"""<a onclick="return showAddAnotherPopup(this);" href='%s' style="%s;">[%s] %s %s</a>%s
                  """ % \
        (view_link, style, employe.id, employe.nom.upper(), employe.prenom.title(), edit)
    _employe.allow_tags = True
    _employe.short_description = u"Employé ([code] NOM Prénom)"
    _employe.admin_order_field = "employe__nom"
   
    def save_formset(self, request, form, formset, change):
        instances = formset.save(commit=False)
        for instance in instances:
            if instance.__class__ == rh.DossierCommentaire:
                instance.owner = request.user
            instance.save()


class DossierPieceAdmin(admin.ModelAdmin):
    pass


class DossierCommentaireAdmin(admin.ModelAdmin):
    pass


class EmployeAdmin(AUFMetadataAdminMixin, ProtectRegionMixin, admin.ModelAdmin):
    alphabet_filter = 'nom'
    DEFAULT_ALPHABET = u'ABCDEFGHIJKLMNOPQRSTUVWXYZ'
    search_fields = ('id', 'nom', 'prenom', 'nom_affichage', )
    ordering = ('nom', )
    form = EmployeAdminForm
    list_display = ('_nom', '_dossiers', 'date_modification',  'user_modification', 'actif',)
    list_filter = ('rh_dossiers__poste__implantation__region',  'rh_dossiers__poste__implantation', 'actif', )
    inlines = (AyantDroitInline,
               DossierROInline,
               EmployePieceInline,
               EmployeCommentaireInline)
    fieldsets = AUFMetadataAdminMixin.fieldsets + (
        ('Identification', {
            'fields': (('nom', 'prenom'), ('nom_affichage', 'genre'), 'nationalite', 'date_naissance', )
        }),
        ('Informations personnelles', {
            'fields': ('situation_famille', 'date_entree', )
        }),
        ('Coordonnées', {
            'fields': (('tel_domicile', 'tel_cellulaire'), ('adresse', 'ville'), ('code_postal', 'province'), 'pays', )
        }),
        )

    def _nom(self, obj):
        view_link = reverse('employe_apercu', args=(obj.id,))
        edit_link = reverse('admin:rh_employe_change', args=(obj.id,))
        return u"""<a onclick="return showAddAnotherPopup(this);" href='%s'>[%s] %s %s</a>
                  <a href="%s" title="Modifier l'employé"><img src="%simg/user_edit.png" /></a>""" % \
        (view_link, obj.id, obj.nom.upper(), obj.prenom.title(), edit_link, settings.MEDIA_URL,)
    _nom.allow_tags = True
    _nom.short_description = u"Employé ([code] NOM Prénom)"
    _nom.admin_order_field = "nom"

    def _dossiers(self, obj):
        l = []
        for d in obj.rh_dossiers.all().order_by('-date_debut'):
            style = ""
            edit = u"""<a href="%s" title="Modifier le dossier"><img src="%simg/page_edit.png" /></a>""" % (reverse('admin:rh_dossier_change', args=(d.id,)), settings.MEDIA_URL,)
            if d.date_fin is not None:
                edit = ""
                style = u"color: grey";
            link = u"""<li><a style="%s;" onclick="return showAddAnotherPopup(this);" href='%s'>%s : %s</a>%s</li>""" % \
                (style,
                 reverse('dossier_apercu', args=(d.id,)),
                 d.date_debut.year,
                 d.poste,
                 edit,
                 )
            l.append(link)
        return "<ul>%s</ul>" % "\n".join(l)
    _dossiers.allow_tags = True
    _dossiers.short_description = u"Dossiers"

    def queryset(self, request):
        qs = super(EmployeAdmin, self).queryset(request) 
        return qs.select_related(depth=1).order_by('nom')

    def save_formset(self, request, form, formset, change):
        instances = formset.save(commit=False)
        for instance in instances:
            if instance.__class__ == rh.EmployeCommentaire:
                instance.owner = request.user
            instance.save()


class EmployeCommentaireAdmin(admin.ModelAdmin):
    pass


class EmployePieceAdmin(admin.ModelAdmin):
    pass


class FamilleEmploiAdmin(AUFMetadataAdminMixin, admin.ModelAdmin):
    list_display = ('nom', 'date_modification', 'user_modification', 'actif', )
    inlines = (TypePosteInline,)
    fieldsets = AUFMetadataAdminMixin.fieldsets + ( 
        (None, {
            'fields': ('nom', )
        }),
        )


class OrganismeBstgAdmin(AUFMetadataAdminMixin, ProtectRegionMixin, admin.ModelAdmin):
    search_fields = ('nom',)
    list_display = ('nom', 'type', 'pays', 'date_modification', 'user_modification', 'actif', )
    list_filter = ('type', )
    inlines = (DossierROInline,)
    fieldsets = AUFMetadataAdminMixin.fieldsets + ( 
        (None, {
            'fields': ('nom', 'type', 'pays', )
        }),
        )


class PosteAdmin(AUFMetadataAdminMixin, ProtectRegionMixin, admin.ModelAdmin, AjaxSelect):
    form =  make_ajax_form(rh.Poste, {
        'implantation' : 'implantations',
        'type_poste' : 'typepostes',
        'responsable' : 'postes',
        'valeur_point_min' : 'valeurpoints',
        'valeur_point_max' : 'valeurpoints',
    })
    alphabet_filter = 'nom'
    search_fields = ('nom', 
        'implantation__code', 
        'implantation__nom', 
        'implantation__region__code', 
        'implantation__region__nom',
        )
    list_display = (
        '_nom', 
        '_occupe_par',
        'implantation', 
        '_service', 
        'date_debut', 
        'date_fin',
        'date_modification',
        'user_modification',
        'actif',
        )
    list_filter = ('service', 
        'implantation__region', 
        'implantation',
        'type_poste',
        'type_poste__famille_emploi',
        'vacant',
        'actif',
        )
    fieldsets = AUFMetadataAdminMixin.fieldsets + (
        (None, {
            'fields': (('nom', 'nom_feminin'), 'implantation', 'type_poste',
                       'service', 'responsable')
        }),
        ('Contrat', {
            'fields': (('regime_travail', 'regime_travail_nb_heure_semaine'), )
        }),
        ('Recrutement', {
            'fields': (('local', 'expatrie', 'mise_a_disposition', 'appel'),)
        }),
        ('Rémunération', {
            'fields': (('classement_min', 'valeur_point_min', 'devise_min', 'salaire_min', 'indemn_min', 'autre_min', ),
                       ('classement_max', 'valeur_point_max' ,'devise_max', 'salaire_max', 'indemn_max', 'autre_max', ),
                        )
        }),
        ('Comparatifs de rémunération', {
            'fields': ('devise_comparaison',
                       ('comp_locale_min', 'comp_locale_max'),
                       ('comp_universite_min', 'comp_universite_max'),
                       ('comp_fonctionpub_min', 'comp_fonctionpub_max'),
                       ('comp_ong_min', 'comp_ong_max'),
                       ('comp_autre_min', 'comp_autre_max'))
        }),
        ('Justification', {
            'fields': ('justification',)
        }),
        ('Autres Metadata', {
            'fields': ('date_debut', 'date_fin')
        }),
    )

    inlines = (PosteFinancementInline,
               PostePieceInline,
               DossierROInline,
               PosteCommentaireInline, )

    
    def _service(self, obj):
        return obj.service

    def _nom(self, poste):
        link = u"""<a onclick="return showAddAnotherPopup(this);" href='%s'>%s</a> <a href="%s" title="Modifier le poste"><img src="%simg/page_edit.png" /></a>""" % \
                (reverse('poste_apercu', args=(poste.id,)),
                 poste.nom,
                 reverse('admin:rh_poste_change', args=(poste.id,)),
                 settings.MEDIA_URL,
                 )
        return link
    _nom.allow_tags = True
    _nom.short_description = u'Nom'
    _nom.admin_order_field = 'nom'

    def _occupe_par(self, obj):
        """Formatte la méthode Poste.occupe_par() pour l'admin"""
        output = u"Vacant"
        if obj.actif is False:
            return u"-"
        employes = obj.occupe_par()
        if employes:
            l = []
            for e in employes:
                link = "<a href='%s'>%s</a>" % \
                    (reverse('admin:rh_employe_change', args=(e.id,)),
                     e)
                l.append(link)
            output = "\n<br />".join(l)
        return output
    _occupe_par.allow_tags = True
    _occupe_par.short_description = "Occupé par"   

    def save_formset(self, request, form, formset, change):
        instances = formset.save(commit=False)
        for instance in instances:
            if instance.__class__ == rh.PosteCommentaire:
                instance.owner = request.user
            instance.save()
        formset.save_m2m()


class PosteCommentaireAdmin(admin.ModelAdmin):
    pass


class PosteFinancementAdmin(admin.ModelAdmin):
    pass


class PostePieceAdmin(admin.ModelAdmin):
    fk_name = 'poste'


class RemunerationAdmin(admin.ModelAdmin):
    pass


class ResponsableImplantationAdmin(AUFMetadataAdminMixin, admin.ModelAdmin):
    fieldsets = AUFMetadataAdminMixin.fieldsets + ( 
        (None, {
            'fields': ('employe', 'implantation', ),
        }),
        )
    

class ServiceAdmin(AUFMetadataAdminMixin, admin.ModelAdmin):
    list_display = ('nom', 'date_modification', 'user_modification', 'actif', )
    fieldsets = AUFMetadataAdminMixin.fieldsets + ( 
        (None, {
            'fields': ('nom', ),
        }),
        )

class StatutAdmin(AUFMetadataAdminMixin, admin.ModelAdmin):
    list_display = ('code', 'nom', 'date_modification', 'user_modification', 'actif', )
    fieldsets = AUFMetadataAdminMixin.fieldsets + ( 
        (None, {
            'fields': ('code', 'nom', ),
        }),
        )

class TauxChangeAdmin(admin.ModelAdmin):
    list_display = ('taux', 'devise', 'annee', 'date_modification', 'user_modification', )
    list_filter = ('devise', )
    fieldsets = AUFMetadataAdminMixin.fieldsets + ( 
        (None, {
            'fields': ('taux', 'devise', 'annee', ),
        }),
        )

class TypeContratAdmin(admin.ModelAdmin):
    list_display = ('nom', 'nom_long', 'date_modification', 'user_modification', 'actif', )
    fieldsets = AUFMetadataAdminMixin.fieldsets + ( 
        (None, {
            'fields': ('nom', 'nom_long', ),
        }),
        )


class TypePosteAdmin(AUFMetadataAdminMixin, admin.ModelAdmin):
    search_fields = ('nom', 'nom_feminin', )
    list_display = ('nom', 'famille_emploi', 'date_modification', 'user_modification', 'actif', )
    list_filter = ('famille_emploi', )
    fieldsets = AUFMetadataAdminMixin.fieldsets + ( 
        (None, {
            'fields': ('nom', 'nom_feminin', 'is_responsable', 'famille_emploi',  )
        }),
        )


class TypeRemunerationAdmin(AUFMetadataAdminMixin, admin.ModelAdmin):
    list_display = ('nom', 'type_paiement', 'nature_remuneration', 'date_modification', 'user_modification', 'actif', )
    fieldsets = AUFMetadataAdminMixin.fieldsets + ( 
        (None, {
            'fields': ('nom', 'type_paiement', 'nature_remuneration', )
        }),
        )


class TypeRevalorisationAdmin(AUFMetadataAdminMixin, admin.ModelAdmin):
    list_display = ('nom', 'date_modification', 'user_modification', 'actif', )
    fieldsets = AUFMetadataAdminMixin.fieldsets + ( 
        (None, {
            'fields': ('nom', )
        }),
        )


class ValeurPointAdmin(AUFMetadataAdminMixin, admin.ModelAdmin):
    list_display = ('_devise_code', '_devise_nom', 'annee', 'valeur',  'date_modification', 'user_modification', )
    list_filter = ('annee', 'devise', )
    fieldsets = AUFMetadataAdminMixin.fieldsets + ( 
        (None, {
            'fields': ('valeur', 'devise', 'implantation', 'annee', )
        }),
        )

    def _devise_code(self, obj):
        return obj.devise.code
    _devise_code.short_description = "Code de la devise"

    def _devise_nom(self, obj):
        return obj.devise.nom
    _devise_nom.short_description = "Nom de la devise"


admin.site.register(rh.Classement, ClassementAdmin)
admin.site.register(rh.Devise, DeviseAdmin)
admin.site.register(rh.Dossier, DossierAdmin)
admin.site.register(rh.Employe, EmployeAdmin)
admin.site.register(rh.FamilleEmploi, FamilleEmploiAdmin)
admin.site.register(rh.OrganismeBstg, OrganismeBstgAdmin)
admin.site.register(rh.Poste, PosteAdmin)
admin.site.register(rh.ResponsableImplantation, ResponsableImplantationAdmin)
admin.site.register(rh.Service, ServiceAdmin)
admin.site.register(rh.Statut, StatutAdmin)
admin.site.register(rh.TauxChange, TauxChangeAdmin)
admin.site.register(rh.TypeContrat, TypeContratAdmin)
admin.site.register(rh.TypePoste, TypePosteAdmin)
admin.site.register(rh.TypeRemuneration, TypeRemunerationAdmin)
admin.site.register(rh.TypeRevalorisation, TypeRevalorisationAdmin)
admin.site.register(rh.ValeurPoint, ValeurPointAdmin)