mirror of
https://github.com/netbox-community/netbox.git
synced 2024-05-10 07:54:54 +00:00
Relocate and group generic views
This commit is contained in:
@@ -7,9 +7,7 @@ from django.shortcuts import get_object_or_404, redirect, render
|
||||
from django.utils.html import escape
|
||||
from django.utils.safestring import mark_safe
|
||||
|
||||
from utilities.views import (
|
||||
BulkDeleteView, BulkEditView, BulkImportView, ObjectView, ObjectDeleteView, ObjectEditView, ObjectListView,
|
||||
)
|
||||
from netbox.views import generic
|
||||
from . import filters, forms, tables
|
||||
from .models import SecretRole, Secret, SessionKey, UserKey
|
||||
|
||||
@@ -28,27 +26,27 @@ def get_session_key(request):
|
||||
# Secret roles
|
||||
#
|
||||
|
||||
class SecretRoleListView(ObjectListView):
|
||||
class SecretRoleListView(generic.ObjectListView):
|
||||
queryset = SecretRole.objects.annotate(secret_count=Count('secrets')).order_by(*SecretRole._meta.ordering)
|
||||
table = tables.SecretRoleTable
|
||||
|
||||
|
||||
class SecretRoleEditView(ObjectEditView):
|
||||
class SecretRoleEditView(generic.ObjectEditView):
|
||||
queryset = SecretRole.objects.all()
|
||||
model_form = forms.SecretRoleForm
|
||||
|
||||
|
||||
class SecretRoleDeleteView(ObjectDeleteView):
|
||||
class SecretRoleDeleteView(generic.ObjectDeleteView):
|
||||
queryset = SecretRole.objects.all()
|
||||
|
||||
|
||||
class SecretRoleBulkImportView(BulkImportView):
|
||||
class SecretRoleBulkImportView(generic.BulkImportView):
|
||||
queryset = SecretRole.objects.all()
|
||||
model_form = forms.SecretRoleCSVForm
|
||||
table = tables.SecretRoleTable
|
||||
|
||||
|
||||
class SecretRoleBulkDeleteView(BulkDeleteView):
|
||||
class SecretRoleBulkDeleteView(generic.BulkDeleteView):
|
||||
queryset = SecretRole.objects.annotate(secret_count=Count('secrets')).order_by(*SecretRole._meta.ordering)
|
||||
table = tables.SecretRoleTable
|
||||
|
||||
@@ -57,7 +55,7 @@ class SecretRoleBulkDeleteView(BulkDeleteView):
|
||||
# Secrets
|
||||
#
|
||||
|
||||
class SecretListView(ObjectListView):
|
||||
class SecretListView(generic.ObjectListView):
|
||||
queryset = Secret.objects.all()
|
||||
filterset = filters.SecretFilterSet
|
||||
filterset_form = forms.SecretFilterForm
|
||||
@@ -65,7 +63,7 @@ class SecretListView(ObjectListView):
|
||||
action_buttons = ('import', 'export')
|
||||
|
||||
|
||||
class SecretView(ObjectView):
|
||||
class SecretView(generic.ObjectView):
|
||||
queryset = Secret.objects.all()
|
||||
|
||||
def get(self, request, pk):
|
||||
@@ -77,7 +75,7 @@ class SecretView(ObjectView):
|
||||
})
|
||||
|
||||
|
||||
class SecretEditView(ObjectEditView):
|
||||
class SecretEditView(generic.ObjectEditView):
|
||||
queryset = Secret.objects.all()
|
||||
model_form = forms.SecretForm
|
||||
template_name = 'secrets/secret_edit.html'
|
||||
@@ -146,11 +144,11 @@ class SecretEditView(ObjectEditView):
|
||||
})
|
||||
|
||||
|
||||
class SecretDeleteView(ObjectDeleteView):
|
||||
class SecretDeleteView(generic.ObjectDeleteView):
|
||||
queryset = Secret.objects.all()
|
||||
|
||||
|
||||
class SecretBulkImportView(BulkImportView):
|
||||
class SecretBulkImportView(generic.BulkImportView):
|
||||
queryset = Secret.objects.all()
|
||||
model_form = forms.SecretCSVForm
|
||||
table = tables.SecretTable
|
||||
@@ -197,14 +195,14 @@ class SecretBulkImportView(BulkImportView):
|
||||
})
|
||||
|
||||
|
||||
class SecretBulkEditView(BulkEditView):
|
||||
class SecretBulkEditView(generic.BulkEditView):
|
||||
queryset = Secret.objects.prefetch_related('role')
|
||||
filterset = filters.SecretFilterSet
|
||||
table = tables.SecretTable
|
||||
form = forms.SecretBulkEditForm
|
||||
|
||||
|
||||
class SecretBulkDeleteView(BulkDeleteView):
|
||||
class SecretBulkDeleteView(generic.BulkDeleteView):
|
||||
queryset = Secret.objects.prefetch_related('role')
|
||||
filterset = filters.SecretFilterSet
|
||||
table = tables.SecretTable
|
||||
|
Reference in New Issue
Block a user