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

14438 review comments retain old urls

Browse Source
This commit is contained in:
Arthur 2024-02-13 16:02:03 -08:00
parent 613f9f2354
commit ba8329eb4a
2 changed files with 86 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
netbox/extras/urls.py
View File

@ -121,9 +121,14 @@ urlpatterns = [
path('scripts/add/', views.ScriptModuleCreateView.as_view(), name='scriptmodule_add'),
path('scripts/results/<int:job_pk>/', views.ScriptResultView.as_view(), name='script_result'),
path('scripts/<int:pk>/', include(get_model_urls('extras', 'scriptmodule'))),
path('scripts/class/<int:pk>/', views.ScriptView.as_view(), name='script'),
path('scripts/class/<int:pk>/source/', views.ScriptSourceView.as_view(), name='script_source'),
path('scripts/class/<int:pk>/jobs/', views.ScriptJobsView.as_view(), name='script_jobs'),
path('scripts/<str:module>/<str:name>/', views.ScriptModuleView.as_view(), name='script_module'),
path('scripts/<str:module>/<str:name>/source/', views.ScriptModuleSourceView.as_view(), name='script_module_source'),
path('scripts/<str:module>/<str:name>/jobs/', views.ScriptModuleJobsView.as_view(), name='script_module_jobs'),
path('script-modules/<int:pk>/', views.ScriptView.as_view(), name='script'),
path('scripts-modules/<int:pk>/source/', views.ScriptSourceView.as_view(), name='script_source'),
path('scripts-modules/<int:pk>/jobs/', views.ScriptJobsView.as_view(), name='script_jobs'),
# Markdown
path('render/markdown/', views.RenderMarkdownView.as_view(), name="render_markdown"),

81
netbox/extras/views.py
View File

@ -1042,7 +1042,8 @@ def get_script_module(module, request):
return get_object_or_404(ScriptModule.objects.restrict(request.user), file_path__regex=f"^{module}\\.")
class BaseScriptView(ContentTypePermissionRequiredMixin, View):
class BaseScriptView(ContentTypePermissionRequiredMixin, generic.ObjectView):
queryset = Script.objects.all()
script = None
script_class = None
jobs = None
@ -1050,8 +1051,7 @@ class BaseScriptView(ContentTypePermissionRequiredMixin, View):
def get_required_permission(self):
return 'extras.view_script'
def get_script(self, request, pk):
self.script = get_object_or_404(Script.objects.all(), pk=pk)
def _init_vars(self, request):
if self.script.python_class:
self.script_class = self.script.python_class()
else:
@ -1065,6 +1065,15 @@ class BaseScriptView(ContentTypePermissionRequiredMixin, View):
self.jobs = self.script.jobs.all()
return None
def get_script(self, request, pk):
self.script = get_object_or_404(Script.objects.all(), pk=pk)
return self._init_vars(request)
def get_script_by_module_name(self, request, module, name):
module = get_script_module(module, request)
self.script = get_object_or_404(Script.objects.all(), module=module, name=name)
return self._init_vars(request)
class ScriptView(BaseScriptView):
@ -1123,6 +1132,25 @@ class ScriptView(BaseScriptView):
})
class ScriptModuleView(ScriptView):
def get(self, request, module, name):
if ret := self.get_script_by_module_name(request, module, name):
return ret
form = None
if self.script_class:
form = self.script_class.as_form(initial=normalize_querydict(request.GET))
return render(request, 'extras/script.html', {
'job_count': self.jobs.count(),
'module': self.script.module,
'script': self.script,
'script_class': self.script_class,
'form': form,
})
class ScriptSourceView(BaseScriptView):
def get(self, request, pk):
@ -1138,6 +1166,21 @@ class ScriptSourceView(BaseScriptView):
})
class ScriptModuleSourceView(ScriptSourceView):
def get(self, request, module, name):
if ret := self.get_script_by_module_name(request, module, name):
return ret
return render(request, 'extras/script/source.html', {
'job_count': self.jobs.count(),
'module': self.script.module,
'script': self.script,
'script_class': self.script_class,
'tab': 'source',
})
class ScriptJobsView(ContentTypePermissionRequiredMixin, View):
script = None
script_class = None
@ -1148,6 +1191,38 @@ class ScriptJobsView(ContentTypePermissionRequiredMixin, View):
def get(self, request, pk):
self.script = get_object_or_404(Script.objects.all(), pk=pk)
if self.script.python_class:
self.script_class = self.script.python_class()
else:
self.script.delete_if_no_jobs()
if not self.script.id:
messages.error(request, _("Script class has been deleted from module: ") + str(self.script.module))
return redirect('extras:script_list')
self.jobs = self.script.jobs.all()
jobs_table = JobTable(
data=self.jobs,
orderable=False,
user=request.user
)
jobs_table.configure(request)
return render(request, 'extras/script/jobs.html', {
'job_count': self.jobs.count(),
'module': self.script.module,
'script': self.script,
'table': jobs_table,
'tab': 'jobs',
})
class ScriptModuleJobsView(ScriptJobsView):
def get(self, request, module, name):
module = get_script_module(module, request)
self.script = get_object_or_404(Script.objects.all(), module=module, name=name)
if self.script.python_class:
self.script_class = self.script.python_class()
else: