mirror of
https://github.com/netbox-community/netbox.git
synced 2024-05-10 07:54:54 +00:00
60 lines
2.4 KiB
Python
60 lines
2.4 KiB
Python
from django.contrib import admin, messages
|
|
from django.shortcuts import redirect, render
|
|
|
|
from netbox.admin import admin_site
|
|
from .forms import ActivateUserKeyForm
|
|
from .models import UserKey
|
|
|
|
|
|
@admin.register(UserKey, site=admin_site)
|
|
class UserKeyAdmin(admin.ModelAdmin):
|
|
actions = ['activate_selected']
|
|
list_display = ['user', 'is_filled', 'is_active', 'created']
|
|
fields = ['user', 'public_key', 'is_active', 'last_updated']
|
|
readonly_fields = ['user', 'is_active', 'last_updated']
|
|
|
|
def get_readonly_fields(self, request, obj=None):
|
|
# Don't allow a user to modify an existing public key directly.
|
|
if obj and obj.public_key:
|
|
return ['public_key'] + self.readonly_fields
|
|
return self.readonly_fields
|
|
|
|
def get_actions(self, request):
|
|
# Bulk deletion is disabled at the manager level, so remove the action from the admin site for this model.
|
|
actions = super().get_actions(request)
|
|
if 'delete_selected' in actions:
|
|
del actions['delete_selected']
|
|
if not request.user.has_perm('secrets.activate_userkey'):
|
|
del actions['activate_selected']
|
|
return actions
|
|
|
|
def activate_selected(modeladmin, request, queryset):
|
|
"""
|
|
Enable bulk activation of UserKeys
|
|
"""
|
|
try:
|
|
my_userkey = UserKey.objects.get(user=request.user)
|
|
except UserKey.DoesNotExist:
|
|
messages.error(request, "You do not have an active User Key.")
|
|
return redirect('admin:secrets_userkey_changelist')
|
|
|
|
if 'activate' in request.POST:
|
|
form = ActivateUserKeyForm(request.POST)
|
|
if form.is_valid():
|
|
master_key = my_userkey.get_master_key(form.cleaned_data['secret_key'])
|
|
if master_key is not None:
|
|
for uk in form.cleaned_data['_selected_action']:
|
|
uk.activate(master_key)
|
|
return redirect('admin:secrets_userkey_changelist')
|
|
else:
|
|
messages.error(
|
|
request, "Invalid private key provided. Unable to retrieve master key.", extra_tags='error'
|
|
)
|
|
else:
|
|
form = ActivateUserKeyForm(initial={'_selected_action': request.POST.getlist(admin.ACTION_CHECKBOX_NAME)})
|
|
|
|
return render(request, 'activate_keys.html', {
|
|
'form': form,
|
|
})
|
|
activate_selected.short_description = "Activate selected user keys"
|