import os
import logging
from datetime import datetime


def get_logger(root, name=None, debug=True):
    """
    创建带时间戳的日志记录器
    :param root: 日志文件目录
    :param name: 日志名称
    :param debug: 是否开启调试模式
    """
    logger = logging.getLogger(name)
    logger.setLevel(logging.DEBUG)

    # 避免重复添加 Handler
    if logger.hasHandlers():
        logger.handlers.clear()

    # 时间格式改为 年/月/日 时:分:秒
    formatter = logging.Formatter("%(asctime)s - %(message)s", "%Y/%m/%d %H:%M:%S")

    # 控制台输出
    console_handler = logging.StreamHandler()
    console_handler.setFormatter(formatter)
    console_handler.setLevel(logging.DEBUG if debug else logging.INFO)
    logger.addHandler(console_handler)

    # 文件输出（仅非 debug 模式）
    if not debug:
        os.makedirs(root, exist_ok=True)
        logfile = os.path.join(root, "run.log")
        print(f"Create Log File in: {logfile}")
        file_handler = logging.FileHandler(logfile, mode="w", encoding="utf-8")
        file_handler.setLevel(logging.DEBUG)
        file_handler.setFormatter(formatter)
        logger.addHandler(file_handler)

    return logger


if __name__ == "__main__":
    time_str = datetime.now().strftime("%Y%m%d%H%M%S")
    print(time_str)

    logger = get_logger("./logs", debug=False)  # 改成 False 测试文件输出
    logger.debug("this is a {} debug message".format(1))
    logger.info("this is an info message")
    logger.debug("this is a debug message")
    logger.info("this is an info message")