| 123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051 |
- 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)
|
