Files
penracourses/anatomy/decorators.py
T
Ross d4ae7dbf6c .
2021-04-29 23:18:40 +01:00

14 lines
554 B
Python
Executable File

from django.core.exceptions import PermissionDenied
from .models import Exam, AnatomyQuestion
def user_is_author_or_anatomy_checker(function):
def wrap(request, *args, **kwargs):
question = AnatomyQuestion.objects.get(pk=kwargs['pk'])
if request.user in question.author.all() or request.user.groups.filter(name='anatomy_checker').exists():
return function(request, *args, **kwargs)
else:
raise PermissionDenied
wrap.__doc__ = function.__doc__
wrap.__name__ = function.__name__
return wrap