Files
penracourses/rapids/filters.py
T
Ross be643c6ade .
2021-04-26 14:03:31 +01:00

48 lines
1.7 KiB
Python
Executable File

import django_filters
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", "region", "examination",
"laterality",
#"site",
"created_date", "open_access", "author")
def __init__(self, data=None, queryset=None, prefix=None, strict=None, user=None, request=None):
if not request.user.groups.filter(name="rapid_checker").exists():
queryset = queryset.filter(open_access=True) | queryset.filter(author__id=request.user.id)
super(RapidFilter, self).__init__(data=data, queryset=queryset, prefix=prefix, request=request)
pass
class RapidUserAnswerFilter(django_filters.FilterSet):
class Meta:
model = CidUserAnswer
fields = (
"cid",
"normal",
"answer",
#"answer_compare",
"exam",
"created",
"updated",
)
def __init__(self, data=None, queryset=None, prefix=None, strict=None, user=None, request=None):
if not request.user.groups.filter(name="rapid_checker").exists():
queryset = queryset.filter(open_access=True) | queryset.filter(author__id=request.user.id)
super(RapidUserAnswerFilter, self).__init__(data=data, queryset=queryset, prefix=prefix, request=request)
pass