给 Python 入口函数加可选 logger，兼容被调用与独立运行

Administrator 字数: 945 阅读耗时: 2 分钟 2024/11/01 博客独享热度: 7

工具函数既可能 python app.py 单独跑，也可能被调度器/其他脚本 import。把 logger 做成可选参数最省事。

推荐结构

import logging

def setup_logging() -> logging.Logger:
    logging.basicConfig(level=logging.INFO, format="%(asctime)s %(levelname)s %(name)s: %(message)s")
    return logging.getLogger("MainScriptLogger")

def main(logger: logging.Logger | None = None) -> None:
    if logger is None:
        logger = logging.getLogger("MainScriptLogger")
        if not logger.handlers:
            logger = setup_logging()
    logger.info("脚本开始执行")
    try:
        ...
        logger.info("脚本执行完成")
    except Exception:
        logger.exception("脚本执行失败")
        raise

if __name__ == "__main__":
    main(setup_logging())