Ross
103 lines
2.7 KiB
Python
Executable File
103 lines
2.7 KiB
Python
Executable File
import django_filters
|
|
|
|
from .models import Rapid, UserAnswer, Exam
|
|
from django.contrib.auth.models import User
|
|
|
|
|
|
def get_exams(request):
|
|
if request is None:
|
|
return Exam.objects.none()
|
|
|
|
if not request.user.groups.filter(name="rapid_checker").exists():
|
|
queryset = Exam.objects.filter(author__id=request.user.id)
|
|
else:
|
|
queryset = Exam.objects.all()
|
|
|
|
return queryset
|
|
|
|
|
|
def get_authors(request):
|
|
if request is None:
|
|
return User.objects.none()
|
|
|
|
if not request.user.groups.filter(name="rapid_checker").exists():
|
|
queryset = User.objects.filter(id=request.user.id)
|
|
else:
|
|
queryset = User.objects.all()
|
|
|
|
return queryset
|
|
|
|
|
|
class RapidFilter(django_filters.FilterSet):
|
|
exams = django_filters.ModelMultipleChoiceFilter(
|
|
queryset=get_exams, null_label="No exam"
|
|
)
|
|
author = django_filters.ModelMultipleChoiceFilter(
|
|
queryset=get_authors, null_label="No author"
|
|
)
|
|
|
|
class Meta:
|
|
model = Rapid
|
|
fields = {
|
|
"normal": ["exact"],
|
|
"abnormality": ["exact"],
|
|
"region": ["exact"],
|
|
"examination": ["exact"],
|
|
"laterality": ["exact"],
|
|
# "site",
|
|
"created_date": ["exact"],
|
|
"open_access": ["exact"],
|
|
}
|
|
|
|
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 = UserAnswer
|
|
fields = (
|
|
"cid",
|
|
"user",
|
|
"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.is_superuser:
|
|
if not request.user.groups.filter(name="rapid_checker").exists():
|
|
queryset = queryset.filter(exam__open_access=True) | queryset.filter(
|
|
exam__author__id=request.user.id
|
|
)
|
|
super(RapidUserAnswerFilter, self).__init__(
|
|
data=data, queryset=queryset, prefix=prefix, request=request
|
|
)
|
|
pass
|