modules/infographics_generator/utils/logger.py

import os
import logging
from logging import getLogger

def setup_logger():
    """Configure and return the logger for infographics generator"""
    # 创建tmp目录（如果不存在）
    os.makedirs("tmp", exist_ok=True)

    # 配置日志
    logger = getLogger(__name__)
    logger.setLevel(logging.INFO)

    # 移除所有现有的处理器
    for handler in logger.handlers[:]:
        logger.removeHandler(handler)

    # 创建文件处理器
    file_handler = logging.FileHandler('tmp/log.txt')
    file_handler.setLevel(logging.INFO)

    # 创建格式化器
    formatter = logging.Formatter('%(asctime)s - %(name)s - %(levelname)s - %(message)s')
    file_handler.setFormatter(formatter)

    # 添加处理器到logger
    logger.addHandler(file_handler)
    
    return logger