Prefetch IndexedRegulationText and category parents

Enterprise-CMCS · Aug 26, 2024 · 45b2399 · 45b2399
1 parent 31ceccc
commit 45b2399
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/solution/backend/content_search/views.py b/solution/backend/content_search/views.py
@@ -13,7 +13,7 @@
 )
 from resources.utils import get_citation_filter, string_to_bool
 
-from .models import ContentIndex
+from .models import ContentIndex, IndexedRegulationText
 from .serializers import ContentSearchSerializer
 
 
@@ -133,9 +133,12 @@ def list(self, request, *args, **kwargs):
 
         # Prefetch all related data
         query = query.prefetch_related(
+            Prefetch("reg_text", IndexedRegulationText.objects.all()),
             Prefetch("resource", AbstractResource.objects.select_subclasses().prefetch_related(
                 Prefetch("cfr_citations", AbstractCitation.objects.select_subclasses()),
-                Prefetch("category", AbstractCategory.objects.select_subclasses()),
+                Prefetch("category", AbstractCategory.objects.select_subclasses().prefetch_related(
+                    Prefetch("parent", AbstractCategory.objects.select_subclasses()),
+                )),
                 Prefetch("subjects", Subject.objects.all()),
             )),
         )