data_analysis logger added

Author: IamMQaisar

analyze/data_analysis.py

@@ -3,10 +3,10 @@
 """
 
 # Standard library
+import logging
 import os.path
 import re
 import sys
-import traceback
 import warnings
 
 # Third-party
@@ -24,6 +24,14 @@
 # Set the current working directory
 CWD = os.path.dirname(os.path.abspath(__file__))
 
+# Set up logging
+logging.basicConfig(
+    level=logging.INFO,
+    format="%(asctime)s - %(levelname)s - %(message)s",
+    handlers=[logging.FileHandler("logfile.log"), logging.StreamHandler()],
+)
+logger = logging.getLogger(__name__)
+
 
 def tags_frequency(csv_path, column_names):
     """
@@ -495,15 +503,13 @@ def main():
 
 
 if __name__ == "__main__":
-    # Exception handling
     try:
         main()
     except SystemExit as e:
         sys.exit(e.code)
     except KeyboardInterrupt:
-        print("INFO (130) Halted via KeyboardInterrupt.", file=sys.stderr)
+        logger.info("Halted via KeyboardInterrupt.")
         sys.exit(130)
     except Exception:
-        print("ERROR (1) Unhandled exception:", file=sys.stderr)
-        print(traceback.print_exc(), file=sys.stderr)
-    sys.exit(1)
+        logger.exception("Unhandled exception:")
+        sys.exit(1)