PEP8
[auf_rh_dae.git] / project / lib.py
index 3670aca..10ca2cd 100644 (file)
@@ -1,16 +1,50 @@
 # -*- encoding: utf-8 -*-
 
-import datamaster_modeles.models as ref
-import rh.models as rh
+from datamaster_modeles import models as ref
+from django.contrib.auth.models import Group
+
+from project.rh import models as rh
+
 
 def get_employe_from_user(user):
     """
-    Retourne un employé AUF (rh.Employe) à partir de son user Django. 
+    Retourne un employé AUF (rh.Employe) à partir de son user Django.
     """
     try:
         ref_employe = ref.Authentification.objects.get(courriel=user.email).id
         employe = rh.Employe.objects.get(id=ref_employe.id)
+        ref_employe = ref.Employe.objects.get(id=employe.id)
+        employe.courriel = ref_employe.courriel
+        employe.tel_pro_poste = ref_employe.telephone_poste
+        employe.tel_pro_ip = ref_employe.telephone_ip
+    except:
+        employe = rh.Employe.objects.none()
+    return employe
+
+
+def get_employe_from_id(id):
+    """
+    Retourne un employé AUF (rh.Employe) à partir de son id.
+    """
+    try:
+        employe = rh.Employe.objects.get(id=id)
+        employe.courriel = ref.Employe.objects.get(id=employe.id).courriel
     except:
-        #raise Exception(u"L'employé avec le courriel %s n'a pas été trouvé dans le référentiel." % user.email)
         employe = rh.Employe.objects.none()
     return employe
+
+
+def safe_create_groupe(name=None, id=None):
+    """
+    Création d'un groupe prédéfini. Retourne None, quand la création
+    ne peut se faire. (C'est le cas au syncdb, quand la table de groupe
+    n'a pas été créée encore).
+    """
+    try:
+        if name:
+            grp, created = Group.objects.get_or_create(name=name)
+        elif id:
+            grp, created = Group.objects.get_or_create(id=id)
+    except:
+        return None
+    return grp