[#3068] [#3069] [#3133] Réécriture du rapport de masse salariale

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

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

from datetime import date

from django.db import models
from django.db.models import Q
from django.db.models.query import QuerySet

from auf.django.metadata.managers import NoDeleteManager, NoDeleteQuerySet

from project.groups import get_employe_from_user
from project.groups import \
        grp_drh, grp_drh2, grp_accior, grp_abf, grp_haute_direction, \
        grp_service_utilisateurs


class SecurityManager(models.Manager):

    prefixe_service = None
    prefixe_implantation = None

    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.
        """
        employe = get_employe_from_user(user)

        ############################################
        # TRAITEMENT NORMAL
        ############################################
        # REGION
        q = Q(**{self.prefixe_implantation: employe.implantation.region})

        # SERVICE
        if self.prefixe_service \
           and grp_service_utilisateurs in user.groups.all():
            q = q | Q(**{self.prefixe_service: employe.service})

        liste = self.get_query_set().filter(q)

        ############################################
        # TRAITEMENT ACCIOR
        ############################################
        if grp_accior in user.groups.all():
            liste = self.get_query_set().all()

        ############################################
        # TRAITEMENT ABF
        ############################################
        if grp_abf in user.groups.all():
            liste = self.get_query_set().all()

        ############################################
        # TRAITEMENT HAUTE DIRECTION
        ############################################
        if grp_haute_direction in user.groups.all():
            liste = self.get_query_set().all()

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

        if user.is_superuser:
            liste = self.get_query_set().all()

        return liste


class ActifsQuerySet(QuerySet):

    def _actifs(self, debut_field, fin_field, date_min=None, date_max=None,
                annee=None):
        qs = self
        if annee:
            janvier = date(annee, 1, 1)
            decembre = date(annee, 12, 31)
            date_min = max(janvier, date_min) if date_min else janvier
            date_max = min(decembre, date_max) if date_max else decembre
        if not date_min and not date_max:
            date_min = date_max = date.today()
        if date_min:
            qs = qs.filter(
                Q(**{fin_field + '__gte': date_min}) |
                Q(**{fin_field: None})
            )
        if date_max:
            qs = qs.filter(
                Q(**{debut_field + '__lte': date_max}) |
                Q(**{debut_field: None})
            )
        return qs

    def actifs(self, *args, **kwargs):
        return self._actifs('date_debut', 'date_fin', *args, **kwargs)


class PosteQuerySet(NoDeleteQuerySet, ActifsQuerySet):
    pass


class PosteManager(SecurityManager, NoDeleteManager):
    prefixe_service = "service"
    prefixe_implantation = "implantation__region"

    def get_query_set(self):
        return PosteQuerySet(self.model).filter(supprime=False) \
                .select_related('type_poste')

    def actifs(self, *args, **kwargs):
        return self.get_query_set().actifs(*args, **kwargs)

    def ma_region_ou_service(self, user):
        return super(PosteManager, self).ma_region_ou_service(user)


class DossierQuerySet(NoDeleteQuerySet, ActifsQuerySet):
    pass


class DossierManager(SecurityManager, NoDeleteManager):
    prefixe_service = "poste__service"
    prefixe_implantation = "poste__implantation__region"

    def get_query_set(self):
        return DossierQuerySet(self.model) \
                .filter(supprime=False) \
                .select_related('poste', 'employe')

    def actifs(self, *args, **kwargs):
        return self.get_query_set().actifs(*args, **kwargs)


class RemunerationQuerySet(NoDeleteQuerySet, ActifsQuerySet):

    def actifs(self, *args, **kwargs):
        return self \
                ._actifs('date_debut', 'date_fin', *args, **kwargs) \
                ._actifs(
                    'dossier__date_debut', 'dossier__date_fin', *args, **kwargs
                )


class RemunerationManager(NoDeleteManager):

    def get_query_set(self):
        return RemunerationQuerySet(self.model).filter(supprime=False)

    def actifs(self, *args, **kwargs):
        return self.get_query_set().actifs(*args, **kwargs)


class EmployeQuerySet(NoDeleteQuerySet, ActifsQuerySet):

    def actifs(self, date_min=None, date_max=None, annee=None):
        return self._actifs('dossiers__date_debut', 'dossiers__date_fin') \
                .distinct()


class EmployeManager(NoDeleteManager):
    def get_query_set(self):
        return EmployeQuerySet(self.model) \
                .filter(supprime=False)

    def actifs(self, *args, **kwargs):
        return self.get_query_set().actifs(*args, **kwargs)


class PosteComparaisonManager(SecurityManager):
    use_for_related_fields = True
    prefixe_implantation = "implantation__region"


class DossierComparaisonManager(SecurityManager):
    use_for_related_fields = True
    prefixe_implantation = "implantation__region"


class DeviseManager(NoDeleteManager):
    pass


class ServiceManager(NoDeleteManager):
    pass


class TypeRemunerationManager(NoDeleteManager):
    pass