Fix Token Admin page

IFRCGo · May 28, 2024 · 45ccb13 · 45ccb13
1 parent c32dd45
commit 45ccb13
Showing 1 changed file with 3 additions and 4 deletions.
diff --git a/api/admin.py b/api/admin.py
@@ -12,7 +12,7 @@
 from django.conf import settings
 from django_admin_listfilter_dropdown.filters import RelatedDropdownFilter
 from rest_framework.authtoken.admin import TokenAdmin
-from rest_framework.authtoken.models import Token
+from rest_framework.authtoken.models import Token, TokenProxy
 from reversion_compare.admin import CompareVersionAdmin
 # from reversion.models import Revision
 
@@ -58,9 +58,8 @@ def get_inline_instances(self, request, obj=None):
 class GoTokenAdmin(TokenAdmin):
     search_fields = ('user__username', 'user__email',)
 
-
-# admin.site.unregister(Token)
-# admin.site.register(Token, GoTokenAdmin)
+admin.site.unregister(TokenProxy)
+admin.site.register(TokenProxy, GoTokenAdmin)
 
 
 class HasRelatedEventFilter(admin.SimpleListFilter):