.

sbas/filters.py

@@ -0,0 +1,62 @@
+import django_filters
+
+from .models import 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="anatomy_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="sba_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)
+    author = django_filters.ModelMultipleChoiceFilter(queryset=get_authors)
+    class Meta:
+        model = Question
+        fields = ("exams", "category", "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="sba_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="sba_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