|
| 1 | +# simple_logger.py |
| 2 | +""" |
| 3 | +Minimal logger setup that encourages per-module loggers. |
| 4 | +
|
| 5 | +Usage: |
| 6 | + from openhands.core.logger import get_logger |
| 7 | + logger = get_logger(__name__) |
| 8 | + logger.info("Hello from this module!") |
| 9 | +""" |
| 10 | + |
| 11 | +import logging |
| 12 | +import os |
| 13 | +from logging.handlers import TimedRotatingFileHandler |
| 14 | + |
| 15 | +# ========= ENV (loaded at import) ========= |
| 16 | +LEVEL_MAP = ( |
| 17 | + logging.getLevelNamesMapping() |
| 18 | + if hasattr(logging, "getLevelNamesMapping") |
| 19 | + else logging._nameToLevel |
| 20 | +) |
| 21 | + |
| 22 | +ENV_LOG_LEVEL_STR = os.getenv("LOG_LEVEL", "INFO").upper() |
| 23 | +ENV_LOG_LEVEL = LEVEL_MAP.get(ENV_LOG_LEVEL_STR, logging.INFO) |
| 24 | +ENV_LOG_TO_FILE = os.getenv("LOG_TO_FILE", "false").lower() in {"1", "true", "yes"} |
| 25 | +ENV_LOG_DIR = os.getenv("LOG_DIR", "logs") |
| 26 | +ENV_ROTATE_WHEN = os.getenv("LOG_ROTATE_WHEN", "midnight") |
| 27 | +ENV_BACKUP_COUNT = int(os.getenv("LOG_BACKUP_COUNT", "7")) |
| 28 | +ENV_FORMAT = os.getenv( |
| 29 | + "LOG_FORMAT", |
| 30 | + "%(asctime)s | %(levelname)s | %(name)s | %(pathname)s:%(lineno)d | %(message)s", |
| 31 | +) |
| 32 | +ENV_AUTO_CONFIG = os.getenv("LOG_AUTO_CONFIG", "true").lower() in {"1", "true", "yes"} |
| 33 | + |
| 34 | + |
| 35 | +# ========= SETUP ========= |
| 36 | +def setup_logging( |
| 37 | + level: int | None = None, |
| 38 | + log_to_file: bool | None = None, |
| 39 | + log_dir: str | None = None, |
| 40 | + fmt: str | None = None, |
| 41 | + when: str | None = None, |
| 42 | + backup_count: int | None = None, |
| 43 | +) -> None: |
| 44 | + """Configure the root logger. All child loggers inherit this setup.""" |
| 45 | + lvl = ENV_LOG_LEVEL if level is None else level |
| 46 | + to_file = ENV_LOG_TO_FILE if log_to_file is None else log_to_file |
| 47 | + directory = ENV_LOG_DIR if log_dir is None else log_dir |
| 48 | + format_str = ENV_FORMAT if fmt is None else fmt |
| 49 | + rotate_when = ENV_ROTATE_WHEN if when is None else when |
| 50 | + keep = ENV_BACKUP_COUNT if backup_count is None else backup_count |
| 51 | + |
| 52 | + root = logging.getLogger() |
| 53 | + root.setLevel(lvl) |
| 54 | + root.handlers = [] # reset |
| 55 | + |
| 56 | + formatter = logging.Formatter(format_str) |
| 57 | + |
| 58 | + ch = logging.StreamHandler() |
| 59 | + ch.setLevel(lvl) |
| 60 | + ch.setFormatter(formatter) |
| 61 | + root.addHandler(ch) |
| 62 | + |
| 63 | + if to_file: |
| 64 | + os.makedirs(directory, exist_ok=True) |
| 65 | + fh = TimedRotatingFileHandler( |
| 66 | + os.path.join(directory, "app.log"), |
| 67 | + when=rotate_when, |
| 68 | + backupCount=keep, |
| 69 | + encoding="utf-8", |
| 70 | + ) |
| 71 | + fh.setLevel(lvl) |
| 72 | + fh.setFormatter(formatter) |
| 73 | + root.addHandler(fh) |
| 74 | + |
| 75 | + |
| 76 | +def get_logger(name: str) -> logging.Logger: |
| 77 | + """Return a logger for the given module name.""" |
| 78 | + return logging.getLogger(name) |
| 79 | + |
| 80 | + |
| 81 | +# Auto-configure if desired |
| 82 | +if ENV_AUTO_CONFIG: |
| 83 | + setup_logging() |
0 commit comments