From 5723a59b550bb655f007267b80f352a6a3aeb1b6 Mon Sep 17 00:00:00 2001
From: Wilson Gaturu <wilsongaturu@gmail.com>
Date: Mon, 16 Oct 2023 11:52:51 +0300
Subject: [PATCH] fixes n+1 in judgment admin

---
 peachjam/admin.py | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/peachjam/admin.py b/peachjam/admin.py
index 3e62a247b..8a02fe810 100644
--- a/peachjam/admin.py
+++ b/peachjam/admin.py
@@ -762,6 +762,11 @@ 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")
+        return super().formfield_for_foreignkey(db_field, request, **kwargs)
+
 
 @admin.register(Predicate)
 class PredicateAdmin(admin.ModelAdmin):