PEP8
[auf_rh_dae.git] / project / lib.py
CommitLineData
58e9fdc2
OL
1# -*- encoding: utf-8 -*-
2
b31ce2d7
EMS
3from datamaster_modeles import models as ref
4from django.contrib.auth.models import Group
5
6from project.rh import models as rh
7
58e9fdc2
OL
8
9def get_employe_from_user(user):
10 """
b31ce2d7 11 Retourne un employé AUF (rh.Employe) à partir de son user Django.
58e9fdc2
OL
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:
58e9fdc2
OL
21 employe = rh.Employe.objects.none()
22 return employe
23
b31ce2d7 24
58e9fdc2
OL
25def get_employe_from_id(id):
26 """
b31ce2d7 27 Retourne un employé AUF (rh.Employe) à partir de son id.
58e9fdc2
OL
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
b31ce2d7
EMS
35
36
58e9fdc2
OL
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)
b31ce2d7 46 elif id:
58e9fdc2
OL
47 grp, created = Group.objects.get_or_create(id=id)
48 except:
49 return None
50 return grp