mirror of
https://github.com/encode/django-rest-framework.git
synced 2025-07-24 23:19:47 +03:00
Update admin.py
This commit is contained in:
parent
22d206c1e0
commit
6cb548e3be
|
@ -4,6 +4,7 @@ from django.contrib.admin.views.main import ChangeList
|
||||||
from django.contrib.auth import get_user_model
|
from django.contrib.auth import get_user_model
|
||||||
from django.core.exceptions import ValidationError
|
from django.core.exceptions import ValidationError
|
||||||
from django.urls import reverse
|
from django.urls import reverse
|
||||||
|
from django.contrib.admin.sites import AlreadyRegistered
|
||||||
|
|
||||||
from rest_framework.authtoken.models import Token, TokenProxy
|
from rest_framework.authtoken.models import Token, TokenProxy
|
||||||
|
|
||||||
|
@ -48,5 +49,8 @@ class TokenAdmin(admin.ModelAdmin):
|
||||||
token = Token.objects.get(key=obj.key)
|
token = Token.objects.get(key=obj.key)
|
||||||
return super().delete_model(request, token)
|
return super().delete_model(request, token)
|
||||||
|
|
||||||
|
try:
|
||||||
admin.site.register(TokenProxy, TokenAdmin)
|
admin.site.register(TokenProxy, TokenAdmin)
|
||||||
|
except AlreadyRegistered:
|
||||||
|
# handle already registered
|
||||||
|
pass
|
||||||
|
|
Loading…
Reference in New Issue
Block a user