mirror of
https://github.com/netbox-community/netbox.git
synced 2024-05-10 07:54:54 +00:00
f7a2eb8aef
* Initial work on #11890 * Consolidate get_scripts() and get_reports() functions * Introduce proxy models for script & report modules * Add add/delete views for reports & scripts * Add deletion links for modules * Enable resolving scripts/reports from module class * Remove get_modules() utility function * Show results in report/script lists * Misc cleanup * Fix file uploads * Support automatic migration for submodules * Fix module child ordering * Template cleanup * Remove ManagedFile views * Move is_script(), is_report() into extras.utils * Fix URLs for nested reports & scripts * Misc cleanup
68 lines
2.8 KiB
Python
68 lines
2.8 KiB
Python
import time
|
|
|
|
from django.contrib.contenttypes.models import ContentType
|
|
from django.core.management.base import BaseCommand
|
|
from django.utils import timezone
|
|
|
|
from extras.choices import JobResultStatusChoices
|
|
from extras.models import JobResult, ReportModule
|
|
from extras.reports import run_report
|
|
|
|
|
|
class Command(BaseCommand):
|
|
help = "Run a report to validate data in NetBox"
|
|
|
|
def add_arguments(self, parser):
|
|
parser.add_argument('reports', nargs='+', help="Report(s) to run")
|
|
|
|
def handle(self, *args, **options):
|
|
|
|
for module in ReportModule.objects.all():
|
|
for report in module.reports.values():
|
|
if module.name in options['reports'] or report.full_name in options['reports']:
|
|
|
|
# Run the report and create a new JobResult
|
|
self.stdout.write(
|
|
"[{:%H:%M:%S}] Running {}...".format(timezone.now(), report.full_name)
|
|
)
|
|
|
|
report_content_type = ContentType.objects.get(app_label='extras', model='report')
|
|
job_result = JobResult.enqueue_job(
|
|
run_report,
|
|
report.full_name,
|
|
report_content_type,
|
|
None,
|
|
job_timeout=report.job_timeout
|
|
)
|
|
|
|
# Wait on the job to finish
|
|
while job_result.status not in JobResultStatusChoices.TERMINAL_STATE_CHOICES:
|
|
time.sleep(1)
|
|
job_result = JobResult.objects.get(pk=job_result.pk)
|
|
|
|
# Report on success/failure
|
|
if job_result.status == JobResultStatusChoices.STATUS_FAILED:
|
|
status = self.style.ERROR('FAILED')
|
|
elif job_result == JobResultStatusChoices.STATUS_ERRORED:
|
|
status = self.style.ERROR('ERRORED')
|
|
else:
|
|
status = self.style.SUCCESS('SUCCESS')
|
|
|
|
for test_name, attrs in job_result.data.items():
|
|
self.stdout.write(
|
|
"\t{}: {} success, {} info, {} warning, {} failure".format(
|
|
test_name, attrs['success'], attrs['info'], attrs['warning'], attrs['failure']
|
|
)
|
|
)
|
|
self.stdout.write(
|
|
"[{:%H:%M:%S}] {}: {}".format(timezone.now(), report.full_name, status)
|
|
)
|
|
self.stdout.write(
|
|
"[{:%H:%M:%S}] {}: Duration {}".format(timezone.now(), report.full_name, job_result.duration)
|
|
)
|
|
|
|
# Wrap things up
|
|
self.stdout.write(
|
|
"[{:%H:%M:%S}] Finished".format(timezone.now())
|
|
)
|