bs-python-utils 0.8.2__tar.gz → 0.8.3__tar.gz

{bs_python_utils-0.8.2 → bs_python_utils-0.8.3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: bs-python-utils
-Version: 0.8.2
+Version: 0.8.3
 Summary: Utilities programs for my Python code
 Requires-Python: >=3.12
 Description-Content-Type: text/markdown
@@ -24,6 +24,7 @@ Requires-Dist: seaborn>=0.13.2
 Requires-Dist: statsmodels>=0.14.5
 Requires-Dist: streamlit>=1.49.1
 Requires-Dist: vega-datasets>=0.9.0
+Requires-Dist: colorama>=0.4.6
 Dynamic: license-file
 
 ## bs-python-utils

{bs_python_utils-0.8.2 → bs_python_utils-0.8.3}/bs_python_utils/bs_logging.py

@@ -1,14 +1,82 @@
 """Utilities for logging:
 
+* `get_logger` creates and returns a logger with colored output
 * `init_logger` initializes and returns a customized logger
-* `log_execution` ia a decorator to log entry into and ext from a function.
+* `log_execution` is a decorator to log entry into and ext from a function.
 """
 
 import functools
 import logging
+import sys
 from pathlib import Path
 from typing import Callable
 
+from colorama import Fore, Style, init
+
+# Initialize colorama
+init(autoreset=True)
+
+COLOR_MAP = {
+    "DEBUG": Fore.CYAN,
+    "INFO": Fore.GREEN,
+    "WARNING": Fore.YELLOW,
+    "ERROR": Fore.RED,
+    "CRITICAL": Fore.MAGENTA,
+}
+
+
+class ColorFormatter(logging.Formatter):
+    def format(self, record):
+        # Mirror the standard Formatter logic so we still get message, asctime, and exception text
+        record.message = record.getMessage()
+        if self.usesTime():
+            record.asctime = self.formatTime(record, self.datefmt)
+
+        color = COLOR_MAP.get(record.levelname, "")
+        level = f"{color}{record.levelname}{Style.RESET_ALL}"
+        output = f"{record.asctime} [{record.name}] {level}: {record.message}"
+
+        if record.exc_info and not record.exc_text:
+            record.exc_text = self.formatException(record.exc_info)
+        if record.exc_text:
+            output = f"{output}\n{record.exc_text}"
+
+        return output
+
+
+def get_logger(
+    name: str,
+    level=logging.INFO,
+    log_to_file: bool = False,
+) -> logging.Logger:
+    """Create and return a logger with colored output.
+
+    Args:
+        name: Name of the logger.
+        level: Minimum logging level.
+        log_to_file: If True, also log to a file named '{name}.log'.
+
+    Returns:
+        Configured logger instance.
+
+    Examples:
+        >>> logger = get_logger("my_logger", level=logging.DEBUG, log_to_file=True)
+        >>> logger.debug("This is a debug message.")
+    """
+    logger = logging.getLogger(name)
+    if logger.handlers:
+        return logger  # Already configured
+    logger.setLevel(level)
+    formatter = ColorFormatter("%(asctime)s", datefmt="%H:%M:%S")
+    stream_handler = logging.StreamHandler(sys.stdout)
+    stream_handler.setFormatter(formatter)
+    logger.addHandler(stream_handler)
+    if log_to_file:
+        file_handler = logging.FileHandler(f"{name}.log")
+        file_handler.setFormatter(formatter)
+        logger.addHandler(file_handler)
+    return logger
+
 
 def init_logger(
     logger_name: str,
@@ -95,3 +163,6 @@ def log_execution(func: Callable) -> Callable:
         return result
 
     return wrapper
+
+    return wrapper
+    return wrapper

{bs_python_utils-0.8.2 → bs_python_utils-0.8.3}/bs_python_utils.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: bs-python-utils
-Version: 0.8.2
+Version: 0.8.3
 Summary: Utilities programs for my Python code
 Requires-Python: >=3.12
 Description-Content-Type: text/markdown
@@ -24,6 +24,7 @@ Requires-Dist: seaborn>=0.13.2
 Requires-Dist: statsmodels>=0.14.5
 Requires-Dist: streamlit>=1.49.1
 Requires-Dist: vega-datasets>=0.9.0
+Requires-Dist: colorama>=0.4.6
 Dynamic: license-file
 
 ## bs-python-utils

{bs_python_utils-0.8.2 → bs_python_utils-0.8.3}/bs_python_utils.egg-info/requires.txt

@@ -17,3 +17,4 @@ seaborn>=0.13.2
 statsmodels>=0.14.5
 streamlit>=1.49.1
 vega-datasets>=0.9.0
+colorama>=0.4.6

{bs_python_utils-0.8.2 → bs_python_utils-0.8.3}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "bs-python-utils"
-version = "0.8.2"
+version = "0.8.3"
 description = "Utilities programs for my Python code"
 readme = "README.md"
 requires-python = ">=3.12"
@@ -24,6 +24,7 @@ dependencies = [
     "statsmodels>=0.14.5",
     "streamlit>=1.49.1",
     "vega-datasets>=0.9.0",
+    "colorama>=0.4.6",
 ]
 
 [tool.setuptools]