[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 django.template.defaultfilters import date
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
from groups import grp_drh
import models as rh
import filters


# 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):
        user_groups = request.user.groups.all()

        # Lock pour autoriser uniquement les DRH à utiliser RH
        if not request.user.is_superuser and not grp_drh in user_groups:
            return False

        if len(user_groups) == 0 and not request.user.is_superuser:
            return False

        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 PosteComparaisonInline(AUFMetadataInlineAdminMixin, admin.TabularInline):
    model = rh.PosteComparaison


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', )
    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',
    )
    list_filter = (
        'poste__implantation__region',
        'poste__implantation',
        'poste__type_poste',
        'poste__type_poste__famille_emploi',
        'rh_contrats__type_contrat',
        'date_debut',
        'date_fin',
    )
    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',
            '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',
            'date_debut__gte',
            'date_debut__isnull',
            'date_fin__lte',
            'date_fin__isnull',
            ):
            return True


    def _id(self, d):
        apercu_link = u"""<a title="Aperçu du dossier" onclick="return showAddAnotherPopup(this);" href='%s'><img src="%simg/loupe.png" /></a>""" % \
                (reverse('dossier_apercu', args=(d.id,)),
                 settings.STATIC_URL,
                 )
        link = u"""%s&nbsp;<a href="%s" title="Modifier le dossier"><strong>%s</strong></a>""" % \
             (apercu_link,
             reverse('admin:rh_dossier_change', args=(d.id,)),
             d.id,
             )
        return link
    _id.allow_tags = True
    _id.short_description = u"Dossier__#"
    _id.admin_order_field = 'id'
    

    def _date_debut(self, obj):
        return date(obj.date_debut)

    _date_debut.short_description = u'Occupation début'
    _date_debut.admin_order_field = 'date_debut'

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

    def _poste(self, dossier):
        link = u"""<a title="Aperçu du poste" onclick="return showAddAnotherPopup(this);" href='%s'><img src="%simg/loupe.png" /></a> <a href="%s" title="Modifier le poste">%s</a>""" % \
                (reverse('poste_apercu', args=(dossier.poste.id,)),
                 settings.STATIC_URL,
                 reverse('admin:rh_poste_change', args=(dossier.poste.id,)),
                 dossier.poste,
                 )
        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,))

        style = ""
        view = u"""<a href="%s" title="Aperçu l'employé" onclick="return showAddAnotherPopup(this);"><img src="%simg/loupe.png" /></a>""" % (view_link, settings.STATIC_URL,)
        return u"""%s<a href='%s' style="%s;">[%s] %s %s</a>""" % \
        (view, edit_link, style, employe.id, employe.nom.upper(), employe.prenom.title())
    _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.date_creation = datetime.datetime.now()
            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 = ('_apercu', '_nom', '_dossiers', 'date_modification',  'user_modification', )
    list_filter = ('rh_dossiers__poste__implantation__region',  'rh_dossiers__poste__implantation', 'nb_postes', )
    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 _apercu(self, obj):
        return u"""<a title="Aperçu de l'employé" onclick="return showAddAnotherPopup(this);" href='%s'><img src="%simg/loupe.png" /></a>""" % \
                (reverse('employe_apercu', args=(obj.id,)), settings.STATIC_URL)
    _apercu.allow_tags = True
    _apercu.short_description = u""
    _apercu.admin_order_field = ""

    def _nom(self, obj):
        edit_link = reverse('admin:rh_employe_change', args=(obj.id,))
        return u"""<a href='%s'><strong>[%s] %s %s</strong></a>""" % \
        (edit_link, obj.id, obj.nom.upper(), obj.prenom.title(),)
    _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'):
            apercu = u"""<a title="Aperçu du dossier" href="%s" onclick="return showAddAnotherPopup(this);" title="Aperçu du dossier"><img src="%simg/loupe.png" /></a>""" % \
                (reverse('dossier_apercu', args=(d.id,)), settings.STATIC_URL,)
            link = u"""<li>%s<a href='%s'>%s : %s</a></li>""" % \
                (apercu,
                 reverse('admin:rh_dossier_change', args=(d.id,)),
                 d.date_debut.year,
                 d.poste,
                 )

            # Dossier terminé en gris non cliquable
            if d.date_fin is not None:
                link = u"""<li style="color: grey">%s : %s</li>""" % \
                    (d.date_debut.year,
                     d.poste,
                    )

            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.date_creation = datetime.datetime.now()
            instance.save()


class EmployeCommentaireAdmin(admin.ModelAdmin):
    pass


class EmployePieceAdmin(admin.ModelAdmin):
    pass


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


class OrganismeBstgAdmin(AUFMetadataAdminMixin, admin.ModelAdmin):
    search_fields = ('nom',)
    list_display = ('nom', 'type', 'pays', 'date_modification', 'user_modification', )
    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 = (
        '_apercu',
        '_nom', 
        '_occupe_par',
        'implantation', 
        '_service', 
        'date_debut', 
        'date_fin',
        'date_modification',
        'user_modification',
        )
    list_filter = (
        'service', 
        'implantation__region', 
        'implantation',
        'type_poste',
        'type_poste__famille_emploi',
        'date_debut',
        'date_fin',
        'vacant',
        )
    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 Méta-données', {
            'fields': ('date_debut', 'date_fin')
        }),
    )

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

    def lookup_allowed(self, key, value):
        if key in (
            'date_debut__gte',
            'date_debut__isnull',
            'date_fin__lte',
            'date_fin__isnull',
            ):
            return True

    
    def _apercu(self, poste):
        link = u"""<a onclick="return showAddAnotherPopup(this);" title="Aperçu du poste" href='%s'><img src="%simg/loupe.png" /> %s</a>""" % \
                (reverse('poste_apercu', args=(poste.id,)),
                 settings.STATIC_URL,
                 poste.id,
                 )
        return link
    _apercu.allow_tags = True
    _apercu.short_description = 'Poste __#'
    _apercu.admin_order_field = 'id'

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

    def _nom(self, poste):
        link = u"""<a href="%s" title="Modifier le poste"><strong>%s</strong></a>""" % \
                 (reverse('admin:rh_poste_change', args=(poste.id,)),
                 poste.nom,
                 )
        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.date_fin is not None and obj.date_fin < datetime.datetime.now():
            return u"s/o"
        employes = obj.occupe_par()
        if employes:
            l = []
            for e in employes:
                link = "<a href='%s' title='Aperçu de l\'employer' onclick='return showAddAnotherPopup(this)'><img src='%simg/loupe.png' /></a> <a href='%s'>%s</a>" % \
                     (reverse('employe_apercu', args=(e.id,)),
                     settings.STATIC_URL,
                     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.date_creation = datetime.datetime.now()
            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', )
    fieldsets = AUFMetadataAdminMixin.fieldsets + ( 
        (None, {
            'fields': ('nom', ),
        }),
        )

class StatutAdmin(AUFMetadataAdminMixin, admin.ModelAdmin):
    list_display = ('code', 'nom', 'date_modification', 'user_modification', )
    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', )
    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', )
    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', )
    fieldsets = AUFMetadataAdminMixin.fieldsets + ( 
        (None, {
            'fields': ('nom', 'type_paiement', 'nature_remuneration', )
        }),
        )


class TypeRevalorisationAdmin(AUFMetadataAdminMixin, admin.ModelAdmin):
    list_display = ('nom', 'date_modification', 'user_modification', )
    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)