Routeur: Ajout temporaire des modules workflow et dae en écriture

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

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

from django.shortcuts import get_object_or_404

from project import groups
from project.decorators import redirect_interdiction
from project.rh import models as rh

from project.dae import models as dae
from project.dae.groups import dae_groupes
from project.dae.workflow import ETATS_EDITABLE, ETATS_VALIDE


def user_in_dae_groupes(user):
    """
    Teste si un user Django fait parti des groupes prédéfinis de DAE.
    """
    for g in [g.name for g in user.groups.all()]:
        if g in dae_groupes:
            return True
    return False

def dae_groupe_requis(fn):
    """
    L'accès à la plateforme nécessite d'appartenir au moins à un groupe, ou
    d'être superuser. De cette manière l'autentification AUF fonctionnera
    toujours, il suffit de mettre le nouvel employé dans le groupe qui le
    concerne.
    """
    def inner(request, *args, **kwargs):
        user = request.user

        if user.is_superuser or user_in_dae_groupes(user):
            return fn(request, *args, **kwargs)
        else:
            msg = u"Votre compte ne permet pas d'accéder à cette partie " \
                    u"de l'application."
            return redirect_interdiction(request, msg)
    return inner


def poste_dans_ma_region_ou_service(fn):
    """
    Test si le user connecté appartient bien à la même région ou service que
    le poste.
    """
    def inner(request, *args, **kwargs):
        user = request.user
        key = kwargs.get('key', None)

        # Autoriser la création d'une nouvelle demande
        if key is None:
            return fn(request, *args, **kwargs)

        # Rechercher dans la demande, la région ou le service associé
        if '-' in key:
            source, id = key.split('-')
            if source == 'dae':
                Poste = dae.Poste
            elif source == 'rh':
                Poste = rh.Poste
            else:
                raise Exception("source inconnue : dae ou rh")
        else:
            id = key
            Poste = dae.Poste
        postes = Poste.objects.ma_region_ou_service(user).filter(id=id)
        if len(postes) > 0:
            return fn(request, *args, **kwargs)
        else:
            msg = u"Vous n'avez pas le droit de consulter ce poste."
            return redirect_interdiction(request, msg)
    return inner


def dossier_dans_ma_region_ou_service(fn):
    """
    Test si le user connecté appartient bien à la même région ou service que
    le poste.
    """
    def inner(request, *args, **kwargs):
        user = request.user
        poste_key = kwargs.get('key', None)
        dossier_id = kwargs.get('dossier_id', None)

        # Si on s'intéresse à un dossier, on teste la validation avec le
        # poste associé
        if dossier_id is not None:
            dossiers = dae.Dossier.objects.ma_region_ou_service(user) \
                    .filter(id=dossier_id)
            if len(dossiers) > 0:
                return fn(request, *args, **kwargs)
            else:
                msg = u"Vous n'avez pas le droit de consulter " \
                        u"ce dossier d'embauche."
                return redirect_interdiction(request, msg)

        # Autoriser la création d'une nouvelle demande
        if poste_key is None:
            return fn(request, *args, **kwargs)
        # On est en train de répondre à un poste
        else:
            return poste_dans_ma_region_ou_service(fn)(
                request, *args, **kwargs
            )
    return inner


def vieux_dossier_dans_ma_region_ou_service(fn):
    """
    Test si le user connecté appartient bien à la même région ou service que
    le poste.
    """
    def inner(request, *args, **kwargs):
        user = request.user
        dossier_id = kwargs.get('dossier_id', None)
        dossiers = rh.Dossier.objects.ma_region_ou_service(user) \
                .filter(id=dossier_id)
        if len(dossiers) > 0:
            return fn(request, *args, **kwargs)
        else:
            msg = u"Vous n'avez pas le droit de consulter " \
                    u"ce dossier d'embauche."
            return redirect_interdiction(request, msg)

    return inner


def employe_dans_ma_region_ou_service(fn):
    """
    Test d'accès à un employé
    """
    def inner(request, *args, **kwargs):
        from project.dae.forms import _employe_choices
        liste_employes = _employe_choices(None, request)
        autorises = [k for k, nom in liste_employes]
        employe_key = kwargs.get('employe_key')
        if employe_key in autorises:
            return fn(request, *args, **kwargs)
        else:
            msg = u"Vous n'avez pas le droit de consulter cet employé."
            return redirect_interdiction(request, msg)

    return inner


def dossier_est_modifiable(fn):
    def inner(request, *args, **kwargs):
        dossier_id = kwargs.get('dossier_id', None)
        user_groupes = [g.name for g in request.user.groups.all()]
        if dossier_id is not None:
            dossier = dae.Dossier.objects.get(id=dossier_id)
            if not (dossier.etat in ETATS_EDITABLE and
                    (groups.DRH_NIVEAU_1 in user_groupes or groups.DRH_NIVEAU_2 in user_groupes or
                     dossier in dae.Dossier.objects \
                     .mes_choses_a_faire(request.user).all())):
                msg = u"Ce dossier d'embauche ne peut plus être modifié."
                return redirect_interdiction(request, msg)
        return fn(request, *args, **kwargs)
    return inner


def poste_est_modifiable(fn):
    def inner(request, *args, **kwargs):
        key = kwargs.get('key', None)
        if key is not None and key.split('-')[0] == 'dae':
            poste_id = key.split('-')[1]
            poste = dae.Poste.objects.get(id=poste_id)
            user_groupes = [g.name for g in request.user.groups.all()]
            if groups.DRH_NIVEAU_1 not in user_groupes and \
               groups.DRH_NIVEAU_2 not in user_groupes and \
               (poste.etat not in ETATS_EDITABLE \
                or poste not in dae.Poste.objects \
                .mes_choses_a_faire(request.user).all()):
                msg = u"Ce poste ne peut plus être modifié."
                return redirect_interdiction(request, msg)

        return fn(request, *args, **kwargs)
    return inner


def get_contrat(fn):
    """Ce décorateur s'attend à ce que le premier argument de la vue décorée
       soit l'ID d'un contrat. Il vérifie les permissions, puis transforme
       l'ID en objet Contrat."""
    def inner(request, contrat_id, *args, **kwargs):
        contrat = get_object_or_404(dae.Contrat, pk=contrat_id)
        dossier = contrat.dossier
        user_groupes = [g.name for g in request.user.groups.all()]
        dossier_de_ma_region =\
            len(dae.Dossier.objects.ma_region_ou_service(request.user).filter(id=dossier.id)) == 1

        if not (dossier.etat in ETATS_VALIDE and
                (groups.DRH_NIVEAU_1 in user_groupes or
                    groups.DRH_NIVEAU_2 in user_groupes or
                    dossier_de_ma_region)):
            return redirect_interdiction(request)
        return fn(request, contrat, *args, **kwargs)
    return inner