83 lines
2.7 KiB
Python
83 lines
2.7 KiB
Python
import logging
|
||
import sys
|
||
import os
|
||
|
||
TRACE = 5
|
||
DEBUG = logging.DEBUG
|
||
INFO = logging.INFO
|
||
WARNING = logging.WARNING
|
||
ERROR = logging.ERROR
|
||
CRITICAL = logging.CRITICAL
|
||
NOTSET = logging.NOTSET
|
||
|
||
def get_logger(name=None):
|
||
if name is None:
|
||
name = 'alkaid_release_platform'
|
||
# 创建一个 logger
|
||
logger = logging.getLogger(name)
|
||
if logger.handlers:
|
||
return logger
|
||
|
||
#定义一个新的日志级别，用于输出trace信息
|
||
logging.TRACE = TRACE
|
||
logging.addLevelName(logging.TRACE, "TRACE")
|
||
def trace(self, message, *args, **kws):
|
||
self._log(logging.TRACE, message, args, **kws)
|
||
logging.Logger.trace = trace
|
||
|
||
logger.setLevel(INFO)
|
||
class LogColors:
|
||
RESET = "\033[0m"
|
||
TRACE = "\033[35m" # 紫色
|
||
DEBUG = "\033[34m" # 蓝色
|
||
INFO = "\033[32m" # 绿色
|
||
WARNING = "\033[33m" # 黄色
|
||
ERROR = "\033[31m" # 红色
|
||
CRITICAL = "\033[41m" # 红底白字
|
||
|
||
# 创建一个log的颜色格式化器，使得输出的log信息在控制台上有颜色区分
|
||
class ColoredFormatter(logging.Formatter):
|
||
def __init__(self, msg, use_color=True):
|
||
logging.Formatter.__init__(self, msg)
|
||
self.use_color = use_color
|
||
def format(self, record):
|
||
level_color = {
|
||
logging.TRACE: LogColors.TRACE,
|
||
logging.DEBUG: LogColors.DEBUG,
|
||
logging.INFO: LogColors.INFO,
|
||
logging.WARNING: LogColors.WARNING,
|
||
logging.ERROR: LogColors.ERROR,
|
||
logging.CRITICAL: LogColors.CRITICAL
|
||
}
|
||
color = level_color.get(record.levelno, LogColors.RESET)
|
||
message = super().format(record)
|
||
return f"{color}{message}{LogColors.RESET}" if os.isatty(sys.stdout.fileno()) else message
|
||
|
||
# 创建一个带颜色的控制台处理器
|
||
console_handler = logging.StreamHandler(sys.stdout)
|
||
console_handler.setLevel(INFO)
|
||
# 创建一个带颜色的格式化器并将其添加到处理器
|
||
formatter = ColoredFormatter('[%(asctime)s][%(levelname)s][%(filename)s:%(lineno)d]: %(message)s')
|
||
console_handler.setFormatter(formatter)
|
||
|
||
# 将处理器添加到 logger
|
||
logger.addHandler(console_handler)
|
||
|
||
# 定义一个方法，用于设置日志级别，并将其绑定到 logger 对象
|
||
def set_level(self, level):
|
||
lv_map = {
|
||
"trace":TRACE,
|
||
"debug":DEBUG,
|
||
"info":INFO,
|
||
"warning":WARNING,
|
||
"error":ERROR,
|
||
"critical":CRITICAL
|
||
}
|
||
level = lv_map.get(level, DEBUG)
|
||
for handler in self.handlers:
|
||
handler.setLevel(level)
|
||
logger.setLevel(level)
|
||
logging.Logger.set_level = set_level
|
||
return logger
|
||
|