#1490

[auf_rh_dae.git] / project / dae / managers.py

# -*- encoding: utf-8 -*-

from django.db import models
from django.db.models import Q
from utils import is_user_dans_service, get_employe_from_user
from workflow import POSTE_ETAT_HAUTE_DIRECTION, POSTE_ETAT_POLE_FINANCIER, MAP_GROUPE_ETATS_A_FAIRE
from workflow import dae_groupes, \
                     grp_administrateurs, \
                     grp_gestionnaires, \
                     grp_directeurs_bureau, \
                     grp_drh, \
                     grp_pole_financier, \
                     grp_haute_direction, \
                     grp_service_utilisateurs, \
                     grp_directeurs_service, \
                     grp_correspondants_rh


class SecurityManager(models.Manager):

    prefixe_implantation = None

    def mes_choses_a_faire(self, user):
        q = Q()
        for g in user.groups.all():
            etats = MAP_GROUPE_ETATS_A_FAIRE.get(g, ())
            for etat in etats:
                q = q | Q(etat=etat)

        qs = self.ma_region_ou_service(user).filter(q)
        return qs

    def ma_region_ou_service(self, user):
        """
        Filtrage des postes en fonction du user connecté (region / service)
        On s'intéresse aussi au groupe auquel appartient le user car certains groupes
        peuvent tout voir.
        """

        # On s'assure que le manager travaille sur un modèle avec WF, autrement
        # on ne teste pas sur l'état
        try:
            self.model._meta.get_field_by_name('etat')
            workflowed_model = True
        except:
            workflowed_model = False

        employe = get_employe_from_user(user)

        ############################################
        # TRAITEMENT NORMAL
        ############################################        

        # SERVICE
        if is_user_dans_service(user):
            q = Q(**{ '%s' % self.prefixe_implantation : employe.implantation })
        # REGION
        else:
            q = Q(**{ '%s__region' % self.prefixe_implantation : employe.implantation.region })
        liste = self.get_query_set().filter(q)

        ############################################
        # TRAITEMENT POLE FINANCIER
        ############################################        
        if workflowed_model and grp_pole_financier in user.groups.all():
            liste = self.get_query_set().filter(etat=POSTE_ETAT_POLE_FINANCIER)

        ############################################
        # TRAITEMENT HAUTE DIRECTION
        ############################################        
        if workflowed_model and grp_haute_direction in user.groups.all():
            liste = self.get_query_set().filter(etat=POSTE_ETAT_HAUTE_DIRECTION)

        ############################################
        # TRAITEMENT DRH
        ############################################        
        if grp_drh in user.groups.all():
            liste = self.get_query_set()
        
        return liste


class PosteManager(SecurityManager):
    """
    Chargement de tous les objets FK existants sur chaque QuerySet.
    """
    prefixe_implantation = "implantation"

    def ma_region_ou_service(self, user):
        return super(PosteManager, self).filter(actif=True)

    def get_query_set(self):
        fkeys = (
            'id_rh',
            'responsable',
            'implantation',
            'type_poste',
            'service',
            'classement_min',
            'classement_max',
            'valeur_point_min',
            'valeur_point_max',
        )
        return super(PosteManager, self).get_query_set() \
                                        .select_related(*fkeys).all()


class DossierManager(SecurityManager):
    prefixe_implantation = "poste__implantation"

    def ma_region_ou_service(self, user):
        return super(DossierManager, self).filter(poste__actif=True)