NeoAnd/netbox
1
0
Fork 0
mirror of https://github.com/netbox-community/netbox.git synced 2025-08-24 08:25:17 -06:00
Code Issues Actions 4 Packages Projects Releases Wiki Activity

Establish parity with DeviceRenderConfigView

Browse Source
This commit is contained in:
Jeremy Stretch 2023-08-14 15:08:17 -04:00
parent 73935be764
commit c0fae30684
1 changed files with 33 additions and 16 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

49
netbox/virtualization/views.py
View File

@ -4,6 +4,7 @@ from collections import defaultdict
from django.contrib import messages from django.contrib import messages
from django.db import transaction from django.db import transaction
from django.db.models import Prefetch, Sum from django.db.models import Prefetch, Sum
from django.http import HttpResponse
from django.shortcuts import get_object_or_404, redirect, render from django.shortcuts import get_object_or_404, redirect, render
from django.urls import reverse from django.urls import reverse
from django.utils.translation import gettext as _ from django.utils.translation import gettext as _
@ -380,6 +381,17 @@ class VirtualMachineInterfacesView(generic.ObjectChildrenView):
) )
@register_model_view(VirtualMachine, 'configcontext', path='config-context')
class VirtualMachineConfigContextView(ObjectConfigContextView):
queryset = VirtualMachine.objects.annotate_config_context_data()
base_template = 'virtualization/virtualmachine.html'
tab = ViewTab(
label=_('Config Context'),
permission='extras.view_configcontext',
weight=2000
)
@register_model_view(VirtualMachine, 'render-config') @register_model_view(VirtualMachine, 'render-config')
class VirtualMachineRenderConfigView(generic.ObjectView): class VirtualMachineRenderConfigView(generic.ObjectView):
queryset = VirtualMachine.objects.all() queryset = VirtualMachine.objects.all()
@ -387,15 +399,31 @@ class VirtualMachineRenderConfigView(generic.ObjectView):
tab = ViewTab( tab = ViewTab(
label=_('Render Config'), label=_('Render Config'),
permission='extras.view_configtemplate', permission='extras.view_configtemplate',
weight=2000 weight=2100
) )
def get(self, request, **kwargs):
instance = self.get_object(**kwargs)
context = self.get_extra_context(request, instance)
# If a direct export has been requested, return the rendered template content as a
# downloadable file.
if request.GET.get('export'):
response = HttpResponse(context['rendered_config'], content_type='text')
filename = f"{instance.name or 'config'}.txt"
response['Content-Disposition'] = f'attachment; filename="{filename}"'
return response
return render(request, self.get_template_name(), {
'object': instance,
'tab': self.tab,
**context,
})
def get_extra_context(self, request, instance): def get_extra_context(self, request, instance):
# Compile context data # Compile context data
context_data = { context_data = instance.get_config_context()
'virtualmachine': instance, context_data.update({'virtualmachine': instance})
}
context_data.update(**instance.get_config_context())
# Render the config template # Render the config template
rendered_config = None rendered_config = None
@ -413,17 +441,6 @@ class VirtualMachineRenderConfigView(generic.ObjectView):
} }
@register_model_view(VirtualMachine, 'configcontext', path='config-context')
class VirtualMachineConfigContextView(ObjectConfigContextView):
queryset = VirtualMachine.objects.annotate_config_context_data()
base_template = 'virtualization/virtualmachine.html'
tab = ViewTab(
label=_('Config Context'),
permission='extras.view_configcontext',
weight=2100
)
@register_model_view(VirtualMachine, 'edit') @register_model_view(VirtualMachine, 'edit')
class VirtualMachineEditView(generic.ObjectEditView): class VirtualMachineEditView(generic.ObjectEditView):
queryset = VirtualMachine.objects.all() queryset = VirtualMachine.objects.all()