chore(admin): better token search in admin (#4497)

Flagsmith · Aug 14, 2024 · f9cc1b0 · f9cc1b0
1 parent a395a47
commit f9cc1b0
Showing 1 changed file with 11 additions and 0 deletions.
diff --git a/api/custom_auth/admin.py b/api/custom_auth/admin.py
@@ -0,0 +1,11 @@
+from django.contrib import admin
+from rest_framework.authtoken.admin import TokenAdmin
+from rest_framework.authtoken.models import TokenProxy
+
+
+class CustomTokenAdmin(TokenAdmin):
+    search_fields = ("user__email",)
+
+
+admin.site.unregister(TokenProxy)
+admin.site.register(TokenProxy, CustomTokenAdmin)