8ea7d486c2ae38ae74eb6b52462be3082c72d6f5
1 # -*- coding: utf-8 -*-
5 from django
.utils
.encoding
import smart_unicode
6 from django
.template
import Library
7 from django
.utils
.http
import urlencode
9 from datamaster_modeles
.models
import Implantation
, Region
10 from rh
.models
import TypeContrat
16 COMBLE_CHOICES
= (('c', 'Comblé'), ('n', 'Vacant'))
19 @register.inclusion_tag('admin/filter.html', takes_context
=True)
20 def filter_comble(context
):
21 return {'title': 'comblé',
22 'choices': prepare_choices(COMBLE_CHOICES
, 'comble', context
)}
25 @register.inclusion_tag('admin/filter_select.html', takes_context
=True)
26 def 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'])}
31 @register.inclusion_tag('admin/filter_select.html', takes_context
=True)
32 def filter_implantation(context
):
33 return {'title': u
"implantation",
34 'choices': prepare_choices(Implantation
.objects
.values_list('id', 'nom'), 'implantation', context
)}
37 @register.inclusion_tag('admin/filter_select.html', takes_context
=True)
38 def 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'])}
43 @register.inclusion_tag('admin/filter_select.html', takes_context
=True)
44 def filter_implantation_contrat(context
):
45 return {'title': u
"implantation",
46 'choices': prepare_choices(Implantation
.objects
.values_list('id', 'nom'), 'dossier__poste__implantation', context
)}
49 @register.inclusion_tag('admin/filter_select.html', takes_context
=True)
50 def filter_type_contrat(context
):
51 return {'title': u
"type de contrat",
52 'choices': prepare_choices(TypeContrat
.objects
.values_list('id', 'nom'), 'type_contrat', context
)}
54 @register.inclusion_tag('admin/filter_select.html', takes_context
=True)
55 def filter_echeance_contrat(context
):
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)
61 field_name
= 'date_fin'
62 return {'title': u
"échéance",
63 'choices': prepare_choices_date(field_name
, context
, links
=(
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')}),
74 @register.inclusion_tag('admin/filter_select.html', takes_context
=True)
75 def 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)
81 field_name
= 'date_debut'
82 return {'title': u
"date début",
83 'choices': prepare_choices_date(field_name
, context
, links
=(
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')}),
92 @register.inclusion_tag('admin/filter_select.html', takes_context
=True)
93 def filter_a_venir(context
):
94 today
= datetime
.date
.today()
95 year
, month
, day
= datetime
.date
.today().timetuple()[:3]
96 mois_prochain
= datetime
.date(year
+((month
+1)/13), (month
+1)%12, 1)
97 this_month
= datetime
.date(year
, month
, 1)
99 field_name
= 'date_debut'
100 return {'title': u
"à venir",
101 'choices': prepare_choices_date(field_name
, context
, links
=(
103 ('à venir', {'%s__gt' % field_name
: today
.strftime('%Y-%m-%d')}),
104 ('à venir mois prochain', {'%s__gte' % field_name
: mois_prochain
.strftime('%Y-%m-%d')}),
105 ('à venir ce mois', {'%s__gte' % field_name
: this_month
.strftime('%Y-%m-%d'),
106 '%s__lt' % field_name
: mois_prochain
.strftime('%Y-%m-%d')}),
113 @register.inclusion_tag('admin/filter_select.html', takes_context
=True)
114 def filter_region_remun(context
):
115 return {'title': u
"région",
116 'choices': prepare_choices(Region
.objects
.values_list('id', 'nom'), 'dossiers__poste__implantation__region', context
, remove
=['pays', 'nord_sud'])}
119 @register.inclusion_tag('admin/filter_select.html', takes_context
=True)
120 def filter_implantation_remun(context
):
121 return {'title': u
"implantation",
122 'choices': prepare_choices(Implantation
.objects
.values_list('id', 'nom'), 'dossiers__poste__implantation', context
)}
124 @register.inclusion_tag('rh/rapports/table_header.html', takes_context
=True)
125 def table_header(context
, headers
):
126 return {'headers': headers
}
128 def get_query_string(request
, new_params
=None, remove
=None):
129 if new_params
is None: new_params
= {}
130 if remove
is None: remove
= []
131 p
= dict(request
.GET
.items())
136 for k
, v
in new_params
.items():
142 return '?%s' % urlencode(p
)
145 def prepare_choices(choices
, query_param
, context
, remove
=[]):
146 request
= context
['request']
147 query_val
= request
.GET
.get(query_param
)
148 result
= [{'selected': query_val
is None,
149 'query_string': get_query_string(request
, {}, [query_param
] + remove
),
152 result
.append({'selected': smart_unicode(k
) == query_val
,
153 'query_string': get_query_string(request
, {query_param
: k
}, remove
),
158 def prepare_choices_date(field_name
, context
, links
, remove
=[]):
159 request
= context
['request']
161 field_generic
= '%s__' % field_name
162 date_params
= dict([(k
, v
) for k
, v
in params
.items() if k
.startswith(field_generic
)])
167 for title
, param_dict
in links
:
168 result
.append({'selected': date_params
== param_dict
,
169 'query_string': get_query_string(request
, param_dict
, [field_generic
]),
174 def __init__(self
, request
, headers
, order_field_type
, order_field
, default_order_type
="asc", not_sortable
=[]):
175 self
.request
= request
176 self
.order_field_type
= order_field_type
177 self
.order_field
= order_field
178 self
.header_defs
= headers
179 self
.params
= dict(request
.GET
)
180 if default_order_type
not in ('asc', 'desc'):
181 raise AttributeError(u
"L'ordre par défaut n'est pas valide")
182 self
.default_order_type
= default_order_type
183 self
.current_order_field
= self
.params
[self
.order_field
][0] if self
.order_field
in self
.params
else None
184 self
.current_order_field_type
= self
.params
[self
.order_field_type
][0] if self
.order_field_type
in self
.params
else None
185 self
.not_sortable
= not_sortable
188 for h
in self
.header_defs
:
190 if h
[0] == self
.current_order_field
:
191 order_type
= {'asc': 'desc', 'desc': 'asc'}[self
.current_order_field_type
]
192 class_order
= "%sending" % self
.current_order_field_type
194 order_type
= self
.default_order_type
196 params
[self
.order_field_type
] = [order_type
]
197 params
[self
.order_field
] = [h
[0]]
198 url
= "?%s" % "&".join(['%s=%s' % (param
, value
[0]) for param
, value
in params
.items()])
200 "class_attr": "sorted %s" % class_order
if self
.field_sorted(h
[0]) else "",
201 "sortable": self
.field_sorted(h
[0]),
206 def field_sorted(self
, field
):
207 return True if field
not in self
.not_sortable
else False