mirror/netbox-community-netbox
1
0
Fork 0
mirror of https://github.com/netbox-community/netbox.git synced 2024-05-10 07:54:54 +00:00
Code Activity

Fixed runreport management command

Browse Source
This commit is contained in:
Jeremy Stretch
2017-09-28 12:50:52 -04:00
parent 571b817f04
commit b65e9fe0f5
1 changed files with 9 additions and 15 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@ -16,36 +16,30 @@ class Command(BaseCommand):
def handle(self, *args, **options): def handle(self, *args, **options):
# Gather all reports to be run # Gather all available reports
reports = get_reports() reports = get_reports()
# Run reports # Run reports
for module_name, report in reports: for module_name, report_list in reports:
for report_name, report_cls in report: for report in report_list:
report_name_full = '{}.{}'.format(module_name, report_name) if module_name in options['reports'] or report.full_namel in options['reports']:
if module_name in options['reports'] or report_name_full in options['reports']:
# Run the report # Run the report and create a new ReportResult
self.stdout.write( self.stdout.write(
"[{:%H:%M:%S}] Running {}.{}...".format(timezone.now(), module_name, report_name) "[{:%H:%M:%S}] Running {}...".format(timezone.now(), report.full_name)
) )
report = report_cls() report.run()
result = report.run()
# Record the results
ReportResult.objects.filter(report=report_name_full).delete()
ReportResult(report=report_name_full, failed=report.failed, data=result).save()
# Report on success/failure # Report on success/failure
status = self.style.ERROR('FAILED') if report.failed else self.style.SUCCESS('SUCCESS') status = self.style.ERROR('FAILED') if report.failed else self.style.SUCCESS('SUCCESS')
for test_name, attrs in result.items(): for test_name, attrs in report.result.data.items():
self.stdout.write( self.stdout.write(
"\t{}: {} success, {} info, {} warning, {} failed".format( "\t{}: {} success, {} info, {} warning, {} failed".format(
test_name, attrs['success'], attrs['info'], attrs['warning'], attrs['failed'] test_name, attrs['success'], attrs['info'], attrs['warning'], attrs['failed']
) )
) )
self.stdout.write( self.stdout.write(
"[{:%H:%M:%S}] {}.{}: {}".format(timezone.now(), module_name, report_name, status) "[{:%H:%M:%S}] {}: {}".format(timezone.now(), report.full_name, status)
) )
# Wrap things up # Wrap things up