Utilise date.today() pour comparaison pour chercher dossier principal

[auf_rh_dae.git] / project / dae / templatetags / dae.py

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

import os
import datetime

from django import template
from django.db.models import Q
# -*- coding: utf-8 -*-

from project import groups

from project.dae.workflow import ETATS_EDITABLE
from project.dae.forms import (
    remun_formset_factory_factory,
    ReadOnlyRemunFormSet,
    PosteCompReadOnlyRemunFormSet,
    DossierCompReadOnlyRemunFormSet,
    )
from project.dae import models as dae
from project.rh import models as rh


register = template.Library()


@register.filter
def test_membre_drh(user):
    grps = [g.name for g in user.groups.all()]
    if groups.DRH_NIVEAU_1 in grps or groups.DRH_NIVEAU_2 in grps:
        return True
    else:
        return False


@register.filter
def peut_ajouter(user):
    grp_ok = (groups.ADMINISTRATEURS,
            groups.CORRESPONDANT_RH,
            groups.DRH_NIVEAU_1,
            groups.DRH_NIVEAU_2)
    for g in [g.name for g in user.groups.all()]:
        if g in grp_ok:
            return True
    return False


@register.filter
def est_editable(obj, user):
    klass = obj.__class__
    groupes_users = [g.name for g in user.groups.all()]
    if obj.etat in ETATS_EDITABLE and \
       (obj in klass.objects.mes_choses_a_faire(user).all() or \
            groups.DRH_NIVEAU_1 in groupes_users or \
            groups.DRH_NIVEAU_2 in groupes_users):
        return True
    else:
        return False


@register.filter
def basename(path):
    return os.path.basename(path)


@register.inclusion_tag('dae/sort_header.html', takes_context=True)
def sort_header(context, field, title):
    """
    Génère une entête qu'on peut cliquer pour trier
    la colonne correspondante dans une table.
    """
    qs = context['request'].GET.copy()
    current = qs.get('tri', None)
    if current == field:
        cls = 'header headerSortUp'
        qs['tri'] = '-' + field
    elif current == '-' + field:
        cls = 'header headerSortDown'
        qs['tri'] = field
    else:
        cls = 'header'
        qs['tri'] = field
    return {'title': title, 'qs': qs.urlencode(), 'cls': cls}


@register.inclusion_tag('dae/pagination.html', takes_context=True)
def pagination(context, page):
    """Génère la navigation permettant de se promener de page en page."""
    qs = context['request'].GET
    previous_qs = qs.copy()
    previous_qs['page'] = page.previous_page_number()
    next_qs = qs.copy()
    next_qs['page'] = page.next_page_number()
    return {'page': page,
            'previous_qs': previous_qs.urlencode(),
            'next_qs': next_qs.urlencode()
            }


@register.inclusion_tag('dae/embauche-remun.html', takes_context=True)
def remun_form(context, dossier):

    return {
        'annee_remun': None,
        'remunForm': ReadOnlyRemunFormSet(instance=dossier)
        }


@register.inclusion_tag('dae/embauche-remun.html', takes_context=True)
def rh_remun_form(context, dossier):

    return {
        'annee_remun': None,
        'remunForm': RHReadOnlyRemunFormSet(instance=dossier)
        }


@register.inclusion_tag('dae/embauche-remun.html', takes_context=True)
def poste_cmp_remun_form(context, poste_cmp):
    return {
        'annee_remun': None,
        'remunForm': PosteCompReadOnlyRemunFormSet(instance=poste_cmp)
        }


@register.inclusion_tag('dae/embauche-remun.html', takes_context=True)
def dossier_cmp_remun_form(context, dossier_cmp):
    return {
        'annee_remun': None,
        'remunForm': DossierCompReadOnlyRemunFormSet(instance=dossier_cmp)
        }


@register.inclusion_tag('dae/embauche-remun.html', takes_context=True)
def rh_remun_form_for_year(context, dossier, year):

    fs = remun_formset_factory_factory(
        read_only=True,
        parent_model=rh.Dossier,
        model=rh.Remuneration,
        )

    def _get_qs(inst):
        year_start = datetime.date(year, 1, 1)
        year_end = datetime.date(year, 12, 31)
        qs = inst.model.objects.filter(
            Q(dossier=inst.instance) & (
                Q(date_debut__lte=year_end, date_fin__gte=year_start) |
                Q(date_debut__lte=year_end, date_fin__isnull=True) |
                Q(date_debut__isnull=True, date_fin__gte=year_start) |
                Q(date_debut__isnull=True, date_fin__isnull=True)
                ))
        return qs
        
    fs.get_queryset = _get_qs

    return {
        'annee_remun': year,
        'remunForm': fs(instance=dossier)
        }


@register.simple_tag
def remun_ajuste(remun, annee=None):
    return '%.2f' % remun.montant_ajuste_euros(annee)