AI_customer_service/utils/logger_config.py

import os
import logging
from colorama import Fore, Style, init

current_dir = os.path.dirname(os.path.abspath(__file__))
log_path = os.path.join(current_dir, "..", "logs", "intent.log")

# 初始化 colorama
init(autoreset=True)

class ColoredFormatter(logging.Formatter):
    COLORS = {
        'DEBUG': Fore.CYAN,
        'INFO': Fore.GREEN,
        'WARNING': Fore.YELLOW,
        'ERROR': Fore.RED,
        'CRITICAL': Fore.RED + Style.BRIGHT,
    }

    def format(self, record):
        levelname = record.levelname
        if levelname in self.COLORS:
            record.levelname = f"{self.COLORS[levelname]}{levelname}{Style.RESET_ALL}"
            record.msg = f"{self.COLORS[levelname]}{record.msg}{Style.RESET_ALL}"
        return super().format(record)

def setup_logger(name):
    # 配置日志
    logger = logging.getLogger(name)
    logger.setLevel(logging.DEBUG)

    # 创建控制台处理器
    # console_handler = logging.StreamHandler()
    # console_handler.setLevel(logging.DEBUG)

    # 创建文件处理器
    file_handler = logging.FileHandler(log_path, encoding='utf-8') 
    file_handler.setLevel(logging.DEBUG) 

    # 创建格式化器
    formatter = ColoredFormatter('%(asctime)s - %(levelname)s - %(funcName)s - %(message)s')
    # console_handler.setFormatter(formatter)
    file_handler.setFormatter(formatter) 

    # 将处理器添加到日志记录器
    # logger.addHandler(console_handler)
    logger.addHandler(file_handler) 

    return logger