answers filter 추가

Author: clogic29

apps/answer/filters.py

@@ -1,5 +1,9 @@
+from django.db.models import Q
 from ninja import FilterSchema
 
 
 class AnswerFilter(FilterSchema):
-    pass
+    models: list[str]
+
+    def filter_models(self, data):
+        return Q(model_answers__model_id__in=data)

apps/answer/views.py

@@ -1,6 +1,8 @@
-from ninja import Router
+from django.db.models import Count, Q
+from ninja import Router, Query
 from ninja.pagination import paginate
 
+from apps.answer.filters import AnswerFilter
 from apps.answer.models import UserAnswer
 from apps.answer.schemas import UserAnswerSchema
 from contrib.django.ninja.pagination import ContribPageNumberPagination
@@ -10,6 +12,13 @@
 
 @router.get('/answers', response=list[UserAnswerSchema])
 @paginate(ContribPageNumberPagination, page_size=4)
-def get_answers(request):
-    answers = UserAnswer.objects.all()
+def get_answers(request, filter: AnswerFilter = Query(...)):
+    answers = UserAnswer.objects.prefetch_related("model_answers").annotate(
+        matching_names=Count(
+            'model_answers',
+            filter=Q(model_answers__model_id__in=filter.models)
+        )
+    ).filter(
+        matching_names=len(filter.models)
+    )
     return answers