This commit is contained in:
Quim
2019-02-12 16:51:26 +01:00
parent b0caccdc89
commit ccf2e4b1d1

View File

@ -838,32 +838,29 @@ class vulnWhispererQualysVuln(vulnWhispererBase):
else: else:
self.logger.info('Processing report ID: {}'.format(report_id)) self.logger.info('Processing report ID: {}'.format(report_id))
vuln_ready = self.qualys_scan.process_data(scan_id=report_id) vuln_ready = self.qualys_scan.process_data(scan_id=report_id)
if not vuln_ready.empty: vuln_ready['scan_name'] = scan_name
vuln_ready['scan_name'] = scan_name vuln_ready['scan_reference'] = report_id
vuln_ready['scan_reference'] = report_id vuln_ready.rename(columns=self.COLUMN_MAPPING, inplace=True)
vuln_ready.rename(columns=self.COLUMN_MAPPING, inplace=True)
record_meta = ( record_meta = (
scan_name, scan_name,
scan_reference, scan_reference,
launched_date, launched_date,
report_name, report_name,
time.time(), time.time(),
vuln_ready.shape[0], vuln_ready.shape[0],
self.CONFIG_SECTION, self.CONFIG_SECTION,
report_id, report_id,
1, 1,
) )
self.record_insert(record_meta) self.record_insert(record_meta)
if output_format == 'json': if output_format == 'json':
with open(relative_path_name, 'w') as f: with open(relative_path_name, 'w') as f:
f.write(vuln_ready.to_json(orient='records', lines=True)) f.write(vuln_ready.to_json(orient='records', lines=True))
f.write('\n') f.write('\n')
self.logger.info('Report written to {}'.format(report_name)) self.logger.info('Report written to {}'.format(report_name))
else:
return False
except Exception as e: except Exception as e:
self.logger.error('Could not process {}: {}'.format(report_id, str(e))) self.logger.error('Could not process {}: {}'.format(report_id, str(e)))