fix

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

from django.template import Library
from django.db import connection
from django import forms

register = Library()

@register.simple_tag
def query_string_builder(cl, name, value):
    return cl.get_query_string({name: value})

@register.simple_tag
def add_selected(cl, key, value):
    return 'selected="selected" ' if key in cl.params and cl.params[key] == value else ''

@register.inclusion_tag('admin/rh/annee_select.html')
def recherche_par_annees(cl):
    cursor = connection.cursor()
    cursor.execute("SELECT year(date_debut) FROM rh_dossier WHERE year(date_debut) IS NOT NULL GROUP BY year(date_debut)")
    set_annees = set(row[0] for row in cursor.fetchall())
    cursor.execute("SELECT year(date_fin) FROM rh_dossier WHERE year(date_fin) IS NOT NULL GROUP BY year(date_fin)")
    for row in cursor.fetchall():
        set_annees.add(row[0])
    list_annees = list(set_annees)
    list_annees.insert(0, '')

    class RechercheTemporelle(forms.Form):
        periode = forms.ChoiceField(
                choices=((cl.get_query_string({'periode': p }, ('annee', 'date_debut', 'date_fin')), p) for p in cl.PERIODE_CHOICE),
                    widget=forms.Select(attrs = {
                        'onchange' : """window.location=window.location.pathname+this.options[this.selectedIndex].value""",
                        }))
        annee = forms.ChoiceField(choices=((cl.get_query_string({'annee': a}, ('periode', 'date_debut', 'date_fin')), a) for a in list_annees),
                    widget=forms.Select(attrs = {
                        'onchange' : """window.location=window.location.pathname+this.options[this.selectedIndex].value""",
                        }))
        date_debut = forms.DateField()
        date_fin = forms.DateField()
    params = cl.params
    if 'periode' in params:
        params['periode'] = cl.get_query_string({'periode': params['periode']})
    if 'annee' in params:
        params['annee'] = cl.get_query_string({'annee': params['annee']})
    f = RechercheTemporelle(params)
    return {
            'form': f,
            'plage_date_querystring': cl.get_query_string(remove=('annee', 'periode'))
            }