From d7fc63c9523eaa0789ed5a39ba20bfd15a2ad70d Mon Sep 17 00:00:00 2001 From: Austin Taylor Date: Sat, 30 Dec 2017 14:40:49 -0500 Subject: [PATCH] Better database handling --- vulnwhisp/vulnwhisp.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/vulnwhisp/vulnwhisp.py b/vulnwhisp/vulnwhisp.py index f6ddfcf..3249763 100755 --- a/vulnwhisp/vulnwhisp.py +++ b/vulnwhisp/vulnwhisp.py @@ -567,7 +567,7 @@ class vulnWhispererQualys(vulnWhispererBase): % self.path_check(str(generated_report_id))) else: print('{error} Could not process report ID: %s'.format(error=bcolors.FAIL) % status) - self.conn.close() + except Exception as e: print('{error} - Could not process %s - %s'.format(error=bcolors.FAIL) % (report_id, e)) return vuln_ready @@ -597,7 +597,8 @@ class vulnWhispererQualys(vulnWhispererBase): scan_reference=r['reference']) else: self.vprint('{info} No new scans to process. Exiting...'.format(info=bcolors.INFO)) - exit(0) + self.conn.close() + exit(0)