diff --git a/rapids/filters.py b/rapids/filters.py index 4a9cf44c..5404c707 100755 --- a/rapids/filters.py +++ b/rapids/filters.py @@ -1,12 +1,23 @@ import django_filters -from .models import Rapid, CidUserAnswer +from .models import Rapid, CidUserAnswer, Exam +def get_exams(request): + if request is None: + return Exam.objects.none() + + if not request.user.groups.filter(name="rapid_checker").exists(): + queryset = Exam.filter(author__id=request.user.id) + else: + queryset = Exam.objects.all() + + return queryset class RapidFilter(django_filters.FilterSet): + exams = django_filters.ModelMultipleChoiceFilter(queryset=get_exams) class Meta: model = Rapid - fields = ("normal", "abnormality", "exams", "region", "examination", + fields = ("normal", "abnormality", "region", "examination", "laterality", #"site", "created_date", "open_access", "author")