Cyberdefense/VulnWhisperer
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

write output to .tmp then rename

Browse Source
This commit is contained in:
pemontto
2019-05-01 10:33:37 +01:00
parent be06f4811a
commit ea864d09ac
1 changed files with 11 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
vulnwhisp/vulnwhisp.py
View File

@ -519,7 +519,8 @@ class vulnWhispererNessus(vulnWhispererBase):
vuln_ready['scan_source'] = self.CONFIG_SECTION vuln_ready['scan_source'] = self.CONFIG_SECTION
vuln_ready['scan_time'] = norm_time vuln_ready['scan_time'] = norm_time
vuln_ready.to_json(relative_path_name, orient='records', lines=True) vuln_ready.to_json(relative_path_name + '.tmp', orient='records', lines=True)
os.rename(relative_path_name + '.tmp', relative_path_name)
self.logger.info('{records} records written to {path} '.format(records=vuln_ready.shape[0], self.logger.info('{records} records written to {path} '.format(records=vuln_ready.shape[0],
path=relative_path_name)) path=relative_path_name))
@ -630,10 +631,10 @@ class vulnWhispererQualys(vulnWhispererBase):
vuln_ready['scan_time'] = launched_date vuln_ready['scan_time'] = launched_date
if output_format == 'json': if output_format == 'json':
vuln_ready.to_json(relative_path_name, orient='records', lines=True) vuln_ready.to_json(relative_path_name + '.tmp', orient='records', lines=True)
elif output_format == 'csv': elif output_format == 'csv':
vuln_ready.to_csv(relative_path_name, index=False, header=True) vuln_ready.to_csv(relative_path_name + '.tmp', index=False, header=True)
os.rename(relative_path_name + '.tmp', relative_path_name)
self.logger.info('{records} records written to {path} '.format(records=vuln_ready.shape[0], self.logger.info('{records} records written to {path} '.format(records=vuln_ready.shape[0],
path=relative_path_name)) path=relative_path_name))
@ -793,7 +794,8 @@ class vulnWhispererOpenVAS(vulnWhispererBase):
vuln_ready['scan_time'] = launched_date vuln_ready['scan_time'] = launched_date
vuln_ready['scan_source'] = self.CONFIG_SECTION vuln_ready['scan_source'] = self.CONFIG_SECTION
vuln_ready.to_json(relative_path_name, orient='records', lines=True) vuln_ready.to_json(relative_path_name + '.tmp', orient='records', lines=True)
os.rename(relative_path_name + '.tmp', relative_path_name)
self.logger.info('{records} records written to {path} '.format(records=vuln_ready.shape[0], self.logger.info('{records} records written to {path} '.format(records=vuln_ready.shape[0],
path=relative_path_name)) path=relative_path_name))
@ -916,8 +918,10 @@ class vulnWhispererQualysVuln(vulnWhispererBase):
return self.exit_code return self.exit_code
if output_format == 'json': if output_format == 'json':
vuln_ready.to_json(relative_path_name, orient='records', lines=True) vuln_ready.to_json(relative_path_name + '.tmp', orient='records', lines=True)
elif output_format == 'csv':
vuln_ready.to_csv(relative_path_name + '.tmp', index=False, header=True)
os.rename(relative_path_name + '.tmp', relative_path_name)
self.logger.info('{records} records written to {path} '.format(records=vuln_ready.shape[0], self.logger.info('{records} records written to {path} '.format(records=vuln_ready.shape[0],
path=relative_path_name)) path=relative_path_name))