This commit is contained in:
Ross
2021-04-28 10:57:29 +01:00
parent 94538521d2
commit 2acf5a1f48
+15 -1
View File
@@ -3,6 +3,7 @@ import django_filters
from .models import AnatomyQuestion, CidUserAnswer, Exam
#from generic.filters import UserAnswerFilter
from django.contrib.auth.models import User
def get_exams(request):
@@ -16,14 +17,27 @@ def get_exams(request):
return queryset
def get_authors(request):
if request is None:
return User.objects.none()
if not request.user.groups.filter(name="anatomy_checker").exists():
queryset = User.objects.filter(id=request.user.id)
else:
queryset = User.objects.all()
return queryset
class AnatomyQuestionFilter(django_filters.FilterSet):
exams = django_filters.ModelMultipleChoiceFilter(queryset=get_exams)
author = django_filters.ModelMultipleChoiceFilter(queryset=get_authors)
class Meta:
model = AnatomyQuestion
fields = ("question_type",
"examination",
"modality", "region",
"body_part", "created_date", "open_access", "author")
"body_part", "created_date", "open_access")
def __init__(self, data=None, queryset=None, prefix=None, strict=None, user=None, request=None):
if not request.user.groups.filter(name="anatomy_checker").exists():