diff --git a/hypy_utils/logging_utils.py b/hypy_utils/logging_utils.py new file mode 100644 index 0000000..3b663b7 --- /dev/null +++ b/hypy_utils/logging_utils.py @@ -0,0 +1,24 @@ +import logging +import os + + +def setup_logger(debug: bool = os.environ.get("DEBUG", False)): + # Try to use rich for pretty printing + try: + from rich.logging import RichHandler + handler = RichHandler(rich_tracebacks=True) + + from rich.traceback import install + install(show_locals=True) + except ImportError: + handler = logging.StreamHandler() + + # Initialize debug logger + logging.basicConfig( + level="NOTSET" if debug else "INFO", + format="%(message)s", + datefmt="[%X]", + handlers=[handler] + ) + + return logging.getLogger("a2")