Merge branch 'hotfix' into olivier

[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 get_employe_from_user
from workflow import MAP_GROUPE_ETATS_A_FAIRE, DOSSIER_ETAT_FINALISE
from workflow import grp_drh, \
                     grp_drh2, \
                     grp_accior, \
                     grp_abf, \
                     grp_haute_direction, \
                     grp_service_utilisateurs

from rh.managers import DossierManager as RHDossierManager
from rh.managers import PosteManager as RHPosteManager
from rh.managers import PosteComparaisonManager as RHPosteComparaisonManager
from rh.managers import DossierComparaisonManager as RHDossierComparaisonManager


class TodoManagerMixin(object):

    def mes_choses_a_faire(self, user):
        q = Q()
        employe = get_employe_from_user(user)
        rien_a_faire = True
        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 == grp_service_utilisateurs:
                    q2 &= Q(**{self.prefixe_service: employe.service})
                elif g not in (grp_accior, grp_abf, grp_haute_direction, grp_drh, grp_drh2):
                    q2 &= Q(**{self.prefixe_implantation: employe.implantation.region})
                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

class DossierFinaliseManager(models.Manager):
    """
    Dossiers finalisés, qui ne sont pas encore importés
    """
    def get_query_set(self):
        from dae.models import ImportDossier
        dae_importees = [str(i.dae_id) for i in ImportDossier.objects.all()]
        qs = super(DossierFinaliseManager,
                self).get_query_set().filter(etat=DOSSIER_ETAT_FINALISE,).extra(where=['id\
                    NOT IN (%s)' % ", ".join(dae_importees)])
        return qs