detquantlib 3.11.0__tar.gz → 3.12.0__tar.gz

{detquantlib-3.11.0 → detquantlib-3.12.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: detquantlib
-Version: 3.11.0
+Version: 3.12.0
 Summary: An internal library containing functions and classes that can be used across Quant models.
 License-File: LICENSE.txt
 Author: DET
@@ -11,6 +11,7 @@ Classifier: Programming Language :: Python :: 3.11
 Classifier: Programming Language :: Python :: 3.12
 Classifier: Programming Language :: Python :: 3.13
 Classifier: Programming Language :: Python :: 3.14
+Requires-Dist: colorlog (>=6.10.1,<7.0.0)
 Requires-Dist: numpy (>=2.2.6,<3.0.0)
 Requires-Dist: pandas (>=2.3.3,<3.0.0)
 Requires-Dist: paramiko (>=4.0.0,<5.0.0)

detquantlib-3.12.0/detquantlib/utils/logging.py

@@ -0,0 +1,122 @@
+# Python built-in packages
+import logging
+import sys
+import threading
+from pathlib import Path
+from types import TracebackType
+
+# Third-party packages
+from colorlog import ColoredFormatter
+
+
+def setup_logger(log_dir: Path):
+    """
+    Initializes logging with both console and file handlers.
+
+    Args:
+        log_dir: Full path to the log file
+    """
+    # Clear existing handlers to avoid duplicate logs
+    for handler in logging.root.handlers[:]:
+        logging.root.removeHandler(handler)
+
+    # Console handler with colors
+    console_formatter = CustomFormatter(
+        fmt=(
+            "%(log_color)s%(asctime)-8s%(reset)s "
+            "%(log_color)s[%(levelname)s]%(reset)s "
+            "%(log_color)s%(message)s"
+        ),
+        datefmt="%Y-%m-%d %H:%M:%S",
+        reset=True,
+    )
+    console_handler = logging.StreamHandler()
+    console_handler.setLevel(logging.INFO)
+    console_handler.setFormatter(console_formatter)
+
+    # File handler (plain, no colors)
+    file_formatter = logging.Formatter(
+        fmt="%(asctime)s [%(levelname)s] %(message)s",
+        datefmt="%Y-%m-%d %H:%M:%S",
+    )
+    file_handler = logging.FileHandler(log_dir, mode="a", encoding="utf-8")
+    file_handler.setLevel(logging.INFO)
+    file_handler.setFormatter(file_formatter)
+
+    # Configure root logger
+    root_logger = logging.getLogger()
+    root_logger.setLevel(logging.INFO)
+    root_logger.addHandler(console_handler)
+    root_logger.addHandler(file_handler)
+
+    # Capture uncaught exceptions
+    sys.excepthook = exception_handler
+    threading.excepthook = thread_exception_handler
+
+
+def exception_handler(
+    exc_type: type[BaseException], exc_value: BaseException, exc_traceback: TracebackType
+):
+    """
+    Handles uncaught exceptions in the main thread and logs them.
+
+    Args:
+        exc_type: Exception type
+        exc_value: Exception instance
+        exc_traceback: Traceback object
+    """
+    if issubclass(exc_type, KeyboardInterrupt):
+        sys.__excepthook__(exc_type, exc_value, exc_traceback)
+        return
+
+    logging.critical(
+        "Uncaught exception",
+        exc_info=(exc_type, exc_value, exc_traceback),
+    )
+
+
+def thread_exception_handler(args: threading.ExceptHookArgs):
+    """
+    Handles uncaught exceptions in threads and logs them.
+
+    Args:
+        args: Thread exception arguments
+    """
+    logging.critical(
+        f"Uncaught thread exception in {args.thread.name}",
+        exc_info=(args.exc_type, args.exc_value, args.exc_traceback),
+    )
+
+
+class CustomFormatter(ColoredFormatter):
+    """Custom formatter to override default colorlog colors."""
+
+    # ANSI color codes
+    COLORS = dict(
+        BRIGHT_WHITE="\033[97m",
+        LIGHT_WHITE="\033[38;5;250m",
+        BRIGHT_YELLOW="\033[93m",
+        RED="\033[91m",
+        RESET="\033[0m",
+    )
+
+    def _get_escape_code(self, log_colors: dict, level_name: str) -> str:
+        """
+        Returns ANSI escape codes for log levels.
+
+        Overrides DEBUG/INFO to light-white, CRITICAL to non-bold red. WARNING/ERROR remain
+        default.
+
+        Args:
+            log_colors: Mapping of log levels to colors
+            level_name: Log level name
+
+        Returns:
+            ANSI escape code string
+        """
+        if level_name in ("DEBUG", "INFO"):
+            return CustomFormatter.COLORS["LIGHT_WHITE"]
+        if level_name == "CRITICAL":
+            return super()._get_escape_code(log_colors, "ERROR")
+        else:
+            return super()._get_escape_code(log_colors, level_name)

{detquantlib-3.11.0 → detquantlib-3.12.0}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "detquantlib"
-version = "3.11.0"
+version = "3.12.0"
 description = "An internal library containing functions and classes that can be used across Quant models."
 authors = ["DET"]
 readme = "README.md"
@@ -20,6 +20,7 @@ numpy = "^2.2.6"
 pandas = "^2.3.3"
 pyarrow = "^21.0.0"
 scipy = "^1.15.2"
+colorlog = "^6.10.1"
 
 [tool.poetry.group.dev.dependencies]
 toml = "^0.10.2"