Fix N+1 issue in ops-learning

Author: szabozoltan69

assets

@@ -20,7 +20,7 @@
 from rest_framework.response import Response
 from rest_framework.settings import api_settings
 
-from api.models import Country, Region
+from api.models import AppealDocument, Country, Region
 from deployments.models import SectorTag
 from main.permissions import DenyGuestUserMutationPermission, DenyGuestUserPermission
 from main.utils import SpreadSheetContentNegotiation
@@ -1202,6 +1202,7 @@ def get_queryset(self):
             return qs.select_related(
                 "appeal_code",
             ).prefetch_related(
+                "appeal_code__event__appeals",
                 "sector",
                 "sector_validated",
                 "organization",
@@ -1216,6 +1217,7 @@ def get_queryset(self):
                 "appeal_code",
             )
             .prefetch_related(
+                "appeal_code__event__appeals",
                 "sector",
                 "sector_validated",
                 "organization",
@@ -1226,6 +1228,26 @@ def get_queryset(self):
             )
         )
 
+    def get_serializer_context(self):
+        context = super().get_serializer_context()
+        if getattr(self, "swagger_fake_view", False):
+            return context
+        view_action = getattr(self, "action", None)
+        if view_action == "list":
+            queryset = self.filter_queryset(self.get_queryset())
+            appeal_document_ids = list(
+                queryset.exclude(appeal_document_id__isnull=True).values_list("appeal_document_id", flat=True).distinct()
+            )
+            if appeal_document_ids:
+                context["appeal_documents_map"] = AppealDocument.objects.in_bulk(appeal_document_ids)
+        elif view_action == "retrieve":
+            lookup_kwarg = self.lookup_url_kwarg or self.lookup_field
+            if lookup_kwarg in self.kwargs:
+                obj = self.get_object()
+                if obj.appeal_document_id:
+                    context["appeal_documents_map"] = AppealDocument.objects.in_bulk([obj.appeal_document_id])
+        return context
+
     def get_serializer_class(self):
         if self.request.method == "GET":
             request_format = self.request.GET.get("format", "json")

per/drf_views.py

@@ -1054,14 +1054,20 @@ class Meta:
         fields = "__all__"
         read_only_fields = ("created_at", "modified_at")
 
-    @staticmethod
-    def get_document_url(obj):
-        if obj.appeal_document_id and (document := AppealDocument.objects.filter(id=obj.appeal_document_id).first()):
+    def _get_document(self, obj):
+        if not obj.appeal_document_id:
+            return None
+        document_map = self.context.get("appeal_documents_map")
+        if document_map is not None:
+            return document_map.get(obj.appeal_document_id)
+        return AppealDocument.objects.filter(id=obj.appeal_document_id).first()
+
+    def get_document_url(self, obj):
+        if document := self._get_document(obj):
             return document.document_url
 
-    @staticmethod
-    def get_document_name(obj):
-        if obj.appeal_document_id and (document := AppealDocument.objects.filter(id=obj.appeal_document_id).first()):
+    def get_document_name(self, obj):
+        if document := self._get_document(obj):
             return document.name
 
 
@@ -1088,14 +1094,20 @@ class Meta:
         )
         exclude = ("learning", "type", "organization", "sector", "per_component")
 
-    @staticmethod
-    def get_document_url(obj):
-        if obj.appeal_document_id and (document := AppealDocument.objects.filter(id=obj.appeal_document_id).first()):
+    def _get_document(self, obj):
+        if not obj.appeal_document_id:
+            return None
+        document_map = self.context.get("appeal_documents_map")
+        if document_map is not None:
+            return document_map.get(obj.appeal_document_id)
+        return AppealDocument.objects.filter(id=obj.appeal_document_id).first()
+
+    def get_document_url(self, obj):
+        if document := self._get_document(obj):
             return document.document_url
 
-    @staticmethod
-    def get_document_name(obj):
-        if obj.appeal_document_id and (document := AppealDocument.objects.filter(id=obj.appeal_document_id).first()):
+    def get_document_name(self, obj):
+        if document := self._get_document(obj):
             return document.name