the logger automatically exits on critical errors

ThomasByr · ThomasByr · commit cab4de5f1113 · 2023-08-30T17:08:34.000+02:00
diff --git a/src/core/app.py b/src/core/app.py
@@ -68,7 +68,6 @@ def __init__(self, args: Namespace) -> None:
     self.log.debug('Received json config file path (%s)', self.args.cfg)
     if not os.path.isfile(self.args.cfg):
       self.log.critical('Invalid json config file path supplied (%s)', self.args.cfg)
-      sys.exit(1)
 
     __bar = bar_factory('\u2501', borders=(' ', ' '), background=' ')
     __spinner = frame_spinner_factory([colored(p, 'cyan') if supports_color else p for p in '⠋⠙⠹⠸⠼⠴⠦⠧⠇⠏'])
@@ -144,7 +143,6 @@ def __get_json_config_path(self) -> str:
 
     if len(found) == 0:
       self.log.critical('No json config file found in file tree')
-      sys.exit(1)
     return found.pop()
 
   def __on_end(self, sig: int, _: Any, /) -> None:
@@ -207,14 +205,12 @@ def __load_points(self, cfg: Config) -> None:
 
             self.log.critical('Failed to parse line: %s (%s:%d)\n%s', line, cfg.file_path,
                               len(self.points) + int(cfg.skip_first_line) - index, e)
-            sys.exit(1)
 
     except FileNotFoundError as e:
       self.log.error('Skipping unknown file: %s', e)
       return
     except Exception as e:                                  # pylint: disable=broad-except
       self.log.critical('Failed to read file: %s\n%s', cfg.file_path, e)
-      sys.exit(1)
     self.log.debug('Loaded %s points from file: \u2026/%s',
                    format(len(self.points) + int(cfg.skip_first_line) - index, '_'), basename)
 
@@ -277,25 +273,21 @@ def __setup(self) -> None:
         self.log.critical(
           'Failed to parse json config file : '
           'maximum nesting level could be reached, please check your file\n%s', e)
-        sys.exit(1)
     if not raw_data:
       self.log.critical('Failed to parse %s : empty file', self.args.cfg)
-      sys.exit(1)
     default: dict[str, Any] = None
     configs: list[dict[str, Any]] = None
     try:
       default = raw_data['default']
       configs = raw_data['configs']
     except KeyError as e:
       self.log.critical('Failed to parse %s : %s', self.args.cfg, e)
-      sys.exit(1)
     cfgs: list[Config] = []
     try:
       for cfg in configs:
         cfgs.append(Config.from_json(json=cfg, **default))
     except ValueError as e:
       self.log.critical('Failed to parse config n°%d : %s', len(cfgs), e)
-      sys.exit(1)
 
     fset: list[int] = None
     if self.args.only and any(map(lambda x: x > len(cfgs), self.args.only)): # pylint: disable=bad-builtin
diff --git a/src/log/fmt.py b/src/log/fmt.py
@@ -1,11 +1,13 @@
 from __future__ import annotations
 
+import sys
+
 import logging
 from typing import Any
 from typing_extensions import override
 from termcolor import colored
 
-__all__ = ['UsefulFormatter']
+__all__ = ['UsefulFormatter', 'UselessHandler']
 
 
 def formatter(
@@ -42,3 +44,12 @@ def format(self, record: logging.LogRecord) -> str:
     log_fmt = formats.get(record.levelno)
     fmt = logging.Formatter(log_fmt, self.dt_fmt, style='%')
     return fmt.format(record)
+
+
+class UselessHandler(logging.StreamHandler):
+
+  @override
+  def emit(self, record: logging.LogRecord) -> None:
+    super().emit(record)
+    if record.levelno >= logging.CRITICAL: # exit on critical errors
+      sys.exit(1)
diff --git a/src/log/logger.py b/src/log/logger.py
@@ -75,7 +75,7 @@ def init_logger(log_lvl: int = logging.INFO) -> bool:
   color = supports_color()
 
   # create console handler with a higher log level
-  console_handler = logging.StreamHandler()
+  console_handler = UselessHandler()
   console_handler.setLevel(log_lvl)
   console_handler.setFormatter(UsefulFormatter(colored_output=color))
 
