Admin: raw id for user on auth Token model

unicef · Sep 18, 2024 · d1ecfb3 · d1ecfb3
1 parent dc5150a
commit d1ecfb3
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/src/etools/applications/users/admin.py b/src/etools/applications/users/admin.py
@@ -23,6 +23,8 @@
 from django_tenants.admin import TenantAdminMixin
 from django_tenants.postgresql_backend.base import FakeTenant
 from django_tenants.utils import get_public_schema_name
+from rest_framework.authtoken.admin import TokenAdmin
+from rest_framework.authtoken.models import TokenProxy
 from unicef_snapshot.admin import ActivityInline, SnapshotModelAdmin
 
 from etools.applications.funds.tasks import sync_all_delegated_frs, sync_country_delegated_fr
@@ -531,10 +533,18 @@ def has_change_permission(self, request, obj=None):
         return False
 
 
+class CustomTokenAdmin(TokenAdmin):
+    raw_id_fields = ('user',)
+
+
 # Re-register UserAdmin
 admin.site.register(get_user_model(), UserAdminPlus)
 admin.site.register(UserProfile, ProfileAdmin)
 admin.site.register(Country, CountryAdmin)
 admin.site.register(WorkspaceCounter)
 admin.site.register(Realm, RealmAdmin)
 admin.site.register(StagedUser, StagedUserAdmin)
+
+
+admin.site.unregister(TokenProxy)
+admin.site.register(TokenProxy, CustomTokenAdmin)