mirror of
https://github.com/netbox-community/netbox.git
synced 2025-07-23 04:22:01 -06:00
* Fixes #14572 - Constrains JobView (and related views) badge to specific named job * Adjust report views to resolve same problem * Fixed PEP8 error * Update netbox/templates/extras/script/base.html Co-authored-by: Jeremy Stretch <jstretch@netboxlabs.com> * Move function to method on PythonModuleMixin * Update netbox/extras/views.py Co-authored-by: Jeremy Stretch <jstretch@netboxlabs.com> * Update netbox/extras/views.py Co-authored-by: Jeremy Stretch <jstretch@netboxlabs.com> * Update netbox/extras/views.py Co-authored-by: Jeremy Stretch <jstretch@netboxlabs.com> * Update netbox/extras/views.py Co-authored-by: Jeremy Stretch <jstretch@netboxlabs.com> * Update to mixin and view --------- Co-authored-by: Jeremy Stretch <jstretch@netboxlabs.com>
This commit is contained in:
parent
79e0d3ae67
commit
fd5392563f
@ -8,6 +8,16 @@ __all__ = (
|
|||||||
|
|
||||||
class PythonModuleMixin:
|
class PythonModuleMixin:
|
||||||
|
|
||||||
|
def get_jobs(self, name):
|
||||||
|
"""
|
||||||
|
Returns a list of Jobs associated with this specific script or report module
|
||||||
|
:param name: The class name of the script or report
|
||||||
|
:return: List of Jobs associated with this
|
||||||
|
"""
|
||||||
|
return self.jobs.filter(
|
||||||
|
name=name
|
||||||
|
)
|
||||||
|
|
||||||
@property
|
@property
|
||||||
def path(self):
|
def path(self):
|
||||||
return os.path.splitext(self.file_path)[0]
|
return os.path.splitext(self.file_path)[0]
|
||||||
|
@ -1057,16 +1057,14 @@ class ReportView(ContentTypePermissionRequiredMixin, View):
|
|||||||
def get(self, request, module, name):
|
def get(self, request, module, name):
|
||||||
module = get_report_module(module, request)
|
module = get_report_module(module, request)
|
||||||
report = module.reports[name]()
|
report = module.reports[name]()
|
||||||
|
jobs = module.get_jobs(report.class_name)
|
||||||
|
|
||||||
object_type = ContentType.objects.get(app_label='extras', model='reportmodule')
|
report.result = jobs.filter(
|
||||||
report.result = Job.objects.filter(
|
|
||||||
object_type=object_type,
|
|
||||||
object_id=module.pk,
|
|
||||||
name=report.name,
|
|
||||||
status__in=JobStatusChoices.TERMINAL_STATE_CHOICES
|
status__in=JobStatusChoices.TERMINAL_STATE_CHOICES
|
||||||
).first()
|
).first()
|
||||||
|
|
||||||
return render(request, 'extras/report.html', {
|
return render(request, 'extras/report.html', {
|
||||||
|
'job_count': jobs.count(),
|
||||||
'module': module,
|
'module': module,
|
||||||
'report': report,
|
'report': report,
|
||||||
'form': ReportForm(scheduling_enabled=report.scheduling_enabled),
|
'form': ReportForm(scheduling_enabled=report.scheduling_enabled),
|
||||||
@ -1078,6 +1076,7 @@ class ReportView(ContentTypePermissionRequiredMixin, View):
|
|||||||
|
|
||||||
module = get_report_module(module, request)
|
module = get_report_module(module, request)
|
||||||
report = module.reports[name]()
|
report = module.reports[name]()
|
||||||
|
jobs = module.get_jobs(report.class_name)
|
||||||
form = ReportForm(request.POST, scheduling_enabled=report.scheduling_enabled)
|
form = ReportForm(request.POST, scheduling_enabled=report.scheduling_enabled)
|
||||||
|
|
||||||
if form.is_valid():
|
if form.is_valid():
|
||||||
@ -1086,6 +1085,7 @@ class ReportView(ContentTypePermissionRequiredMixin, View):
|
|||||||
if not get_workers_for_queue('default'):
|
if not get_workers_for_queue('default'):
|
||||||
messages.error(request, "Unable to run report: RQ worker process not running.")
|
messages.error(request, "Unable to run report: RQ worker process not running.")
|
||||||
return render(request, 'extras/report.html', {
|
return render(request, 'extras/report.html', {
|
||||||
|
'job_count': jobs.count(),
|
||||||
'report': report,
|
'report': report,
|
||||||
})
|
})
|
||||||
|
|
||||||
@ -1103,6 +1103,7 @@ class ReportView(ContentTypePermissionRequiredMixin, View):
|
|||||||
return redirect('extras:report_result', job_pk=job.pk)
|
return redirect('extras:report_result', job_pk=job.pk)
|
||||||
|
|
||||||
return render(request, 'extras/report.html', {
|
return render(request, 'extras/report.html', {
|
||||||
|
'job_count': jobs.count(),
|
||||||
'module': module,
|
'module': module,
|
||||||
'report': report,
|
'report': report,
|
||||||
'form': form,
|
'form': form,
|
||||||
@ -1117,8 +1118,10 @@ class ReportSourceView(ContentTypePermissionRequiredMixin, View):
|
|||||||
def get(self, request, module, name):
|
def get(self, request, module, name):
|
||||||
module = get_report_module(module, request)
|
module = get_report_module(module, request)
|
||||||
report = module.reports[name]()
|
report = module.reports[name]()
|
||||||
|
jobs = module.get_jobs(report.class_name)
|
||||||
|
|
||||||
return render(request, 'extras/report/source.html', {
|
return render(request, 'extras/report/source.html', {
|
||||||
|
'job_count': jobs.count(),
|
||||||
'module': module,
|
'module': module,
|
||||||
'report': report,
|
'report': report,
|
||||||
'tab': 'source',
|
'tab': 'source',
|
||||||
@ -1133,13 +1136,7 @@ class ReportJobsView(ContentTypePermissionRequiredMixin, View):
|
|||||||
def get(self, request, module, name):
|
def get(self, request, module, name):
|
||||||
module = get_report_module(module, request)
|
module = get_report_module(module, request)
|
||||||
report = module.reports[name]()
|
report = module.reports[name]()
|
||||||
|
jobs = module.get_jobs(report.class_name)
|
||||||
object_type = ContentType.objects.get(app_label='extras', model='reportmodule')
|
|
||||||
jobs = Job.objects.filter(
|
|
||||||
object_type=object_type,
|
|
||||||
object_id=module.pk,
|
|
||||||
name=report.class_name
|
|
||||||
)
|
|
||||||
|
|
||||||
jobs_table = JobTable(
|
jobs_table = JobTable(
|
||||||
data=jobs,
|
data=jobs,
|
||||||
@ -1149,6 +1146,7 @@ class ReportJobsView(ContentTypePermissionRequiredMixin, View):
|
|||||||
jobs_table.configure(request)
|
jobs_table.configure(request)
|
||||||
|
|
||||||
return render(request, 'extras/report/jobs.html', {
|
return render(request, 'extras/report/jobs.html', {
|
||||||
|
'job_count': jobs.count(),
|
||||||
'module': module,
|
'module': module,
|
||||||
'report': report,
|
'report': report,
|
||||||
'table': jobs_table,
|
'table': jobs_table,
|
||||||
@ -1232,19 +1230,16 @@ class ScriptView(ContentTypePermissionRequiredMixin, View):
|
|||||||
def get(self, request, module, name):
|
def get(self, request, module, name):
|
||||||
module = get_script_module(module, request)
|
module = get_script_module(module, request)
|
||||||
script = module.scripts[name]()
|
script = module.scripts[name]()
|
||||||
|
jobs = module.get_jobs(script.class_name)
|
||||||
form = script.as_form(initial=normalize_querydict(request.GET))
|
form = script.as_form(initial=normalize_querydict(request.GET))
|
||||||
|
|
||||||
# Look for a pending Job (use the latest one by creation timestamp)
|
# Look for a pending Job (use the latest one by creation timestamp)
|
||||||
object_type = ContentType.objects.get(app_label='extras', model='scriptmodule')
|
script.result = module.get_jobs(script.class_name).exclude(
|
||||||
script.result = Job.objects.filter(
|
|
||||||
object_type=object_type,
|
|
||||||
object_id=module.pk,
|
|
||||||
name=script.name,
|
|
||||||
).exclude(
|
|
||||||
status__in=JobStatusChoices.TERMINAL_STATE_CHOICES
|
status__in=JobStatusChoices.TERMINAL_STATE_CHOICES
|
||||||
).first()
|
).first()
|
||||||
|
|
||||||
return render(request, 'extras/script.html', {
|
return render(request, 'extras/script.html', {
|
||||||
|
'job_count': jobs.count(),
|
||||||
'module': module,
|
'module': module,
|
||||||
'script': script,
|
'script': script,
|
||||||
'form': form,
|
'form': form,
|
||||||
@ -1256,6 +1251,7 @@ class ScriptView(ContentTypePermissionRequiredMixin, View):
|
|||||||
|
|
||||||
module = get_script_module(module, request)
|
module = get_script_module(module, request)
|
||||||
script = module.scripts[name]()
|
script = module.scripts[name]()
|
||||||
|
jobs = module.get_jobs(script.class_name)
|
||||||
form = script.as_form(request.POST, request.FILES)
|
form = script.as_form(request.POST, request.FILES)
|
||||||
|
|
||||||
# Allow execution only if RQ worker process is running
|
# Allow execution only if RQ worker process is running
|
||||||
@ -1279,6 +1275,7 @@ class ScriptView(ContentTypePermissionRequiredMixin, View):
|
|||||||
return redirect('extras:script_result', job_pk=job.pk)
|
return redirect('extras:script_result', job_pk=job.pk)
|
||||||
|
|
||||||
return render(request, 'extras/script.html', {
|
return render(request, 'extras/script.html', {
|
||||||
|
'job_count': jobs.count(),
|
||||||
'module': module,
|
'module': module,
|
||||||
'script': script,
|
'script': script,
|
||||||
'form': form,
|
'form': form,
|
||||||
@ -1293,8 +1290,10 @@ class ScriptSourceView(ContentTypePermissionRequiredMixin, View):
|
|||||||
def get(self, request, module, name):
|
def get(self, request, module, name):
|
||||||
module = get_script_module(module, request)
|
module = get_script_module(module, request)
|
||||||
script = module.scripts[name]()
|
script = module.scripts[name]()
|
||||||
|
jobs = module.get_jobs(script.class_name)
|
||||||
|
|
||||||
return render(request, 'extras/script/source.html', {
|
return render(request, 'extras/script/source.html', {
|
||||||
|
'job_count': jobs.count(),
|
||||||
'module': module,
|
'module': module,
|
||||||
'script': script,
|
'script': script,
|
||||||
'tab': 'source',
|
'tab': 'source',
|
||||||
@ -1309,13 +1308,7 @@ class ScriptJobsView(ContentTypePermissionRequiredMixin, View):
|
|||||||
def get(self, request, module, name):
|
def get(self, request, module, name):
|
||||||
module = get_script_module(module, request)
|
module = get_script_module(module, request)
|
||||||
script = module.scripts[name]()
|
script = module.scripts[name]()
|
||||||
|
jobs = module.get_jobs(script.class_name)
|
||||||
object_type = ContentType.objects.get(app_label='extras', model='scriptmodule')
|
|
||||||
jobs = Job.objects.filter(
|
|
||||||
object_type=object_type,
|
|
||||||
object_id=module.pk,
|
|
||||||
name=script.class_name
|
|
||||||
)
|
|
||||||
|
|
||||||
jobs_table = JobTable(
|
jobs_table = JobTable(
|
||||||
data=jobs,
|
data=jobs,
|
||||||
@ -1325,6 +1318,7 @@ class ScriptJobsView(ContentTypePermissionRequiredMixin, View):
|
|||||||
jobs_table.configure(request)
|
jobs_table.configure(request)
|
||||||
|
|
||||||
return render(request, 'extras/script/jobs.html', {
|
return render(request, 'extras/script/jobs.html', {
|
||||||
|
'job_count': jobs.count(),
|
||||||
'module': module,
|
'module': module,
|
||||||
'script': script,
|
'script': script,
|
||||||
'table': jobs_table,
|
'table': jobs_table,
|
||||||
|
@ -34,7 +34,7 @@
|
|||||||
</li>
|
</li>
|
||||||
<li class="nav-item" role="presentation">
|
<li class="nav-item" role="presentation">
|
||||||
<a class="nav-link{% if tab == 'jobs' %} active{% endif %}" href="{% url 'extras:report_jobs' module=report.module name=report.class_name %}">
|
<a class="nav-link{% if tab == 'jobs' %} active{% endif %}" href="{% url 'extras:report_jobs' module=report.module name=report.class_name %}">
|
||||||
{% trans "Jobs" %} {% badge module.jobs.count %}
|
{% trans "Jobs" %} {% badge job_count %}
|
||||||
</a>
|
</a>
|
||||||
</li>
|
</li>
|
||||||
</ul>
|
</ul>
|
||||||
|
@ -33,7 +33,7 @@
|
|||||||
</li>
|
</li>
|
||||||
<li class="nav-item" role="presentation">
|
<li class="nav-item" role="presentation">
|
||||||
<a class="nav-link{% if tab == 'jobs' %} active{% endif %}" href="{% url 'extras:script_jobs' module=script.module name=script.class_name %}">
|
<a class="nav-link{% if tab == 'jobs' %} active{% endif %}" href="{% url 'extras:script_jobs' module=script.module name=script.class_name %}">
|
||||||
{% trans "Jobs" %} {% badge module.jobs.count %}
|
{% trans "Jobs" %} {% badge job_count %}
|
||||||
</a>
|
</a>
|
||||||
</li>
|
</li>
|
||||||
</ul>
|
</ul>
|
||||||
|
Loading…
Reference in New Issue
Block a user