[#2372] Réorganisation champs Dossier

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

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

from collections import defaultdict
import datetime

from django.db import models
from django.contrib import admin
from django.conf import settings
from django.db.models import Q
from auf.django.metadata.admin import AUFMetadataAdminMixin, AUFMetadataInlineAdminMixin, AUF_METADATA_READONLY_FIELDS
from project.rh import models as rh
from forms import DossierForm, ContratForm
from dae.utils import get_employe_from_user


class ProtectRegionMixin(object):

    def queryset(self, request):
        qs = super(ProtectRegionMixin, self).queryset(request)

        if request.user.is_superuser:
            return qs

        employe = get_employe_from_user(request.user)

        q = Q(**{self.model.prefix_implantation: employe.implantation.region})
        qs = qs.filter(q).distinct()
        return qs

    def has_change_permission(self, request, obj=None):
        if request.user.is_superuser:
            return True

        if obj:
            employe = get_employe_from_user(request.user)
            if employe.implantation.region in obj.get_regions():
                return True
            else:
                return False

        return True


# 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 = models.Model  # à remplacer dans admin.py
    extra = 0


class AyantDroitCommentaireInline(AUFMetadataInlineAdminMixin, admin.TabularInline):
    readonly_fields = ('owner', )
    model = models.Model  # à remplacer dans admin.py
    extra = 1


class ContratInline(AUFMetadataInlineAdminMixin, admin.TabularInline):
    form = ContratForm
    model = models.Model  # à remplacer dans admin.py
    extra = 1


class DossierROInline(ReadOnlyInlineMixin, admin.TabularInline):
    exclude = AUF_METADATA_READONLY_FIELDS
    model = models.Model  # à remplacer dans admin.py


class DossierCommentaireInline(AUFMetadataInlineAdminMixin, admin.TabularInline):
    readonly_fields = ('owner', )
    model = models.Model  # à remplacer dans admin.py
    extra = 1


class DossierPieceInline(admin.TabularInline):
    model = models.Model  # à remplacer dans admin.py
    extra = 4


class EmployeInline(admin.TabularInline):
    model = models.Model  # à remplacer dans admin.py

class EmployeCommentaireInline(AUFMetadataInlineAdminMixin, admin.TabularInline):
    readonly_fields = ('owner', )
    model = models.Model  # à remplacer dans admin.py
    extra = 1


class EmployePieceInline(admin.TabularInline):
    model = models.Model  # à remplacer dans admin.py
    extra = 4


class EvenementInline(AUFMetadataInlineAdminMixin, admin.TabularInline):
    model = models.Model  # à remplacer dans admin.py
    extra = 1


class EvenementRemunerationInline(AUFMetadataInlineAdminMixin, admin.TabularInline):
    model = models.Model  # à remplacer dans admin.py
    extra = 1


class PosteCommentaireInline(AUFMetadataInlineAdminMixin, admin.TabularInline):
    readonly_fields = ('owner', )
    model = models.Model  # à remplacer dans admin.py
    extra = 1


class PosteFinancementInline(admin.TabularInline):
    model = models.Model  # à remplacer dans admin.py


class PostePieceInline(admin.TabularInline):
    model = models.Model  # à remplacer dans admin.py


class RemunerationInline(AUFMetadataInlineAdminMixin, admin.TabularInline):
    model = models.Model  # à remplacer dans admin.py
    extra = 1


class RemunerationROInline(ReadOnlyInlineMixin, RemunerationInline):
    pass


class TypePosteInline(AUFMetadataInlineAdminMixin, admin.TabularInline):
    model = models.Model  # à remplacer dans admin.py


# Admins

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', 'nationalite', 'date_naissance', 'genre', )
        }),
    )

    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):
    fieldsets = AUFMetadataAdminMixin.fieldsets + (
        (None, {
            'fields': ('type', 'echelon', 'degre', 'coefficient',  )
        }),
    )


class CommentaireAdmin(admin.ModelAdmin):
    pass


#class ContratAdmin(AUFMetadataAdminMixin, admin.ModelAdmin):
#    form = ContratForm
#    alphabet_filter = 'dossier__employe__nom'
#    search_fields = ('dossier__employe__nom', 'dossier__employe__prenom', 'dossier__poste__nom', 'dossier__poste__nom_feminin', )
#    list_display = ('id', '_employe', '_poste', 'date_debut', 'date_fin', '_implantation', )
#    fieldsets = AUFMetadataAdminMixin.fieldsets + (
#        (None, {
#            'fields': ('dossier', 'type_contrat', 'date_debut', 'date_fin', )
#        }),
#    )
#
#    def lookup_allowed(self, key, value):
#        if key in ('dossier__employe__nom__istartswith',  ):
#            return True
#
#    def _employe(self, obj):
#        return unicode(obj.dossier.employe)
#    _employe.short_description = "Employé"
#
#    def _poste(self, obj):
#        return obj.dossier.poste.nom
#    _poste.short_description = "Poste"
#
#    def _implantation(self, obj):
#        return obj.dossier.poste.implantation
#    _poste.short_description = "Implantation"

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


