#########################################################
## @file   : logger_utils.py
## @desc   : 日志公共类
## @create : 2025/6/22
## @author : Chengan，doubao AI
## @email  : douboer@gmail.com
#########################################################

import logging

# 公共的日志类
class CommonLogger:
    def __init__(self, log_file=None):
        self.logger = logging.getLogger(__name__)
        self.logger.setLevel(logging.INFO)

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

        # 创建文件处理器（如果指定了日志文件）
        if log_file:
            file_handler = logging.FileHandler(log_file)
            file_handler.setLevel(logging.DEBUG)
            self.logger.addHandler(file_handler)

        # 设置日志格式
        formatter = logging.Formatter('%(asctime)s - %(name)s - %(levelname)s - %(message)s')
        console_handler.setFormatter(formatter)
        if log_file:
            file_handler.setFormatter(formatter)

        # 添加处理器
        self.logger.addHandler(console_handler)

    def get_logger(self):
        return self.logger