import logging
import sys
from fileinput import filename

from colorlog import ColoredFormatter

# 创建一个 logger
logger = logging.getLogger('my_logger')
logger.setLevel(logging.DEBUG)

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

# 定义 SUCCESS 级别的数值 (INFO=20, WARNING=30，所以 SUCCESS 可以设为 25)
SUCCESS = 25
logging.addLevelName(SUCCESS, "SUCCESS")

# 定义一个 success 方法
def success(self, message, *args, **kwargs):
    if self.isEnabledFor(SUCCESS):
        self._log(SUCCESS, message, args, **kwargs)

# 绑定到 Logger 类上
logging.Logger.success = success


# 创建带颜色的格式化器（用于控制台输出）
console_formatter = ColoredFormatter(
    "%(log_color)s%(asctime)s - %(filename)s:%(lineno)d - [%(levelname)s]:%(message)s",
    datefmt="%Y-%m-%d %H:%M:%S",
    reset=True,
    log_colors={
        'SUCCESS': 'green',  # 这里定义 SUCCESS 级别的颜色
        'DEBUG':    'cyan',
        'INFO':     'white',
        'WARNING':  'yellow',
        'ERROR':    'red',
        'CRITICAL': 'red,bg_black',
    },
    secondary_log_colors={
        'message': {
            'SUCCESS': 'green',  # 这里定义 SUCCESS 级别的消息颜色
            'DEBUG':    'cyan',
            'INFO':     'white',
            'WARNING':  'yellow',
            'ERROR':    'red',
            'CRITICAL': 'red,bg_black',
        }
    },
    style='%'
)

# 创建普通格式的格式化器（用于日志文件）
file_formatter = logging.Formatter(
    "%(asctime)s - %(filename)s:%(lineno)d - %(levelname)s:%(name)s:%(message)s",
    datefmt="%Y-%m-%d %H:%M:%S"
)

# 设置格式化器
console_handler.setFormatter(console_formatter)

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

def set_logger_file_handler(p):
    # 创建一个文件处理器，保存日志到本地文件
    file_handler = logging.FileHandler(f'{p}/run.log', encoding='utf-8')
    file_handler.setFormatter(file_formatter)
    logger.addHandler(file_handler)

    # 重定向 stdout 和 stderr 到日志系统
    class LoggerWriter:
        def __init__(self, level):
            self.level = level

        def write(self, message):
            if message.strip():  # 过滤掉空行
                self.level(message.strip())  # 去掉首尾空白字符

        def flush(self):
            pass

    sys.stdout = LoggerWriter(logger.info)
    sys.stderr = LoggerWriter(logger.error)

# 测试日志
# logger.success("This is a debug message")
# logger.debug("This is a debug message")
# logger.info("This is an info message")
# logger.warning("This is a warning message")
# logger.error("This is an error message")
# logger.critical("This is a critical message")