project/dae/decorators.py

   1 # -*- encoding: utf-8 -*-
   2
   3 from django.contrib import messages
   4 from django.contrib.auth.decorators import user_passes_test
   5 from django.core.urlresolvers import reverse
   6 from django.http import HttpResponseRedirect
   7 from models import dae_groupes
   8 from project.dae import models as dae
   9 from project.rh_v1 import models as rh
  10
  11 def user_in_dae_groupes(user):
  12     """
  13     Teste si un user Django fait parti des groupes prédéfinis de DAE.
  14     """
  15     for g in user.groups.all():
  16         if g in dae_groupes:
  17             return True
  18     return False
  19
  20 def redirect_interdiction(request, msg=u"Vous n'avez pas accès à cette page"):
  21   """
  22   Redirection du la page de login avec un message d'erreur.
  23   """
  24   from django.conf import settings
  25   from django.contrib.auth import REDIRECT_FIELD_NAME
  26   from django.utils.http import urlquote
  27   login_url = settings.LOGIN_URL
  28   path = urlquote(request.get_full_path())
  29   tup = login_url, REDIRECT_FIELD_NAME, path
  30   messages.add_message(request, messages.ERROR, "Votre compte ne permet pas d'accéder à cette partie de l'application.")
  31   return HttpResponseRedirect('%s?%s=%s' % tup)
  32
  33
  34 def dae_groupe_requis(fn):
  35     """
  36     L'accès à la plateforme nécessite d'appartenir au moins à un groupe,
  37     ou d'être superuser. De cette manière l'autentification AUF fonctionnera
  38     toujours, il suffit de mettre le nouvel employé dans le groupe qui le concerne.
  39     """
  40     def inner(request, *args, **kwargs):
  41         user = request.user
  42
  43         if user.is_superuser or user_in_dae_groupes(user):
  44             return fn(request, *args, **kwargs)
  45         else :
  46             msg = u"Votre compte ne permet pas d'accéder à cette partie de l'application."
  47             return redirect_interdiction(request, msg)
  48     return inner
  49
  50
  51 def poste_dans_ma_region_ou_service(fn):
  52     """
  53     Test si le user connecté appartient bien à la même région ou service que le poste.
  54     """
  55     def inner(request, *args, **kwargs):
  56         user = request.user
  57         key = kwargs.get('key', None)
  58         try:
  59             source, id = key.split('-')
  60             if source == 'dae':
  61                 Poste = dae.Poste
  62             if source == 'rh':
  63                 Poste = rh.Poste
  64         except:
  65             id = key
  66             Poste = dae.Poste
  67
  68         postes = Poste.objects.ma_region_ou_service(user).filter(id=id)
  69         print postes
  70         if len(postes) > 0:
  71             return fn(request, *args, **kwargs)
  72         else :
  73             msg = u"Vous n'avez pas le droit de consulter ce poste."
  74             return redirect_interdiction(request, msg)
  75     return inner
  76
  77