securite accès vues poste
[auf_rh_dae.git] / project / dae / decorators.py
CommitLineData
5633fa41
OL
1# -*- encoding: utf-8 -*-
2
3from django.contrib import messages
4from django.contrib.auth.decorators import user_passes_test
5from django.core.urlresolvers import reverse
6from django.http import HttpResponseRedirect
7from models import dae_groupes
8from project.dae import models as dae
9from project.rh_v1 import models as rh
10
11def 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
20def 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
34def 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
51def 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