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()