ocr_platform/ocr_utils/log_utils.py

"""
日志工具模块

提供统一的日志配置功能
"""
import sys
from pathlib import Path
from loguru import logger

def setup_logging(log_level: str = "INFO", log_file: str | None = None):
    """
    设置日志配置
    
    Args:
        log_level: 日志级别（DEBUG, INFO, WARNING, ERROR）
        log_file: 日志文件路径（可选）
    """
    logger.remove()
    
    # 控制台输出
    logger.add(
        sys.stdout,
        level=log_level,
        format="<green>{time:YYYY-MM-DD HH:mm:ss}</green> | <level>{level: <8}</level> | <cyan>{name}</cyan>:<cyan>{function}</cyan>:<cyan>{line}</cyan> - <level>{message}</level>"
    )
    
    # 文件输出
    if log_file:
        logger.add(
            log_file,
            level="DEBUG",
            format="{time:YYYY-MM-DD HH:mm:ss} | {level: <8} | {name}:{function}:{line} - {message}",
            rotation="10 MB"
        )