Retirer debug

[auf_rh_dae.git] / project / rh / templatetags / rapports.py

# -*- coding: utf-8 -*-

import datetime

from django.utils.encoding import smart_unicode
from django.template import Library
from django.utils.http import urlencode

from datamaster_modeles.models import Implantation, Region
from rh.models import TypeContrat


register = Library()


COMBLE_CHOICES = (('c', 'Comblé'), ('n', 'Non-comblé'))


@register.inclusion_tag('admin/filter.html', takes_context=True)
def filter_comble(context):
    return {'title': 'comblé',
            'choices': prepare_choices(COMBLE_CHOICES, 'comble', context)}


@register.inclusion_tag('admin/filter_select.html', takes_context=True)
def filter_region(context):
    return {'title': u"région",
            'choices': prepare_choices(Region.objects.values_list('id', 'nom'), 'implantation__region', context, remove=['pays', 'nord_sud'])}


@register.inclusion_tag('admin/filter_select.html', takes_context=True)
def filter_implantation(context):
    return {'title': u"implantation",
            'choices': prepare_choices(Implantation.objects.values_list('id', 'nom'), 'implantation', context)}


@register.inclusion_tag('admin/filter_select.html', takes_context=True)
def filter_region_contrat(context):
    return {'title': u"région",
            'choices': prepare_choices(Region.objects.values_list('id', 'nom'), 'dossier__poste__implantation__region', context, remove=['pays', 'nord_sud'])}


@register.inclusion_tag('admin/filter_select.html', takes_context=True)
def filter_implantation_contrat(context):
    return {'title': u"implantation",
            'choices': prepare_choices(Implantation.objects.values_list('id', 'nom'), 'dossier__poste__implantation', context)}


@register.inclusion_tag('admin/filter_select.html', takes_context=True)
def filter_type_contrat(context):
    return {'title': u"type de contrat",
            'choices': prepare_choices(TypeContrat.objects.values_list('id', 'nom'), 'type_contrat', context)}


@register.inclusion_tag('admin/filter_select.html', takes_context=True)
def filter_echeance_contrat(context):
    return {'title': u"échéance",
            'choices': prepare_choices_date('date_fin', context)}


def get_query_string(request, new_params=None, remove=None):
    if new_params is None: new_params = {}
    if remove is None: remove = []
    p = dict(request.GET.items())
    for r in remove:
        for k in p.keys():
            if k.startswith(r):
                del p[k]
    for k, v in new_params.items():
        if v is None:
            if k in p:
                del p[k]
        else:
            p[k] = v
    return '?%s' % urlencode(p)


def prepare_choices(choices, query_param, context, remove=[]):
    request = context['request']
    query_val = request.GET.get(query_param)
    result = [{'selected': query_val is None,
               'query_string': get_query_string(request, {}, [query_param] + remove),
               'display': 'Tout'}]
    for k, v in choices:
        result.append({'selected': smart_unicode(k) == query_val,
                       'query_string': get_query_string(request, {query_param: k}, remove),
                       'display': v})
    return result


def prepare_choices_date(field_name, context, remove=[]):
    request = context['request']
    params = request.GET
    field_generic = '%s__' % field_name
    date_params = dict([(k, v) for k, v in params.items() if k.startswith(field_generic)])


    today = datetime.date.today()
    three_months = today + datetime.timedelta(days=3*30)
    six_months = today + datetime.timedelta(days=6*30)
    twelve_months = today + datetime.timedelta(days=12*30)

    links = (
        ('Tous', {}),
        ('moins de 3 mois', {'%s__gte' % field_name: today.strftime('%Y-%m-%d'),
                             '%s__lte' % field_name: three_months.strftime('%Y-%m-%d')}),
        ('3 à 6 mois', {'%s__gte' % field_name: three_months.strftime('%Y-%m-%d'),
                        '%s__lte' % field_name: six_months.strftime('%Y-%m-%d')}),
        ('6 à 12 mois', {'%s__gte' % field_name: six_months.strftime('%Y-%m-%d'),
                         '%s__lte' % field_name: twelve_months.strftime('%Y-%m-%d')}),
    )

    result = []
    for title, param_dict in links:
        result.append({'selected': date_params == param_dict,
                       'query_string': get_query_string(request, param_dict, [field_generic]),
                       'display': title})
    return result