93 lines
2.8 KiB
Python
93 lines
2.8 KiB
Python
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")
|