From 97e4f073bfc371a5b35bbeb6ced95f6cd5970cd6 Mon Sep 17 00:00:00 2001
From: Quim <qmontal@protonmail.com>
Date: Fri, 22 Mar 2019 10:38:55 +0100
Subject: [PATCH] added logging to file

---
 bin/vuln_whisperer | 17 ++++++++++++++++-
 1 file changed, 16 insertions(+), 1 deletion(-)

diff --git a/bin/vuln_whisperer b/bin/vuln_whisperer
index d94b5ce..4621c23 100644
--- a/bin/vuln_whisperer
+++ b/bin/vuln_whisperer
@@ -41,7 +41,13 @@ def main():
         stream=sys.stdout,
         level=logging.DEBUG if args.debug else logging.INFO
     )
-    logger = logging.getLogger(name='main')
+    logger = logging.getLogger()
+    # we set up the logger to log as well to file
+    fh = logging.FileHandler('vulnwhisperer.log')
+    fh.setLevel(logging.DEBUG if args.debug else logging.INFO)
+    fh.setFormatter(logging.Formatter("%(asctime)s - %(name)s:%(levelname)s:%(message)s", "%Y-%m-%d %H:%M:%S"))
+    logger.addHandler(fh)
+
     if args.fancy:
         import coloredlogs
         coloredlogs.install(level='DEBUG' if args.debug else 'INFO')
@@ -68,6 +74,7 @@ def main():
 
             	vw.whisper_vulnerabilities()
             # TODO: fix this to NOT be exit 1 unless in error
+            close_logging_handlers()
 	    sys.exit(1)
 
         else:
@@ -82,6 +89,7 @@ def main():
 
             vw.whisper_vulnerabilities()
             # TODO: fix this to NOT be exit 1 unless in error
+            close_logging_handlers()
             sys.exit(1)
 
     except Exception as e:
@@ -90,8 +98,15 @@ def main():
             logger.error('{}'.format(str(e)))
             print('ERROR: {error}'.format(error=e))
         # TODO: fix this to NOT be exit 2 unless in error
+        close_logging_handlers()
         sys.exit(2)
 
+    close_logging_handlers()
+
+def close_logging_handlers():
+    for handler in logger.handlers:
+        handler.close()
+        logger.removeFilter(handler)
 
 if __name__ == '__main__':
     main()