Prefetch personal notes to avoid additional queries

aleksis/apps/alsijil/models.py

@@ -154,7 +154,9 @@ class Documentation(CalendarEvent):
                 datetime_end__gte=datetime_start,
                 amends__in=[e["REFERENCE_OBJECT"] for e in events],
             )
-            .prefetch_related("participations", "teachers")
+            .prefetch_related(
+                "participations", "teachers", "personal_notes", "personal_notes__extra_mark"
+            )
             .select_related("course", "subject")
         )
 

aleksis/apps/alsijil/schema/documentation.py

@@ -117,6 +117,11 @@ class DocumentationType(PermissionsTypeMixin, DjangoFilterMixin, DjangoObjectTyp
                     p for p in root.participations.all() if p.person == info.context.user.person
                 ]
             return []
+
+        # Annotate participations with prefetched documentation data for personal notes
+        for participation in root.participations.all():
+            participation._prefetched_documentation = root
+
         return root.participations.all()
 
 

aleksis/apps/alsijil/schema/participation_status.py

@@ -41,6 +41,12 @@ class ParticipationStatusType(
 
     @staticmethod
     def resolve_notes_with_extra_mark(root: ParticipationStatus, info, **kwargs):
+        if hasattr(root, "_prefetched_documentation"):
+            return [
+                p
+                for p in root._prefetched_documentation.personal_notes.all()
+                if p.person_id == root.person_id and p.extra_mark
+            ]
         return NewPersonalNote.objects.filter(
             person=root.person,
             documentation=root.related_documentation,
@@ -49,6 +55,12 @@ class ParticipationStatusType(
 
     @staticmethod
     def resolve_notes_with_note(root: ParticipationStatus, info, **kwargs):
+        if hasattr(root, "_prefetched_documentation"):
+            return [
+                p
+                for p in root._prefetched_documentation.personal_notes.all()
+                if p.person_id == root.person_id and p.note
+            ]
         return NewPersonalNote.objects.filter(
             person=root.person,
             documentation=root.related_documentation,