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