groups in permissions ; utilisation id plutôt que name
[auf_rh_dae.git] / project / lib.py
index 2c90073..7a2e010 100644 (file)
@@ -10,19 +10,34 @@ def get_employe_from_user(user):
     try:
         ref_employe = ref.Authentification.objects.get(courriel=user.email).id
         employe = rh.Employe.objects.get(id=ref_employe.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 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:
+        employe = rh.Employe.objects.none()
+    return employe
     
-def safe_create_groupe(name):
+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:
-        grp, created = Group.objects.get_or_create(name=name)
+        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