Rapport masse salariale: ajout de formattage pour les chiffres
[auf_rh_dae.git] / project / rh / masse_salariale.py
1 # -*- encoding: utf-8 -*-
2 import time
3 import datetime
4 import csv
5 import StringIO
6 import itertools
7
8 from django.db.models import Q
9
10 from datamaster_modeles import models as ref
11 import rh.ods as ods
12 import rh.models as rh
13
14
15 KEY_DATE_DEBUT = "debut"
16 KEY_DATE_FIN = "fin"
17
18 TYPE_REMUN_BSTG = (3,)
19 TYPE_REMUN_MAD = (2,)
20 TYPE_REMUN_BASE = (1,)
21 TYPE_REMUN_FONC_RESP = (7, 8)
22 TYPE_REMUN_EXPAT = (4,)
23 TYPE_REMUN_LOGEMENT = (6,)
24 TYPE_REMUN_TRANSP = (9,)
25 TYPE_REMUN_13E = (18,)
26 TYPE_PRIME_INTERIM = (19,)
27 TYPE_REMUN_ALL_INDEMNITES = list(itertools.chain(*(TYPE_REMUN_BSTG,
28 TYPE_REMUN_MAD, TYPE_REMUN_BASE, TYPE_REMUN_FONC_RESP,
29 TYPE_REMUN_EXPAT, TYPE_REMUN_LOGEMENT, TYPE_REMUN_TRANSP,
30 TYPE_REMUN_13E, TYPE_PRIME_INTERIM)))
31 TYPE_PRIME_INSTALLATION = (13,)
32 TYPE_PRIME_DEMENAG = (15,)
33 TYPE_PRIME_AVION = (14,)
34 TYPE_PRIME_ALL = list(itertools.chain(
35 *(TYPE_PRIME_INSTALLATION, TYPE_PRIME_DEMENAG, TYPE_PRIME_AVION)
36 ))
37 TYPE_CHARGE_PATRONALE = (17,)
38 TYPE_CHARGE_ALL = list(itertools.chain(*(TYPE_CHARGE_PATRONALE,)))
39 TYPE_NATURE_INDEMN = u"Indemnité"
40 TYPE_NATURE_PAIEMENT = u"Accessoire"
41 TYPE_NATURE_CHARGES = u"Charges"
42
43
44 class MasseSalariale():
45 """ Rapport de la masse salariale. """
46
47 def __init__(self, date_debut, date_fin, custom_filter=None,
48 ne_pas_grouper=False):
49 """ date_debut: date de début pour les données temporelles
50 date_fin: idem
51 custom_filter: dictionnaire des paramètres à passer au queryset.
52 """
53 if not date_debut and not date_fin:
54 return
55
56 date_debut = datetime.date(
57 *time.strptime(date_debut, "%d-%m-%Y")[0:3]
58 )
59 date_fin = datetime.date(*time.strptime(date_fin, "%d-%m-%Y")[0:3])
60
61 rapport_date_delta = date_fin - date_debut
62 rapport_date_delta += datetime.timedelta(days=1)
63
64 self.annee = date_fin.year
65
66 self.devise_base = rh.Devise.objects.filter(code='EUR')[0]
67 self.taux_change = {}
68
69 q_range = self.build_qs("date_", date_debut, date_fin)
70 q_range_d = self.build_qs("dossier__date_", date_debut, date_fin)
71 remunerations = rh.Remuneration.objects.filter(q_range) \
72 .filter(q_range_d) \
73
74 if custom_filter:
75 remunerations = remunerations.filter(**custom_filter)
76
77 remunerations = remunerations.exclude(supprime=True) \
78 .select_related(
79 "dossier", "dossier_employe", "dossier_poste", "type"
80 )
81
82 contenu = {}
83
84 lineariser_dossiers = not ne_pas_grouper
85
86 for r in remunerations:
87 if lineariser_dossiers:
88 key = r.dossier.employe_id
89 else:
90 key = r.dossier_id
91
92 if key not in contenu:
93 contenu[key] = {
94 'dossiers': set(),
95 'remunerations': []
96 }
97 if lineariser_dossiers:
98 contenu[key]['remunerations'].append(r)
99 else:
100 if r.dossier_id == key:
101 contenu[key]['remunerations'].append(r)
102 contenu[key]['dossiers'].add(r.dossier)
103
104 self.rapport = []
105
106 pays_list = {}
107 for pays in ref.Pays.objects.all():
108 pays_list[pays.id] = pays
109
110 valeurs_point_par_imp = \
111 dict(
112 (v.implantation.id, v) for v in \
113 rh.ValeurPoint.objects.filter(annee=self.annee).all()
114 )
115
116 self.headers = (
117 ('bureau', u"Bureau", {'columnwidth': '2cm'}),
118 ('pays', u"Pays", {'columnwidth': '3.5cm'}),
119 ('implantation', u"Implantation", {'columnwidth': '3cm'}),
120 ('valeur_point', u"Valeur du point",
121 {'columnwidth': '2.3cm'}),
122 ('numero_employe', u"Numéro d'employé",
123 {'columnwidth': '2.4cm'}),
124 ('nom', u"Nom", {'columnwidth': '5.4cm'}),
125 ('prenom', u"Prénom", {'columnwidth': '4.8cm'}),
126 ('type_de_poste', u"Type de poste", {'columnwidth': '2.7cm'}),
127 ('intitule_de_poste', u"Intitulé du poste",
128 {'columnwidth': '7.25cm'}),
129 ('niveau', u"Niveau", {'columnwidth': '1.75cm'}),
130 ('point', u"Point", {'columnwidth': '1.75cm'}),
131 ('regime_de_travail', u"Régime de travail",
132 {'columnwidth': '2cm'}),
133 ('local_expatrie', u"Local / Expatrié",
134 {'columnwidth': '2.25cm'}),
135 ('statut', u"Statut", {'columnwidth': '1.25cm'}),
136 ('date_fin_contrat', u"Date de fin de contrat",
137 {'columnwidth': '2cm'}),
138 ('date_debut', u"Date de début", {'columnwidth': '1.92cm'}),
139 ('date_fin', u"Date de fin", {'columnwidth': '1.92cm'}),
140 ('nb_jours', u"Nombre de jours", {'columnwidth': '2.82cm'}),
141 ('devise', u"Devise", {'columnwidth': '1.46cm'}),
142 ('salaire_bstg_annuel', u"Salaire annuel BSTG",
143 {'columnwidth': '2.5cm'}),
144 ('salaire_bstg_total', u"Salaire total BSTG",
145 {'columnwidth': '2.5cm'}),
146 ('organisme_bstg', u"Organisme BSTG",
147 {'columnwidth': '2.9cm'}),
148 ('salaire_theorique', u"Salaire théorique",
149 {'columnwidth': '2.5cm'}),
150 ('salaire_base_brut', u"Salaire de base brut",
151 {'columnwidth': '2.5cm'}),
152 ('salaire_complementaire', u"Salaire complémentaire",
153 {'columnwidth': '2.5cm'}),
154 ('indemnite_fonctions', u"Indemnités de fonctions",
155 {'columnwidth': '2.5cm'}),
156 ('indemnite_expat', u"Indemnités d'expatriation",
157 {'columnwidth': '2.5cm'}),
158 ('indemnite_logement', u"Indemnités de logement",
159 {'columnwidth': '2.5cm'}),
160 ('indemnite_transp', u"Indemnités de transport",
161 {'columnwidth': '2.5cm'}),
162 ('indemnite_13e', u"Indemnités 13e mois",
163 {'columnwidth': '2.5cm'}),
164 ('prime_interim', u"Prime d'intérim",
165 {'columnwidth': '2.5cm'}),
166 ('indemnite_autre', u"Autre indemnités",
167 {'columnwidth': '2.5cm'}),
168 ('indemnite_sous_total', u"Sous-total d'indemnités",
169 {'columnwidth': '2.5cm'}),
170 ('prime_installation', u"Prime d'installation",
171 {'columnwidth': '2.5cm'}),
172 ('prime_demenagement', u"Prime de déménagement",
173 {'columnwidth': '2.5cm'}),
174 ('prime_avion', u"Prime d'avion",
175 {'columnwidth': '2.5cm'}),
176 ('prime_autre', u"Autre prime",
177 {'columnwidth': '2.5cm'}),
178 ('prime_sous_total', u"Total des primes",
179 {'columnwidth': '2.5cm'}),
180 ('charges_patronales', u"Charges patronales",
181 {'columnwidth': '2.5cm'}),
182 ('charges_autre', u"Autres charges patronales",
183 {'columnwidth': '2.5cm'}),
184 ('charges_sous_total', u"Sous-total des charges patronales",
185 {'columnwidth': '2.5cm'}),
186 ('masse_salariale', u"Masse salariale",
187 {'columnwidth': '2.5cm'}),
188 ('masse_salariale_annee', u"Masse salariale %s" % self.annee,
189 {'columnwidth': '2.5cm'}),
190 ('masse_salariale_annee_euro', u"Masse salariale euro %s" % \
191 self.annee, {'columnwidth': '2.5cm'}),
192 )
193
194 grand_total = 0.0
195 grand_total_euro = 0.0
196
197 for item in contenu.values():
198 dossiers = item['dossiers']
199 remuns = item['remunerations']
200
201 for d in dossiers:
202 if d.principal:
203 dossier = list(dossiers)[0]
204
205 regime = float(dossier.poste.regime_travail) / 100
206
207 if dossier.poste.expatrie:
208 statut = "E"
209 else:
210 statut = "L"
211
212 #on détermine la date du début et fin du dossier si année en cours
213 try:
214 d_date_fin = dossier.date_fin \
215 if dossier.date_fin.year == date_fin.year else None
216 except AttributeError:
217 d_date_fin = None
218 try:
219 d_date_debut = dossier.date_debut \
220 if dossier.date_debut.year == date_fin.year else None
221 except AttributeError:
222 d_date_debut = None
223
224 pays = \
225 pays_list[dossier.poste.implantation.adresse_physique_pays.id]
226
227 #on détermine si les rémunérations sont tous dans la même devise
228 devise = remuns[0].devise
229 meme_devise = True
230 for r in remuns[1:]:
231 if devise != r.devise:
232 meme_devise = False
233
234 if not meme_devise:
235 for r in remuns:
236 self.convertir(r)
237
238 bstg_dossier = None
239 for d in dossiers:
240 if d.organisme_bstg:
241 bstg_dossier = d
242
243 bstg_remun = None
244 if bstg_dossier:
245 for r in bstg_dossier.rh_remunerations.all():
246 if r.type.id in TYPE_REMUN_BSTG:
247 bstg_remun = r
248
249 if bstg_remun:
250 bstg_remun_euro = rh.Remuneration(
251 montant=bstg_remun.montant, devise=bstg_remun.devise
252 )
253 self.convertir(bstg_remun_euro)
254
255 salaire_complement = 0.0
256 salaire_base = 0.0
257 indemnites = {
258 'fonc_resp': 0.0,
259 'expat': 0.0,
260 'logement': 0.0,
261 'transp': 0.0,
262 '13e': 0.0,
263 'autre_recurr': 0.0,
264 'interim': 0.0,
265 }
266
267 primes = {
268 'installation': 0.0,
269 'demenagement': 0.0,
270 'avion': 0.0,
271 'autre': 0.0,
272 }
273 charges = {
274 'patronale': 0.0,
275 'autre': 0.0,
276 }
277 for r in remuns:
278 montant = float(r.montant)
279
280 if r.type_id in TYPE_REMUN_MAD:
281 salaire_complement += montant
282
283 if r.type_id in TYPE_REMUN_BASE:
284 salaire_base += montant
285
286 if r.type_id in TYPE_REMUN_FONC_RESP:
287 indemnites['fonc_resp'] += montant
288
289 if r.type_id in TYPE_REMUN_EXPAT:
290 indemnites['expat'] += montant
291
292 if r.type_id in TYPE_REMUN_LOGEMENT:
293 indemnites['logement'] += montant
294
295 if r.type_id in TYPE_REMUN_TRANSP:
296 indemnites['transp'] += montant
297
298 if r.type_id in TYPE_REMUN_13E:
299 indemnites['13e'] += montant
300
301 if r.type_id in TYPE_PRIME_INTERIM:
302 indemnites['interim'] += montant
303
304 if r.type_id not in TYPE_REMUN_ALL_INDEMNITES \
305 and r.type.nature_remuneration == TYPE_NATURE_INDEMN:
306 indemnites['autre_recurr'] += montant
307
308 if r.type_id in TYPE_PRIME_INSTALLATION:
309 primes['installation'] += montant
310
311 if r.type_id in TYPE_PRIME_DEMENAG:
312 primes['demenagement'] += montant
313
314 if r.type_id in TYPE_PRIME_AVION:
315 primes['avion'] += montant
316
317 if r.type_id not in TYPE_PRIME_ALL and \
318 r.type.nature_remuneration == TYPE_NATURE_PAIEMENT:
319 primes['autre'] += montant
320
321 if r.type_id in TYPE_CHARGE_PATRONALE:
322 charges['patronale'] += montant
323
324 if r.type_id not in TYPE_CHARGE_ALL and \
325 r.type.nature_remuneration == TYPE_NATURE_CHARGES:
326 charges['autre'] += montant
327
328 total_indemnites = sum(indemnites.values())
329
330 #Calcul du nombre de jours pour ce dossier.
331 if dossier.date_debut and dossier.date_debut > date_debut and \
332 not dossier.date_fin:
333 date_delta = date_fin - dossier.date_fin
334 elif dossier.date_fin and dossier.date_fin < date_fin and \
335 not dossier.date_debut:
336 date_delta = dossier.date_fin - date_debut
337 elif dossier.date_fin and dossier.date_debut:
338 date_delta = dossier.date_fin - date_debut
339 else:
340 date_delta = rapport_date_delta
341
342 masse_salariale = (salaire_base + total_indemnites + \
343 sum(primes.values()) + sum(charges.values()))
344 masse_salariale_euro = rh.Remuneration(montant=masse_salariale,
345 devise=remuns[0].devise)
346 self.convertir(masse_salariale_euro)
347
348 if dossier.classement and dossier.classement.coefficient:
349 coefficient = dossier.classement.coefficient
350 else:
351 coefficient = ""
352
353 #todo valeur du point si pas présent
354 valeur_point = valeurs_point_par_imp.get(
355 dossier.poste.implantation_id
356 ) or ""
357
358 salaire_theorique = (
359 round(valeur_point.valeur * int(coefficient) * regime, 2) \
360 if valeur_point and coefficient and regime else None)
361
362 rapport_nombre_jours = (float(date_delta.days)
363 / rapport_date_delta.days)
364 item_rapport = {
365 'bureau': dossier.poste.implantation.region.code,
366 'pays': unicode(pays),
367 'implantation': dossier.poste.implantation.nom_court,
368 'type_implantation': dossier.poste.implantation.type,
369 #'imputation': None,
370 'valeur_point': valeur_point,
371 'numero_employe': dossier.employe_id,
372 'nom': dossier.employe.nom.upper(),
373 'prenom': dossier.employe.prenom,
374 'type_de_poste': dossier.poste.type_poste.nom,
375 'intitule_de_poste': dossier.poste.nom,
376 'niveau': unicode(dossier.classement),
377 'point': coefficient,
378 'regime_de_travail': "%s %%" % int(regime * 100),
379 'local_expatrie': statut,
380 'statut': dossier.statut.code,
381 'date_fin_contrat': dossier.date_fin or "",
382 'date_debut': d_date_debut or "",
383 'date_fin': d_date_fin or "",
384 'nb_jours': date_delta.days,
385 'devise': remuns[0].devise,
386 'salaire_bstg_annuel': bstg_remun.montant \
387 if bstg_remun else "",
388 'salaire_bstg_total': bstg_remun_euro.montant \
389 if bstg_remun else "",
390 'organisme_bstg': dossier.organisme_bstg or "",
391 'salaire_theorique': salaire_theorique,
392 'salaire_base_brut': \
393 salaire_base * regime * rapport_nombre_jours,
394 'salaire_complementaire': \
395 salaire_complement * regime *
396 rapport_nombre_jours,
397 #'salaire_total': None
398 'indemnite_fonctions': indemnites['fonc_resp'] * \
399 regime * rapport_nombre_jours,
400 'indemnite_expat': indemnites['expat'] * regime * \
401 rapport_nombre_jours,
402 'indemnite_logement': indemnites['logement'] * \
403 regime * rapport_nombre_jours,
404 'indemnite_transp': indemnites['transp'] * regime * \
405 rapport_nombre_jours,
406 'indemnite_13e': indemnites['13e'] * regime * \
407 rapport_nombre_jours,
408 'prime_interim': indemnites['interim'] * regime * \
409 rapport_nombre_jours,
410 'indemnite_autre': indemnites['autre_recurr'] * \
411 regime * rapport_nombre_jours,
412 'indemnite_sous_total': total_indemnites * regime * \
413 rapport_nombre_jours,
414 'total_brut': (
415 total_indemnites + salaire_base +
416 salaire_complement
417 ) * regime * rapport_nombre_jours,
418 'prime_installation': primes['installation'] * regime * \
419 rapport_nombre_jours,
420 'prime_demenagement': primes['demenagement'] * regime * \
421 rapport_nombre_jours,
422 'prime_avion': primes['avion'] * regime * \
423 rapport_nombre_jours,
424 'prime_autre': primes['autre'] * regime * \
425 rapport_nombre_jours,
426 'prime_sous_total': sum(primes.values()) * regime * \
427 rapport_nombre_jours,
428 'charges_patronales': charges['patronale'],
429 'charges_autre': charges['autre'],
430 'charges_sous_total': sum(charges.values()),
431 'masse_salariale': masse_salariale,
432 'masse_salariale_annee': masse_salariale * \
433 regime * rapport_nombre_jours,
434 'masse_salariale_annee_euro': \
435 masse_salariale_euro.montant * regime *
436 rapport_nombre_jours,
437 }
438
439 grand_total += round(masse_salariale, 2)
440 grand_total_euro += round(masse_salariale_euro.montant * regime
441 * (
442 date_delta.days / rapport_date_delta.days
443 ), 2)
444
445 self.rapport.append(item_rapport)
446
447 self.grand_totaux = (grand_total, grand_total_euro)
448
449 def build_qs(self, prefix, date_debut, date_fin):
450 date_debut_null = \
451 Q(**{"%s%s__isnull" % (prefix, KEY_DATE_DEBUT): True})
452 date_fin_null = \
453 Q(**{"%s%s__isnull" % (prefix, KEY_DATE_FIN): True})
454 date_debut_superieure_ou_egale_a_borne_gauche = \
455 Q(**{"%s%s__gte" % (prefix, KEY_DATE_DEBUT): date_debut})
456 date_debut_inferieure_ou_egale_a_borne_gauche = \
457 Q(**{"%s%s__lte" % (prefix, KEY_DATE_DEBUT): date_debut})
458 date_fin_superieure_ou_egale_a_borne_gauche = \
459 Q(**{"%s%s__gte" % (prefix, KEY_DATE_FIN): date_debut})
460 date_fin_inferieure_ou_egale_a_borne_droite = \
461 Q(**{"%s%s__lte" % (prefix, KEY_DATE_FIN): date_fin})
462 date_debut_inferieure_ou_egale_a_borne_droite = \
463 Q(**{"%s%s__lte" % (prefix, KEY_DATE_DEBUT): date_fin})
464 date_fin_superieure_ou_egale_a_borne_droite = \
465 Q(**{"%s%s__gte" % (prefix, KEY_DATE_FIN): date_fin})
466
467 q_range = \
468 (
469 date_debut_null & date_fin_null
470 ) | (
471 date_debut_inferieure_ou_egale_a_borne_gauche &
472 date_fin_superieure_ou_egale_a_borne_gauche &
473 date_fin_inferieure_ou_egale_a_borne_droite
474 ) | (
475 date_debut_superieure_ou_egale_a_borne_gauche &
476 date_debut_inferieure_ou_egale_a_borne_droite &
477 date_fin_superieure_ou_egale_a_borne_droite
478 ) | (
479 date_debut_inferieure_ou_egale_a_borne_gauche &
480 date_fin_superieure_ou_egale_a_borne_droite
481 ) | (
482 date_debut_null &
483 date_fin_superieure_ou_egale_a_borne_droite
484 ) | (
485 date_debut_inferieure_ou_egale_a_borne_gauche &
486 date_fin_null
487 )
488
489 return q_range
490
491 def convertir(self, remuneration):
492 if remuneration.devise != self.devise_base:
493 remuneration.montant = float(remuneration.montant) * \
494 self.trouver_taux(remuneration.devise).taux
495 remuneration.devise = self.devise_base
496
497 def trouver_taux(self, devise):
498 if devise.code not in self.taux_change:
499 t = rh.TauxChange.objects.filter(
500 devise=devise, annee=self.annee
501 )[0]
502 self.taux_change[devise.code] = t
503 return self.taux_change[devise.code]
504
505 def csv(self):
506 self.csv_handle = StringIO.StringIO()
507 csv_writer = csv.writer(self.csv_handle, delimiter=",",
508 doublequote=False, escapechar="\\", quoting=csv.QUOTE_ALL,
509 )
510 header = [v[0] for v in self.rapport[0]]
511 csv_writer.writerow(header)
512 for row in self.rapport:
513 values = [v[1] for v in row]
514 csv_writer.writerow(
515 [unicode(r).encode('utf-8') for r in values]
516 )
517
518 def ods(self):
519 self.doc = ods.OpenDocumentSpreadsheet()
520 table = self.doc.add_table(name=u'Masse salariale %s' % self.annee)
521
522 for h in self.headers:
523 if len(h) > 2:
524 table.add_column(**h[2])
525
526 table.add_row([h[1] for h in self.headers], rowheight='2cm', verticalalign='middle')
527
528