PEP8
[auf_rh_dae.git] / project / lib.py
... / ...
CommitLineData
1# -*- encoding: utf-8 -*-
2
3from datamaster_modeles import models as ref
4from django.contrib.auth.models import Group
5
6from project.rh import models as rh
7
8
9def get_employe_from_user(user):
10 """
11 Retourne un employé AUF (rh.Employe) à partir de son user Django.
12 """
13 try:
14 ref_employe = ref.Authentification.objects.get(courriel=user.email).id
15 employe = rh.Employe.objects.get(id=ref_employe.id)
16 ref_employe = ref.Employe.objects.get(id=employe.id)
17 employe.courriel = ref_employe.courriel
18 employe.tel_pro_poste = ref_employe.telephone_poste
19 employe.tel_pro_ip = ref_employe.telephone_ip
20 except:
21 employe = rh.Employe.objects.none()
22 return employe
23
24
25def get_employe_from_id(id):
26 """
27 Retourne un employé AUF (rh.Employe) à partir de son id.
28 """
29 try:
30 employe = rh.Employe.objects.get(id=id)
31 employe.courriel = ref.Employe.objects.get(id=employe.id).courriel
32 except:
33 employe = rh.Employe.objects.none()
34 return employe
35
36
37def safe_create_groupe(name=None, id=None):
38 """
39 Création d'un groupe prédéfini. Retourne None, quand la création
40 ne peut se faire. (C'est le cas au syncdb, quand la table de groupe
41 n'a pas été créée encore).
42 """
43 try:
44 if name:
45 grp, created = Group.objects.get_or_create(name=name)
46 elif id:
47 grp, created = Group.objects.get_or_create(id=id)
48 except:
49 return None
50 return grp