Ross
32 lines
1.1 KiB
Python
Executable File
32 lines
1.1 KiB
Python
Executable File
from django.core.exceptions import PermissionDenied
|
|
from .models import Exam, Rapid
|
|
|
|
def user_is_author_or_rapid_checker(function):
|
|
def wrap(request, *args, **kwargs):
|
|
if "pk" in kwargs:
|
|
question_id = kwargs["pk"]
|
|
elif "question_id" in kwargs:
|
|
question_id = kwargs["question_id"]
|
|
else:
|
|
raise PermissionDenied
|
|
|
|
question = Rapid.objects.get(pk=question_id)
|
|
if request.user in question.author.all() or request.user.groups.filter(name='rapid_checker').exists():
|
|
return function(request, *args, **kwargs)
|
|
else:
|
|
raise PermissionDenied
|
|
wrap.__doc__ = function.__doc__
|
|
wrap.__name__ = function.__name__
|
|
return wrap
|
|
|
|
|
|
def user_is_exam_author_or_rapid_checker(function):
|
|
def wrap(request, *args, **kwargs):
|
|
exam = Exam.objects.get(pk=kwargs['pk'])
|
|
if request.user in exam.author.all() or request.user.groups.filter(name='rapid_checker').exists():
|
|
return function(request, *args, **kwargs)
|
|
else:
|
|
raise PermissionDenied
|
|
wrap.__doc__ = function.__doc__
|
|
wrap.__name__ = function.__name__
|
|
return wrap |