52 lines
1.7 KiB
Python
52 lines
1.7 KiB
Python
|
from django.contrib import admin
|
||
|
|
from django.contrib.admin.utils import quote
|
||
|
|
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 rest_framework.authtoken.models import Token, TokenProxy
|
||
|
|
|
||
|
|
User = get_user_model()
|
||
|
|
|
||
|
|
|
||
|
|
class TokenChangeList(ChangeList):
|
||
|
|
"""Map to matching User id"""
|
||
|
|
def url_for_result(self, result):
|
||
|
|
pk = result.user.pk
|
||
|
|
return reverse('admin:%s_%s_change' % (self.opts.app_label,
|
||
|
|
self.opts.model_name),
|
||
|
|
args=(quote(pk),),
|
||
|
|
current_app=self.model_admin.admin_site.name)
|
||
|
|
|
||
|
|
|
||
|
|
class TokenAdmin(admin.ModelAdmin):
|
||
|
|
list_display = ('key', 'user', 'created')
|
||
|
|
fields = ('user',)
|
||
|
|
ordering = ('-created',)
|
||
|
|
actions = None # Actions not compatible with mapped IDs.
|
||
|
|
|
||
|
|
def get_changelist(self, request, **kwargs):
|
||
|
|
return TokenChangeList
|
||
|
|
|
||
|
|
def get_object(self, request, object_id, from_field=None):
|
||
|
|
"""
|
||
|
|
Map from User ID to matching Token.
|
||
|
|
"""
|
||
|
|
queryset = self.get_queryset(request)
|
||
|
|
field = User._meta.pk
|
||
|
|
try:
|
||
|
|
object_id = field.to_python(object_id)
|
||
|
|
user = User.objects.get(**{field.name: object_id})
|
||
|
|
return queryset.get(user=user)
|
||
|
|
except (queryset.model.DoesNotExist, User.DoesNotExist, ValidationError, ValueError):
|
||
|
|
return None
|
||
|
|
|
||
|
|
def delete_model(self, request, obj):
|
||
|
|
# Map back to actual Token, since delete() uses pk.
|
||
|
|
token = Token.objects.get(key=obj.key)
|
||
|
|
return super().delete_model(request, token)
|
||
|
|
|
||
|
|
|
||
|
|
admin.site.register(TokenProxy, TokenAdmin)
|