class DossierAdmin(AUFMetadataAdminMixin, ProtectRegionMixin, admin.ModelAdmin,):
    form = DossierForm
    alphabet_filter = 'employe__nom'
    search_fields = ('employe__nom', 'employe__prenom', 'poste__nom', 'poste__nom_feminin')
    list_display = ('_employe', '_poste', 'date_debut', 'date_fin', 'date_modification', '_actif')
    inlines = (DossierPieceInline, ContratInline,
               RemunerationInline,
               #EvenementInline,
               DossierCommentaireInline,
              )
    fieldsets = AUFMetadataAdminMixin.fieldsets + (
        ('Identification', {
            'fields': ('employe', 'poste', 'statut', 'organisme_bstg',)
        }),
        ('Recrutement', {
            'fields': (('remplacement', 'statut_residence'), )
        }),
        ('Rémunération', {
            'fields': ('classement', ('regime_travail', 'regime_travail_nb_heure_semaine'),)
        }),
        ('Occupation du Poste par cet Employe', {
            'fields': (('date_debut', 'date_fin'), )
        }),
    )

    def lookup_allowed(self, key, value):
        if key in ('employe__nom__istartswith', 'actif__exact', ):
            return True

    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 = u'Employé actif'
    _actif.short_description = u'Employé actif'
    _actif.admin_order_field = 'employe__actif'

    def _poste(self, dossier):
        return unicode(dossier.poste.nom)
    _poste.short_description = u'Poste'
    _poste.admin_order_field = 'poste__nom'

    def _employe(self, dossier):
        return unicode(dossier.employe)
    _employe.short_description = u'Employé'
    _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()

    def render_change_form(self, request, context, *args, **kwargs):
        obj = kwargs['obj']

        thisyear = datetime.date.today().year
        thisyearfilter = Q(date_debut__year=thisyear) | Q(date_fin__year=thisyear)

        remunnow = obj.rh_remuneration_remunerations.filter(thisyearfilter)

        remun_sum = 0
        remun_sum_euro = 0
        sums = defaultdict(int)
        sums_euro = defaultdict(int)
        for r in remunnow:
            nature = r.type.nature_remuneration
            sums[nature] += r.montant
            sums_euro[nature] += r.montant_euro()
            remun_sum += r.montant
            remun_sum_euro += r.montant_euro()

        remun = {}
        sums = dict(sums)
        for n, s in sums.iteritems():
            remun[n] = [sums[n], sums_euro[n]]

        extra = {
            'remun': remun,
            'remun_sum': remun_sum,
            'remun_sum_euro': remun_sum_euro,
        }

        context.update(extra)

        return super(DossierAdmin, self).render_change_form(request, context, *args, **kwargs)


class DossierPieceAdmin(admin.ModelAdmin):
    pass


class DossierCommentaireAdmin(admin.ModelAdmin):
    pass


class EmployeAdmin(AUFMetadataAdminMixin, ProtectRegionMixin, admin.ModelAdmin):
    alphabet_filter = 'nom'
    search_fields = ('id', 'nom', 'prenom', 'nom_affichage', 'actif', )
    list_filter = ('actif', )
    ordering = ('nom', )
    actions = ('desactiver', )
    list_display = ('nom', 'prenom', 'actif', )
    inlines = (AyantDroitInline,
               DossierROInline,
               EmployePieceInline,
               EmployeCommentaireInline)
    fieldsets = AUFMetadataAdminMixin.fieldsets + ( 
        ('Identification', {
            'fields': ('nom', 'prenom', 'nom_affichage', 'nationalite', 'date_naissance', 'genre', )
        }),
        ('Informations personnelles', {
            'fields': ('situation_famille', 'date_entree', )
        }),
        ('Coordonnées', {
            'fields': ('tel_domicile', 'tel_cellulaire', 'adresse', 'ville', 'province', 'code_postal', 'pays', )
        }),
        )

    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 EvenementAdmin(admin.ModelAdmin):
    inlines = (EvenementRemunerationInline,)


class EvenementRemunerationAdmin(admin.ModelAdmin):
    pass


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


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


class PosteAdmin(AUFMetadataAdminMixin, ProtectRegionMixin, admin.ModelAdmin):
    alphabet_filter = 'nom'
    search_fields = ('nom', 'implantation__code', 'implantation__nom', 'implantation__region__code', 'implantation__region__nom', )
    list_display = ('nom', 'implantation', 'service', 'type_poste', 'date_debut', 'date_fin', )
    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', 'classement_max', 'valeur_point_min',
                       'valeur_point_max', 'devise_min', 'devise_max',
                       'salaire_min', 'salaire_max', 'indemn_min',
                       'indemn_max', 'autre_min', '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_validation', 'date_debut', 'date_fin')
        }),
    )

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

    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):
    pass


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', 'actif', )
    fieldsets = AUFMetadataAdminMixin.fieldsets + ( 
        (None, {
            'fields': ('nom', ),
        }),
        )

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

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

class TypeContratAdmin(admin.ModelAdmin):
    inlines = (ContratInline,)


class TypePosteAdmin(AUFMetadataAdminMixin, admin.ModelAdmin):
    search_fields = ('nom', 'nom_feminin', )
    list_display = ('nom', 'famille_emploi', )
    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', )
    #inlines = (RemunerationROInline,) utilité?
    fieldsets = AUFMetadataAdminMixin.fieldsets + ( 
        (None, {
            'fields': ('nom', 'type_paiement', 'nature_remuneration', )
        }),
        )


class TypeRevalorisationAdmin(AUFMetadataAdminMixin, admin.ModelAdmin):
    #inlines = (RemunerationROInline,) utilité?
    fieldsets = AUFMetadataAdminMixin.fieldsets + ( 
        (None, {
            'fields': ('nom', )
        }),
        )


class ValeurPointAdmin(AUFMetadataAdminMixin, admin.ModelAdmin):
    list_display = ('_devise_code', '_devise_nom', 'annee', 'valeur', )
    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"