.

2021-09-30 20:24:50 +01:00
parent 5e6a48480f
commit fd99f51d97
8 changed files with 235 additions and 18 deletions
@@ -0,0 +1,63 @@
+import django_filters
+
+from .models import Category, Question, CidUserAnswer, Exam
+
+#from generic.filters import UserAnswerFilter
+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="physics_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="physics_checker").exists():
+        queryset = User.objects.filter(id=request.user.id)
+    else:
+        queryset = User.objects.all()
+
+    return queryset
+
+
+class QuestionFilter(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")
+    category = django_filters.ModelChoiceFilter(queryset=Category.objects.all(),null_label="Uncategorised")
+    class Meta:
+        model = Question
+        fields = ("exams", "created_date")
+
+    def __init__(self, data=None, queryset=None, prefix=None, strict=None, user=None, request=None):
+        if not request.user.groups.filter(name="physics_checker").exists():
+            queryset = queryset.filter(open_access=True) | queryset.filter(author__id=request.user.id)
+        super(QuestionFilter, self).__init__(data=data, queryset=queryset, prefix=prefix, request=request)
+        pass
+
+class UserAnswerFilter(django_filters.FilterSet):
+    class Meta:
+        model= CidUserAnswer
+        fields = (
+            "cid",
+            #"question",
+            "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="physics_checker").exists():
+            queryset = queryset.filter(open_access=True) | queryset.filter(author__id=request.user.id)
+        super(UserAnswerFilter, self).__init__(data=data, queryset=queryset, prefix=prefix, request=request)
+        pass