PEP8
[auf_rh_dae.git] / project / lib.py
1 # -*- encoding: utf-8 -*-
2
3 from datamaster_modeles import models as ref
4 from django.contrib.auth.models import Group
5
6 from project.rh import models as rh
7
8
9 def 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
25 def 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
37 def 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