gestioCOF/gestioncof/decorators.py
2013-10-01 15:27:19 +02:00

20 lines
511 B
Python

from django_cas.decorators import user_passes_test
def is_cof(user):
try:
profile = user.get_profile()
return profile.is_cof
except:
return False
cof_required = user_passes_test(lambda u: is_cof(u))
cof_required_customdenied = user_passes_test(lambda u: is_cof(u), login_url = "cof-denied")
def is_buro(user):
try:
profile = user.get_profile()
return profile.is_buro
except:
return False
buro_required = user_passes_test(lambda u: is_buro(u))