nextrec 0.1.1__py3-none-any.whl → 0.1.3__py3-none-any.whl

nextrec/basic/loggers.py

@@ -14,36 +14,37 @@ import datetime
 import logging
 
 ANSI_CODES = {
-    'black': '\033[30m',
-    'red': '\033[31m',
-    'green': '\033[32m',
-    'yellow': '\033[33m',
-    'blue': '\033[34m',
-    'magenta': '\033[35m',
-    'cyan': '\033[36m',
-    'white': '\033[37m',
-    'bright_black': '\033[90m',
-    'bright_red': '\033[91m',
-    'bright_green': '\033[92m',
-    'bright_yellow': '\033[93m',
-    'bright_blue': '\033[94m',
-    'bright_magenta': '\033[95m',
-    'bright_cyan': '\033[96m',
-    'bright_white': '\033[97m',
+    "black": "\033[30m",
+    "red": "\033[31m",
+    "green": "\033[32m",
+    "yellow": "\033[33m",
+    "blue": "\033[34m",
+    "magenta": "\033[35m",
+    "cyan": "\033[36m",
+    "white": "\033[37m",
+    "bright_black": "\033[90m",
+    "bright_red": "\033[91m",
+    "bright_green": "\033[92m",
+    "bright_yellow": "\033[93m",
+    "bright_blue": "\033[94m",
+    "bright_magenta": "\033[95m",
+    "bright_cyan": "\033[96m",
+    "bright_white": "\033[97m",
 }
 
-ANSI_BOLD = '\033[1m'
-ANSI_RESET = '\033[0m'
-ANSI_ESCAPE_PATTERN = re.compile(r'\033\[[0-9;]*m')
+ANSI_BOLD = "\033[1m"
+ANSI_RESET = "\033[0m"
+ANSI_ESCAPE_PATTERN = re.compile(r"\033\[[0-9;]*m")
 
 DEFAULT_LEVEL_COLORS = {
-    'DEBUG': 'cyan',
-    'INFO': None,
-    'WARNING': 'yellow',
-    'ERROR': 'red',
-    'CRITICAL': 'bright_red',
+    "DEBUG": "cyan",
+    "INFO": None,
+    "WARNING": "yellow",
+    "ERROR": "red",
+    "CRITICAL": "bright_red",
 }
 
+
 class AnsiFormatter(logging.Formatter):
     def __init__(
         self,
@@ -62,16 +63,17 @@ class AnsiFormatter(logging.Formatter):
         record_copy = copy.copy(record)
         formatted = super().format(record_copy)
 
-        if self.auto_color_level and '\033[' not in formatted:
+        if self.auto_color_level and "\033[" not in formatted:
             color = self.level_colors.get(record.levelname)
             if color:
                 formatted = colorize(formatted, color=color)
 
         if self.strip_ansi:
-            return ANSI_ESCAPE_PATTERN.sub('', formatted)
+            return ANSI_ESCAPE_PATTERN.sub("", formatted)
 
         return formatted
 
+
 def colorize(text: str, color: str | None = None, bold: bool = False) -> str:
     """Apply ANSI color and bold formatting to the given text."""
     if not color and not bold:
@@ -89,36 +91,47 @@ def colorize(text: str, color: str | None = None, bold: bool = False) -> str:
 
     return result
 
+
 def setup_logger(log_dir: str | None = None):
     """Set up a logger that logs to both console and a file with ANSI formatting.
-       Only console output has colors; file output is stripped of ANSI codes.    
+    Only console output has colors; file output is stripped of ANSI codes.
     """
     if log_dir is None:
         project_root = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
         log_dir = os.path.join(project_root, "..", "logs")
-    
+
     os.makedirs(log_dir, exist_ok=True)
-    log_file = os.path.join(log_dir, f"nextrec_{datetime.datetime.now().strftime('%Y%m%d_%H%M%S')}.log")
+    log_file = os.path.join(
+        log_dir, f"nextrec_{datetime.datetime.now().strftime('%Y%m%d_%H%M%S')}.log"
+    )
+
+    console_format = "%(message)s"
+    file_format = "%(asctime)s - %(levelname)s - %(message)s"
+    date_format = "%H:%M:%S"
 
-    console_format = '%(message)s'
-    file_format = '%(asctime)s - %(levelname)s - %(message)s'
-    date_format = '%H:%M:%S'
-    
     logger = logging.getLogger()
     logger.setLevel(logging.INFO)
 
     if logger.hasHandlers():
         logger.handlers.clear()
 
-    file_handler = logging.FileHandler(log_file, encoding='utf-8')
+    file_handler = logging.FileHandler(log_file, encoding="utf-8")
     file_handler.setLevel(logging.INFO)
-    file_handler.setFormatter(AnsiFormatter(file_format, datefmt=date_format, strip_ansi=True))
+    file_handler.setFormatter(
+        AnsiFormatter(file_format, datefmt=date_format, strip_ansi=True)
+    )
 
     console_handler = logging.StreamHandler(sys.stdout)
     console_handler.setLevel(logging.INFO)
-    console_handler.setFormatter(AnsiFormatter(console_format, datefmt=date_format, auto_color_level=True,))
+    console_handler.setFormatter(
+        AnsiFormatter(
+            console_format,
+            datefmt=date_format,
+            auto_color_level=True,
+        )
+    )
 
     logger.addHandler(file_handler)
     logger.addHandler(console_handler)
-    
+
     return logger