nextrec 0.1.4__py3-none-any.whl → 0.1.7__py3-none-any.whl

nextrec/basic/loggers.py

@@ -14,37 +14,36 @@ 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,
@@ -63,17 +62,16 @@ 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:
@@ -91,47 +89,36 @@ 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"
-    )
-
-    console_format = "%(message)s"
-    file_format = "%(asctime)s - %(levelname)s - %(message)s"
-    date_format = "%H:%M:%S"
+    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'
+    
     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