fixes n+1 in judgment admin

laws-africa · Oct 16, 2023 · 8c7b0ff · 8c7b0ff
1 parent 1c87a87
commit 8c7b0ff
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/peachjam/admin.py b/peachjam/admin.py
@@ -762,6 +762,16 @@ def get_fieldsets(self, request, obj=None):
 
         return fieldsets
 
+    def formfield_for_foreignkey(self, db_field, request, **kwargs):
+        if db_field.name == "registry":
+            kwargs["queryset"] = CourtRegistry.objects.select_related("court").all()
+        return super().formfield_for_foreignkey(db_field, request, **kwargs)
+
+    def get_queryset(self, request):
+        qs = super().get_queryset(request)
+        print(qs.count())
+        return qs.select_related("court", "registry__court")
+
 
 @admin.register(Predicate)
 class PredicateAdmin(admin.ModelAdmin):