import logging.handlers
import os
from config import DEBUG

log_dir = "./logs"
log_max_bytes = 1024 * 1024 * 10
log_backup_count = 5


def get_logger(name, level=None):
    if not level:
        level = logging.DEBUG if DEBUG else logging.INFO

    # 配置日志
    logger = logging.getLogger(name)
    logger.setLevel(level)
    # 检查日志目录是否存在，如果不存在则创建
    if not os.path.exists(log_dir):
        os.makedirs(log_dir)

    # 创建一个handler，用于写入日志文件
    file_handler = logging.handlers.RotatingFileHandler(f'./{log_dir}/{name}.log', maxBytes=log_max_bytes,
                                                        backupCount=log_backup_count,encoding='utf-8')
    file_handler.setLevel(level)

    # 定义handler的输出格式
    formatter = logging.Formatter('%(asctime)s - %(name)s - %(levelname)s - %(message)s')
    file_handler.setFormatter(formatter)

    # 给logger添加handler
    logger.addHandler(file_handler)
    return logger


# 定义一个模块级别的变量来存储日志记录器实例
_app_logger = None


def get_app_logger():
    global _app_logger
    if _app_logger is None:
        _app_logger = get_logger("app")
    return _app_logger