100 lines
4.2 KiB
Python
100 lines
4.2 KiB
Python
|
from Log4p.plugins_for_core import *
|
|||
|
|
|
|||
|
|
class LogManager:
|
|||
|
|
def __init__(self) -> None:
|
|||
|
|
self.public_formatter = logging.Formatter(
|
|||
|
|
fmt='[%(asctime)s][%(name)s/%(levelname)s][%(funcName)s]:%(message)s',
|
|||
|
|
datefmt='%H:%M:%S'
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
def GetLogger(self, log_name: str = "default",
|
|||
|
|
out_to_console: bool = True,
|
|||
|
|
web_log_mode: bool = False,
|
|||
|
|
WSpost_url: str = "",
|
|||
|
|
HTTPpost_url: str = "",
|
|||
|
|
http_mode: bool = False,
|
|||
|
|
custom_formatter: logging.Formatter = None):
|
|||
|
|
# 确保日志名称有效
|
|||
|
|
log_name = log_name if log_name else "default"
|
|||
|
|
if out_to_console:
|
|||
|
|
log_folder = f'./logs/{log_name}'
|
|||
|
|
if not os.path.exists(log_folder):
|
|||
|
|
os.makedirs(log_folder, exist_ok=True)
|
|||
|
|
|
|||
|
|
logger = logging.getLogger(log_name)
|
|||
|
|
if logger.hasHandlers():
|
|||
|
|
# Logger已经配置过处理器，避免重复配置
|
|||
|
|
return logger
|
|||
|
|
|
|||
|
|
# 颜色配置
|
|||
|
|
log_color_config = {
|
|||
|
|
'DEBUG': 'bold_blue', 'INFO': 'bold_cyan',
|
|||
|
|
'WARNING': 'bold_yellow', 'ERROR': 'red',
|
|||
|
|
'CRITICAL': 'bold_red', 'RESET': 'reset',
|
|||
|
|
'asctime': 'green'
|
|||
|
|
}
|
|||
|
|
if out_to_console:
|
|||
|
|
console_handler = logging.StreamHandler()
|
|||
|
|
console_handler.setLevel(logging.DEBUG)
|
|||
|
|
console_formatter = colorlog.ColoredFormatter(
|
|||
|
|
fmt='%(log_color)s [%(asctime)s][%(name)s/%(levelname)s][%(funcName)s]:%(message)s %(reset)s',
|
|||
|
|
datefmt='%H:%M:%S',
|
|||
|
|
log_colors=log_color_config
|
|||
|
|
)
|
|||
|
|
if custom_formatter:
|
|||
|
|
console_formatter = custom_formatter
|
|||
|
|
|
|||
|
|
if isinstance(console_handler, logging.StreamHandler):
|
|||
|
|
console_formatter = colorlog.ColoredFormatter(fmt=f"%(log_color)s {console_formatter._fmt} %(reset)s",datefmt=console_formatter.datefmt, log_colors=log_color_config)
|
|||
|
|
console_handler.setFormatter(console_formatter)
|
|||
|
|
|
|||
|
|
logger.setLevel(logging.DEBUG)
|
|||
|
|
logger.addHandler(console_handler)
|
|||
|
|
|
|||
|
|
|
|||
|
|
file_handler = logging.FileHandler(
|
|||
|
|
filename=f'logs/{log_name}/{log_name}.log', mode='a', encoding='utf-8')
|
|||
|
|
file_handler.setLevel(logging.INFO)
|
|||
|
|
file_handler.setFormatter(self.public_formatter)
|
|||
|
|
|
|||
|
|
if custom_formatter:
|
|||
|
|
file_handler.setFormatter(custom_formatter)
|
|||
|
|
|
|||
|
|
# 检查代码是否在异步环境中运行
|
|||
|
|
if asyncio.iscoroutinefunction(logging.Handler.emit):
|
|||
|
|
queue = asyncio.Queue()
|
|||
|
|
queue_handler = QueueHandler(queue)
|
|||
|
|
queue_listener = QueueListener(queue, file_handler)
|
|||
|
|
logger.addHandler(queue_handler)
|
|||
|
|
asyncio.ensure_future(queue_listener.start())
|
|||
|
|
else:
|
|||
|
|
logger.addHandler(file_handler)
|
|||
|
|
|
|||
|
|
if web_log_mode and WSpost_url:
|
|||
|
|
websocket_handler = WebsocketHandler(WSpost_url)
|
|||
|
|
websocket_handler.setLevel(logging.INFO)
|
|||
|
|
formatter = self.public_formatter
|
|||
|
|
if custom_formatter:
|
|||
|
|
formatter = custom_formatter
|
|||
|
|
websocket_handler.setFormatter(formatter)
|
|||
|
|
logger.addHandler(websocket_handler)
|
|||
|
|
|
|||
|
|
if http_mode and HTTPpost_url:
|
|||
|
|
# 检查代码是否在异步环境中运行
|
|||
|
|
if asyncio.iscoroutinefunction(logging.Handler.emit):
|
|||
|
|
async_http_hander = AsyncHTTPhandler(HTTPpost_url)
|
|||
|
|
async_http_hander.setLevel(logging.INFO)
|
|||
|
|
formatter = self.public_formatter
|
|||
|
|
if custom_formatter:
|
|||
|
|
formatter = custom_formatter
|
|||
|
|
async_http_hander.setFormatter(formatter)
|
|||
|
|
logger.addHandler(async_http_hander)
|
|||
|
|
http_handler = HTTPhandler(HTTPpost_url)
|
|||
|
|
http_handler.setLevel(logging.INFO)
|
|||
|
|
formatter = self.public_formatter
|
|||
|
|
if custom_formatter:
|
|||
|
|
formatter = custom_formatter
|
|||
|
|
http_handler.setFormatter(formatter)
|
|||
|
|
logger.addHandler(http_handler)
|
|||
|
|
|
|||
|
|
return logger
|