project/dae/managers.py

   1 # -*- encoding: utf-8 -*-
   2
   3 from django.db import models
   4 from django.db.models import Q
   5 from utils import get_employe_from_user
   6 from workflow import POSTE_ETAT_HAUTE_DIRECTION, \
   7                      POSTE_ETAT_ACCIOR, \
   8                      POSTE_ETAT_ABF, \
   9                      DOSSIER_ETAT_ACCIOR, \
  10                      DOSSIER_ETAT_ABF, \
  11                      MAP_GROUPE_ETATS_A_FAIRE
  12 from workflow import dae_groupes, \
  13                      grp_administrateurs, \
  14                      grp_directeurs_bureau, \
  15                      grp_drh, \
  16                      grp_accior, \
  17                      grp_abf, \
  18                      grp_haute_direction, \
  19                      grp_service_utilisateurs, \
  20                      grp_correspondants_rh
  21
  22
  23 class SecurityManager(models.Manager):
  24
  25     prefixe_service = None
  26     prefixe_implantation = None
  27
  28     def mes_choses_a_faire(self, user):
  29         q = Q()
  30         rien_a_faire = True
  31         for g in user.groups.all():
  32             etats = MAP_GROUPE_ETATS_A_FAIRE.get(g, ())
  33             for etat in etats:
  34                 rien_a_faire = False
  35                 q = q | Q(etat=etat)
  36
  37         if rien_a_faire:
  38             qs = self.ma_region_ou_service(user).none()
  39         else:
  40             qs = self.ma_region_ou_service(user).filter(q)
  41
  42         return qs
  43
  44     def ma_region_ou_service(self, user):
  45         """
  46         Filtrage des postes en fonction du user connecté (region / service)
  47         On s'intéresse aussi au groupe auquel appartient le user car certains groupes
  48         peuvent tout voir.
  49         """
  50         # On s'assure que le manager travaille sur un modèle avec WF, autrement
  51         # on ne teste pas sur l'état
  52         try:
  53             self.model._meta.get_field_by_name('etat')
  54             workflowed_model = True
  55         except:
  56             workflowed_model = False
  57
  58         employe = get_employe_from_user(user)
  59
  60         ############################################
  61         # TRAITEMENT NORMAL
  62         ############################################
  63         # REGION
  64         q = Q(**{ self.prefixe_implantation : employe.implantation.region })
  65
  66         # SERVICE
  67         if self.prefixe_service and workflowed_model and \
  68            grp_service_utilisateurs in user.groups.all():
  69             q = q | Q(**{ self.prefixe_service : employe.service})
  70
  71         liste = self.get_query_set().filter(q)
  72
  73
  74         ############################################
  75         # TRAITEMENT ACCIOR
  76         ############################################
  77         if workflowed_model and grp_accior in user.groups.all():
  78             liste = self.get_query_set().all()
  79
  80         ############################################
  81         # TRAITEMENT ABF
  82         ############################################
  83         if workflowed_model and grp_abf in user.groups.all():
  84             liste = self.get_query_set().all()
  85
  86         ############################################
  87         # TRAITEMENT HAUTE DIRECTION
  88         ############################################
  89         if workflowed_model and grp_haute_direction in user.groups.all():
  90             liste = self.get_query_set().all()
  91
  92         ############################################
  93         # TRAITEMENT DRH
  94         ############################################
  95         if grp_drh in user.groups.all():
  96             liste = self.get_query_set().all()
  97
  98         return liste
  99
 100
 101 class PosteManager(SecurityManager):
 102     """
 103     Chargement de tous les objets FK existants sur chaque QuerySet.
 104     """
 105     prefixe_service = "service"
 106     prefixe_implantation = "implantation__region"
 107
 108     def ma_region_ou_service(self, user):
 109         return super(PosteManager, self).ma_region_ou_service(user).filter(actif=True)
 110
 111     def get_query_set(self):
 112         fkeys = (
 113             'id_rh',
 114             'responsable',
 115             'implantation',
 116             'implantation.bureau_rattachement',
 117             'type_poste',
 118             'service',
 119             'classement_min',
 120             'classement_max',
 121             'valeur_point_min',
 122             'valeur_point_max',
 123         )
 124         return super(PosteManager, self).get_query_set() \
 125                                         .select_related(*fkeys).all()
 126
 127
 128 class DossierManager(SecurityManager):
 129     prefixe_service = "poste__service"
 130     prefixe_implantation = "poste__implantation__region"
 131
 132     def get_query_set(self):
 133         fkeys = (
 134             'poste',
 135         )
 136         return super(DossierManager, self).get_query_set() \
 137                                         .select_related(*fkeys).all()
 138
 139     def ma_region_ou_service(self, user):
 140         return super(DossierManager, self).ma_region_ou_service(user).filter(poste__actif=True)
 141
 142
 143 class PosteComparaisonManager(SecurityManager):
 144     use_for_related_fields = True
 145     prefixe_implantation = "implantation__region"