NeoAnd/netbox
1
0
Fork 0
mirror of https://github.com/netbox-community/netbox.git synced 2026-01-07 12:36:55 -06:00
Code Issues Actions 9 Packages Projects Releases Wiki Activity

Tweaked report run logic

Browse Source
This commit is contained in:
Jeremy Stretch
2017-09-21 13:49:04 -04:00
parent 8f1607e010
commit 16d1f9aca8
2 changed files with 30 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files

33
netbox/extras/management/commands/runreport.py
View File

@@ -20,26 +20,39 @@ class Command(BaseCommand):
# Gather all reports to be run
reports = []
for module_name in options['reports']:
# Split the report name off if one has been provided.
report_name = None
if '.' in module_name:
module_name, report_name = module_name.split('.', 1)
# Import the report module
try:
report_module = importlib.import_module('reports.report_{}'.format(module_name))
except ImportError:
self.stdout.write(
"Report '{}' not found. Ensure that the report has been saved as 'report_{}.py' in the reports "
"directory.".format(module_name, module_name)
"Report module '{}' not found. Ensure that the report has been saved as 'report_{}.py' in the "
"reports directory.".format(module_name, module_name)
)
return
for name, cls in inspect.getmembers(report_module, inspect.isclass):
if cls in Report.__subclasses__():
reports.append((name, cls))
# If the name of a particular report has been given, run that. Otherwise, run all reports in the module.
if report_name is not None:
report_cls = getattr(report_module, report_name)
reports = [(report_name, report_cls)]
else:
for name, report_cls in inspect.getmembers(report_module, inspect.isclass):
if report_cls in Report.__subclasses__():
reports.append((name, report_cls))
# Run reports
for name, report in reports:
self.stdout.write("[{:%H:%M:%S}] Running report {}...".format(timezone.now(), name))
report = report()
report.run()
for name, report_cls in reports:
self.stdout.write("[{:%H:%M:%S}] Running {}...".format(timezone.now(), name))
report = report_cls()
results = report.run()
status = self.style.ERROR('FAILED') if report.failed else self.style.SUCCESS('SUCCESS')
self.stdout.write("[{:%H:%M:%S}] {}: {}".format(timezone.now(), name, status))
for test_name, attrs in report.results.items():
for test_name, attrs in results.items():
self.stdout.write(" {}: {} success, {} info, {} warning, {} failed".format(
test_name, attrs['success'], attrs['info'], attrs['warning'], attrs['failed']
))