Update admin.py

This commit is contained in:
L-iet 2023-01-31 12:48:27 -08:00 committed by GitHub
parent 22d206c1e0
commit 6cb548e3be
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -4,6 +4,7 @@ from django.contrib.admin.views.main import ChangeList
from django.contrib.auth import get_user_model
from django.core.exceptions import ValidationError
from django.urls import reverse
from django.contrib.admin.sites import AlreadyRegistered
from rest_framework.authtoken.models import Token, TokenProxy
@ -48,5 +49,8 @@ class TokenAdmin(admin.ModelAdmin):
token = Token.objects.get(key=obj.key)
return super().delete_model(request, token)
admin.site.register(TokenProxy, TokenAdmin)
try:
admin.site.register(TokenProxy, TokenAdmin)
except AlreadyRegistered:
# handle already registered
pass