|
| 1 | +from django.contrib import admin |
| 2 | +from django.contrib import messages |
| 3 | +from rest_framework_api_key.models import APIKey |
| 4 | +from rest_framework_api_key.helpers import generate_key |
| 5 | + |
| 6 | + |
| 7 | +class ApiKeyAdmin(admin.ModelAdmin): |
| 8 | + list_display = ('id', 'name', 'created', 'modified') |
| 9 | + |
| 10 | + fieldsets = ( |
| 11 | + ('Required Information', {'fields': ('name',)}), |
| 12 | + ('Additional Information', {'fields': ('key_message',)}), |
| 13 | + ) |
| 14 | + readonly_fields = ('key_message',) |
| 15 | + |
| 16 | + search_fields = ('id', 'name',) |
| 17 | + |
| 18 | + def has_delete_permission(self, request, obj=None): |
| 19 | + return False |
| 20 | + |
| 21 | + def key_message(self, obj): |
| 22 | + if obj.key: |
| 23 | + return "Hidden" |
| 24 | + return "The API Key will be generated once you click save." |
| 25 | + |
| 26 | + def save_model(self, request, obj, form, change): |
| 27 | + if not obj.key: |
| 28 | + obj.key = generate_key() |
| 29 | + messages.add_message(request, messages.WARNING, ('The API Key for %s is %s. Please note it since you will not be able to see it again.' % (obj.name, obj.key))) |
| 30 | + obj.save() |
| 31 | + |
| 32 | +admin.site.register(APIKey, ApiKeyAdmin) |
0 commit comments