Fix 5453

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

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

from django.db.models import Q

from project import groups
from project.rh.managers import \
        DossierManager as RHDossierManager, \
        PosteManager as RHPosteManager, \
        PosteComparaisonManager as RHPosteComparaisonManager, \
        DossierComparaisonManager as RHDossierComparaisonManager
from project.dae.workflow import MAP_GROUPE_ETATS_A_FAIRE


class TodoManagerMixin(object):

    def mes_choses_a_faire(self, user):
        q = Q()
        employe = groups.get_employe_from_user(user)
        rien_a_faire = True
        for g in [g.name for g in user.groups.all()]:
            etats = MAP_GROUPE_ETATS_A_FAIRE.get(g, ())
            for etat in etats:
                rien_a_faire = False
                q2 = Q(etat=etat)
                if g == groups.SERVICE_UTILISATEURS:
                    q2 &= Q(**{self.prefixe_service: employe.service})
                elif g not in (
                    groups.ACCIOR,
                    groups.ABF,
                    groups.FINANCE,
                    groups.HAUTE_DIRECTION,
                    groups.DRH_NIVEAU_1,
                    groups.DRH_NIVEAU_2,
                ):
                    q2 &= Q(**{
                        self.prefixe_implantation:
                        employe.implantation.zone_administrative
                    })
                q |= q2

        if rien_a_faire:
            qs = self.ma_region_ou_service(user).none()
        else:
            qs = self.ma_region_ou_service(user).filter(q)

        return qs


class DossierManager(RHDossierManager, TodoManagerMixin):
    pass


class PosteManager(RHPosteManager, TodoManagerMixin):
    pass


class PosteComparaisonManager(RHPosteComparaisonManager, TodoManagerMixin):
    pass


class DossierComparaisonManager(RHDossierComparaisonManager, TodoManagerMixin):
    pass