Rapport contrats: ajout filtre date début
[auf_rh_dae.git] / project / rh / templatetags / rapports.py
CommitLineData
fd009814
PP
1# -*- coding: utf-8 -*-
2
72ac5e55
PP
3import datetime
4
fd009814
PP
5from django.utils.encoding import smart_unicode
6from django.template import Library
7from django.utils.http import urlencode
8
9from datamaster_modeles.models import Implantation, Region
23e749e1 10from rh.models import TypeContrat
fd009814
PP
11
12
13register = Library()
14
15
549830eb 16COMBLE_CHOICES = (('c', 'Comblé'), ('n', 'Vacant'))
7821915f
PP
17
18
19@register.inclusion_tag('admin/filter.html', takes_context=True)
20def filter_comble(context):
21 return {'title': 'comblé',
22 'choices': prepare_choices(COMBLE_CHOICES, 'comble', context)}
23
24
fd009814
PP
25@register.inclusion_tag('admin/filter_select.html', takes_context=True)
26def filter_region(context):
27 return {'title': u"région",
28 'choices': prepare_choices(Region.objects.values_list('id', 'nom'), 'implantation__region', context, remove=['pays', 'nord_sud'])}
29
30
31@register.inclusion_tag('admin/filter_select.html', takes_context=True)
32def filter_implantation(context):
33 return {'title': u"implantation",
34 'choices': prepare_choices(Implantation.objects.values_list('id', 'nom'), 'implantation', context)}
35
36
23e749e1
PP
37@register.inclusion_tag('admin/filter_select.html', takes_context=True)
38def filter_region_contrat(context):
39 return {'title': u"région",
40 'choices': prepare_choices(Region.objects.values_list('id', 'nom'), 'dossier__poste__implantation__region', context, remove=['pays', 'nord_sud'])}
41
42
43@register.inclusion_tag('admin/filter_select.html', takes_context=True)
44def filter_implantation_contrat(context):
45 return {'title': u"implantation",
46 'choices': prepare_choices(Implantation.objects.values_list('id', 'nom'), 'dossier__poste__implantation', context)}
47
48
49@register.inclusion_tag('admin/filter_select.html', takes_context=True)
50def filter_type_contrat(context):
51 return {'title': u"type de contrat",
52 'choices': prepare_choices(TypeContrat.objects.values_list('id', 'nom'), 'type_contrat', context)}
53
72ac5e55
PP
54@register.inclusion_tag('admin/filter_select.html', takes_context=True)
55def filter_echeance_contrat(context):
eb204143
JPC
56 today = datetime.date.today()
57 three_months = today + datetime.timedelta(days=3*30)
58 six_months = today + datetime.timedelta(days=6*30)
59 twelve_months = today + datetime.timedelta(days=12*30)
60
61 field_name = 'date_fin'
72ac5e55 62 return {'title': u"échéance",
eb204143
JPC
63 'choices': prepare_choices_date(field_name, context, links=(
64 ('Tous', {}),
65 ('moins de 3 mois', {'%s__gte' % field_name: today.strftime('%Y-%m-%d'),
66 '%s__lte' % field_name: three_months.strftime('%Y-%m-%d')}),
67 ('3 à 6 mois', {'%s__gte' % field_name: three_months.strftime('%Y-%m-%d'),
68 '%s__lte' % field_name: six_months.strftime('%Y-%m-%d')}),
69 ('6 à 12 mois', {'%s__gte' % field_name: six_months.strftime('%Y-%m-%d'),
70 '%s__lte' % field_name: twelve_months.strftime('%Y-%m-%d')}),
71 )
72 )}
73
74@register.inclusion_tag('admin/filter_select.html', takes_context=True)
75def filter_debut_contrat(context):
76 year = datetime.date.today().timetuple()[0]
77 this_year = datetime.date(year, 1, 1)
78 next_year = datetime.date(year + 1, 1, 1)
79 last_year = datetime.date(year - 1, 12,31)
80
81 field_name = 'date_debut'
82 return {'title': u"date début",
83 'choices': prepare_choices_date(field_name, context, links=(
84 ('Tous', {}),
85 ('anées à venirs', {'%s__gte' % field_name: next_year.strftime('%Y-%m-%d')}),
86 ('cette anneée', {'%s__gte' % field_name: this_year.strftime('%Y-%m-%d'),
87 '%s__lt' % field_name: next_year.strftime('%Y-%m-%d')}),
88 ('ans passés', {'%s__lte' % field_name: last_year.strftime('%Y-%m-%d')}),
89 )
90 )}
91
72ac5e55
PP
92
93
af5073aa
PP
94@register.inclusion_tag('admin/filter_select.html', takes_context=True)
95def filter_region_remun(context):
96 return {'title': u"région",
97 'choices': prepare_choices(Region.objects.values_list('id', 'nom'), 'dossiers__poste__implantation__region', context, remove=['pays', 'nord_sud'])}
98
99
100@register.inclusion_tag('admin/filter_select.html', takes_context=True)
101def filter_implantation_remun(context):
102 return {'title': u"implantation",
103 'choices': prepare_choices(Implantation.objects.values_list('id', 'nom'), 'dossiers__poste__implantation', context)}
104
105
fd009814
PP
106def get_query_string(request, new_params=None, remove=None):
107 if new_params is None: new_params = {}
108 if remove is None: remove = []
109 p = dict(request.GET.items())
110 for r in remove:
111 for k in p.keys():
112 if k.startswith(r):
113 del p[k]
114 for k, v in new_params.items():
115 if v is None:
116 if k in p:
117 del p[k]
118 else:
119 p[k] = v
120 return '?%s' % urlencode(p)
121
122
123def prepare_choices(choices, query_param, context, remove=[]):
124 request = context['request']
125 query_val = request.GET.get(query_param)
126 result = [{'selected': query_val is None,
127 'query_string': get_query_string(request, {}, [query_param] + remove),
128 'display': 'Tout'}]
129 for k, v in choices:
130 result.append({'selected': smart_unicode(k) == query_val,
131 'query_string': get_query_string(request, {query_param: k}, remove),
132 'display': v})
133 return result
72ac5e55
PP
134
135
eb204143 136def prepare_choices_date(field_name, context, links, remove=[]):
72ac5e55
PP
137 request = context['request']
138 params = request.GET
139 field_generic = '%s__' % field_name
140 date_params = dict([(k, v) for k, v in params.items() if k.startswith(field_generic)])
141
142
eb204143 143
72ac5e55
PP
144 result = []
145 for title, param_dict in links:
146 result.append({'selected': date_params == param_dict,
147 'query_string': get_query_string(request, param_dict, [field_generic]),
148 'display': title})
149 return result