nice filter

[auf_rh_dae.git] / project / rh / lib.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 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



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

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


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, LinkedInline):
    exclude = AUF_METADATA_READONLY_FIELDS
    model = models.Model  # à remplacer dans admin.py
    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 = 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', '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):
    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 + (
        (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'), )
        }),
    )

    add_form_template = 'admin/change_form.html'

    def queryset(self, request):
        return self.model.actifs.all()

    class Media:
        js = ('js/dossier.js',)

    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.get('obj', None)

        if not obj:
            return super(DossierAdmin, self).render_change_form(request, context, *args, **kwargs)


        remun, remun_sum, remun_sum_euro = calc_remun(obj)

        extra = {
            'remun': remun,
            'remun_sum': remun_sum,
            'remun_sum_euro': remun_sum_euro,
            'employe': obj.employe,
            'poste_nom': obj.poste.nom,
            'poste_service': obj.poste.service,
            'poste_implantation': obj.poste.implantation,
        }

        context.update(extra)

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


class DossierInactifAdmin(DossierAdmin):
    def queryset(self, request):
        return self.model.inactifs.all()


class DossierPieceAdmin(admin.ModelAdmin):
    pass


class DossierCommentaireAdmin(admin.ModelAdmin):
    pass


class EmployeAdminForm(forms.ModelForm):
    class Meta:
        model = rh.Employe

    def __init__(self, *args, **kwargs):
        super(EmployeAdminForm, self).__init__(*args, **kwargs)
        self.fields['date_naissance'].widget = forms.widgets.DateInput()


class EmployeAdmin(AUFMetadataAdminMixin, ProtectRegionMixin, admin.ModelAdmin):
    alphabet_filter = 'nom'
    search_fields = ('id', 'nom', 'prenom', 'nom_affichage', )
    ordering = ('nom', )
    actions = ('desactiver', )
    form = EmployeAdminForm
    list_display = ('id', 'nom', 'prenom', '_dossiers', 'date_modification',  'user_modification',)
    list_display_links = ('id', 'nom',)
    list_filter = ('dossiers__poste__implantation__region',  'dossiers__poste__implantation',)
    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', )
        }),
        )

    add_form_template = 'admin/change_form.html'

    def _dossiers(self, obj):
        l = []
        for d in obj.dossiers.all().order_by('-date_debut'):
            link = "<li><a href='%s'>%s : %s</a></li>" % \
                (reverse('admin:rh_dossier_change', args=(d.id,)),
                 d.date_debut.year,
                 d.poste)
            l.append(link)
        return "<ul>%s</ul>" % "\n".join(l)
    _dossiers.allow_tags = True

    def queryset(self, request):
        qs = super(EmployeAdmin, self).queryset(request) 
        return qs.filter(actif=True).select_related(depth=1)

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

    def render_change_form(self, request, context, *args, **kwargs):
        obj = kwargs.get('obj', None)

        if not obj:
            return super(EmployeAdmin, self).render_change_form(request, context, *args, **kwargs)

        remun = {}
        remun_sum = 0
        remun_sum_euro = 0
        dossiers = obj.dossiers.all()

        for dossier in dossiers:
            this_remun, this_remun_sum, this_remun_sum_euro = calc_remun(dossier)

            for item in this_remun:
                if item not in remun:
                    remun[item] = this_remun[item]
                else:
                    remun[item][0] += this_remun[item][0]
                    remun[item][1] += this_remun[item][1]

            remun_sum += this_remun_sum
            remun_sum_euro += this_remun_sum_euro

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

        context.update(extra)

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


class EmployeInactifAdmin(EmployeAdmin):
    def queryset(self, request):
        return self.model.inactifs.all()


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


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"


def calc_remun(dossier):
        thisyear = datetime.date.today().year
        thisyearfilter = Q(date_debut__year=thisyear) | Q(date_fin__year=thisyear)

        remunnow = dossier.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]]

        return remun, remun_sum, remun_sum_euro