import logging
import os.path
from logging import StreamHandler, FileHandler
from colorama import Fore, Style
from datetime import datetime

定义一个新的日志级别方法

def notice(self, message, args, *kwargs):

if self.isEnabledFor(NOTICE_LEVEL):
    self._log(NOTICE_LEVEL, message, args, **kwargs)

将该方法绑定到 logger 对象

logging.Logger.notice = notice

NOTICE_LEVEL = 19 # < NOTICE(19) < INFO(20) < WARNING(30)
logging.addLevelName(NOTICE_LEVEL, "NOTICE")

class ColorFormatter(logging.Formatter):

"""日志格式化，支持不同级别的颜色"""
COLOR_MAP = {
    logging.DEBUG: Fore.CYAN,
    logging.INFO: Fore.GREEN,
    logging.WARNING: Fore.YELLOW,
    NOTICE_LEVEL: Fore.BLUE,  # NOTICE 级别使用蓝色
    logging.ERROR: Fore.RED,
    logging.CRITICAL: Fore.MAGENTA + Style.BRIGHT,
}

RESET = Style.RESET_ALL

def format(self, record):
    log_color = self.COLOR_MAP.get(record.levelno, "")
    log_fmt = f"{log_color}[%(asctime)s] [%(levelname)s] [%(filename)s:%(lineno)d] %(message)s{self.RESET}"
    formatter = logging.Formatter(log_fmt, datefmt="%Y-%m-%d %H:%M:%S")
    return formatter.format(record)

def notice(self, message, args, *kwargs):

if self.isEnabledFor(NOTICE_LEVEL):
    self._log(NOTICE_LEVEL, message, args, **kwargs)

logging.Logger.notice = notice # 绑定到 Logger

def log_handler(log_file, level):

today = datetime.today().strftime('%Y-%m-%d')

file_format = logging.Formatter(f"[%(asctime)s] [%(levelname)s] [%(filename)s:%(lineno)d] %(message)s",
                                datefmt="%Y-%m-%d %H:%M:%S")

file_handler = FileHandler(f"{log_file}_{today}_{level}.log", mode="a", encoding="utf-8")
if level == 'info':
    file_handler.setLevel(logging.INFO)
elif level == 'debug':
    file_handler.setLevel(logging.DEBUG)
elif level == 'notice':
    file_handler.setLevel(NOTICE_LEVEL)
elif level == 'warning':
    file_handler.setLevel(logging.WARNING)
else:
    file_handler.setLevel(logging.NOTSET)
file_handler.setFormatter(file_format)

return file_handler

def setup_logger(log_dir = 'logs', log_file='app', task_id=0, uuid=None):

"""
设置全局 logger，支持：
1. 控制台日志（带颜色）
2. 文件日志（info, error）
3. 远程服务器日志上报（仅 NOTICE 级别）
"""
global logger, _task_id, _uuid
if task_id:
    _task_id = task_id
if uuid:
    _uuid = uuid
log_file = os.path.join(log_dir,log_file)
new_logger = logging.getLogger("custom_logger")
new_logger.setLevel(logging.DEBUG)  # 设置最低日志级别

# 清除旧的处理器
if new_logger.handlers:
    for handler in new_logger.handlers[:]:
        new_logger.removeHandler(handler)

# 控制台处理器
console_handler = StreamHandler()
console_handler.setLevel(logging.DEBUG)
console_handler.setFormatter(ColorFormatter())
new_logger.addHandler(console_handler)
if log_file:
    new_logger.addHandler(log_handler(log_file, 'debug'))
    new_logger.addHandler(log_handler(log_file, 'error'))
    new_logger.addHandler(log_handler(log_file, 'info'))
    new_logger.addHandler(log_handler(log_file, 'notice'))

# ✅ 替换 `logger` 的内容，而不改变 `logger` 的引用
logger.__dict__.update(new_logger.__dict__)

初始化全局 logger

logger: logging.Logger
setup_logger()