Add list_select_related for performance improvement

jazzband · May 25, 2024 · 7e3a177 · 7e3a177
1 parent 3b71b89
commit 7e3a177
Showing 1 changed file with 2 additions and 10 deletions.
diff --git a/rest_framework_simplejwt/token_blacklist/admin.py b/rest_framework_simplejwt/token_blacklist/admin.py
@@ -24,13 +24,9 @@ class OutstandingTokenAdmin(admin.ModelAdmin):
         "user__id",
         "jti",
     )
+    list_select_related = ("user", )
     ordering = ("user",)
 
-    def get_queryset(self, *args, **kwargs) -> QuerySet:
-        qs = super().get_queryset(*args, **kwargs)
-
-        return qs.select_related("user")
-
     # Read-only behavior defined below
     actions = None
 
@@ -66,13 +62,9 @@ class BlacklistedTokenAdmin(admin.ModelAdmin):
         "token__user__id",
         "token__jti",
     )
+    list_select_related = ("token__user",)
     ordering = ("token__user",)
 
-    def get_queryset(self, *args, **kwargs) -> QuerySet:
-        qs = super().get_queryset(*args, **kwargs)
-
-        return qs.select_related("token__user")
-
     def token_jti(self, obj: BlacklistedToken) -> str:
         return obj.token.jti