project/dae/catalogues.py

   1 # -*- encoding: utf-8 -*-
   2
   3 from django.db.models import Q
   4 from rh import models as rh
   5 from utils import get_employe_from_user, is_user_dans_services_centraux
   6 from workflow import grp_drh
   7
   8 class Responsable(object):
   9     q = ""
  10     def get_query(self,q,request):
  11         if len(q) < 4:
  12             return rh.Poste.objects.none()
  13
  14         self.q = q
  15         postes =  rh.Poste.objects.filter(
  16             Q(nom__icontains=q) |
  17             Q(type_poste__nom__icontains=q) |
  18             Q(dossiers__employe__nom__icontains=q) |
  19             Q(dossiers__employe__prenom__icontains=q)
  20             ).distinct()
  21         return postes
  22
  23     def format_result(self, poste):
  24         q = self.q
  25         filtre = Q(poste=poste) & (Q(poste__nom__icontains=q) | Q(employe__nom__icontains=q) | Q(employe__prenom__icontains=q))
  26         dossiers = rh.Dossier.objects.filter(filtre)
  27
  28         # certains postes ont un id de type de poste qui n'existe pas
  29         try:
  30             nom_poste = poste.type_poste.nom
  31         except:
  32             nom_poste = ""
  33
  34         if len(dossiers) == 1:
  35             dossier = dossiers[0]
  36             if nom_poste == poste.nom:
  37                 return "[%s] %s (%s) (%s)" % (poste.implantation.id, nom_poste, poste.id, dossier.employe)
  38             else:
  39                 return "[%s] %s %s (%s) (%s)" % (poste.implantation.id, nom_poste, poste.nom, poste.id, dossier.employe)
  40         else:
  41             dossiers = poste.dossiers.all()
  42             if len(dossiers) > 0:
  43                 complement1 = dossiers[0].poste.nom
  44                 employe = unicode(dossiers[0].employe)
  45                 return "[%s] %s %s (%s) (%s)" % (poste.implantation.id, nom_poste, complement1, poste.id, employe)
  46
  47     def format_item(self, poste):
  48         """ the display of a currently selected object in the area below the search box. html is OK """
  49         return self.format_result(poste)
  50
  51     def get_objects(self, ids):
  52         """ given a list of ids, return the objects ordered as you would like them on the admin page.
  53             this is for displaying the currently selected items (in the case of a ManyToMany field)
  54         """
  55         return rh.Poste.objects.filter(pk__in=ids)
  56
  57 class Dossier(object):
  58
  59     def get_query(self,q,request):
  60
  61         employe = get_employe_from_user(request.user)
  62         prefixe_implantation = 'poste__implantation'
  63
  64         q_recherche = Q(poste__nom__icontains=q) | \
  65             Q(poste__type_poste__nom__icontains=q) | \
  66             Q(employe__nom__icontains=q) | \
  67             Q(employe__prenom__icontains=q)
  68
  69         if is_user_dans_services_centraux(request.user):
  70             q_place = Q(**{ '%s' % prefixe_implantation : employe.implantation })
  71         else:
  72             q_place = Q(**{ '%s__region' % prefixe_implantation : employe.implantation.region })
  73
  74
  75         if grp_drh in request.user.groups.all():
  76             q_filtre = q_recherche
  77         else:
  78             q_filtre = q_place & q_recherche
  79         return rh.Dossier.objects.filter(q_filtre).distinct()
  80
  81     def format_result(self, dossier):
  82         return dossier.__unicode__()
  83
  84     def format_item(self, dossier):
  85         """ the display of a currently selected object in the area below the search box. html is OK """
  86         return self.format_result(dossier)
  87
  88     def get_objects(self, ids):
  89         """ given a list of ids, return the objects ordered as you would like them on the admin page.
  90             this is for displaying the currently selected items (in the case of a ManyToMany field)
  91         """
  92         return rh.Dossier.objects.filter(pk__in=ids)
  93
  94 class Poste(object):
  95
  96     def get_query(self,q,request):
  97         employe = get_employe_from_user(request.user)
  98         prefixe_implantation = 'poste__implantation'
  99
 100         q_recherche = Q(poste__nom__icontains=q) | Q(poste__type_poste__nom__icontains=q)
 101
 102         if is_user_dans_services_centraux(request.user):
 103             q_place = Q(**{ '%s' % prefixe_implantation : employe.implantation })
 104         else:
 105             q_place = Q(**{ '%s__region' % prefixe_implantation : employe.implantation.region })
 106
 107
 108         if grp_drh in request.user.groups.all():
 109             q_filtre = q_recherche
 110         else:
 111             q_filtre = q_place & q_recherche
 112         return rh.Dossier.objects.filter(q_filtre).distinct()
 113
 114     def format_result(self, dossier):
 115         return u"[%s] %s" % (dossier.poste.implantation, dossier.poste.type_poste.nom)
 116
 117     def format_item(self, dossier):
 118         """ the display of a currently selected object in the area below the search box. html is OK """
 119         return self.format_result(dossier)
 120
 121     def get_objects(self, ids):
 122         """ given a list of ids, return the objects ordered as you would like them on the admin page.
 123             this is for displaying the currently selected items (in the case of a ManyToMany field)
 124         """
 125         return rh.Dossier.objects.filter(pk__in=ids)