securite accès vues poste

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

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

from django.contrib import messages
from django.contrib.auth.decorators import user_passes_test
from django.core.urlresolvers import reverse
from django.http import HttpResponseRedirect
from models import dae_groupes
from project.dae import models as dae
from project.rh_v1 import models as rh

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

def redirect_interdiction(request, msg=u"Vous n'avez pas accès à cette page"):
  """
  Redirection du la page de login avec un message d'erreur.
  """
  from django.conf import settings
  from django.contrib.auth import REDIRECT_FIELD_NAME
  from django.utils.http import urlquote
  login_url = settings.LOGIN_URL
  path = urlquote(request.get_full_path())
  tup = login_url, REDIRECT_FIELD_NAME, path
  messages.add_message(request, messages.ERROR, "Votre compte ne permet pas d'accéder à cette partie de l'application.")
  return HttpResponseRedirect('%s?%s=%s' % tup)


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 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)
        try:
            source, id = key.split('-')
            if source == 'dae':
                Poste = dae.Poste
            if source == 'rh':
                Poste = rh.Poste
        except:
            id = key
            Poste = dae.Poste

        postes = Poste.objects.ma_region_ou_service(user).filter(id=id)
        print postes
        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