mirror of
https://github.com/jihe520/MathModelAgent.git
synced 2026-10-19 05:40:21 +08:00
Sanjin
1921f9da0c
add apikeydialog 在网页配置 apikey 不需要再环境变量 update docker file and github action change prompt 跳过问题核验导致不非数学建模问题
59 lines
1.6 KiB
Python
59 lines
1.6 KiB
Python
import os
|
|
import sys
|
|
import time
|
|
from loguru import logger as _logger
|
|
|
|
|
|
class LoggerInitializer:
|
|
def __init__(self):
|
|
self.log_path = os.path.join(os.getcwd(), "logs")
|
|
self.__ensure_log_directory_exists()
|
|
self.log_path_error = os.path.join(
|
|
self.log_path, f"{time.strftime('%Y-%m-%d')}_error.log"
|
|
)
|
|
|
|
def __ensure_log_directory_exists(self):
|
|
"""
|
|
确保日志目录存在，如果不存在则创建
|
|
"""
|
|
if not os.path.exists(self.log_path):
|
|
os.mkdir(self.log_path)
|
|
|
|
@staticmethod
|
|
def __filter(log: dict):
|
|
"""
|
|
自定义日志过滤器，添加trace_id
|
|
"""
|
|
return log
|
|
|
|
def init_log(self):
|
|
"""
|
|
初始化日志配置
|
|
"""
|
|
# 自定义日志格式
|
|
format_str = (
|
|
"<green>{time:YYYY-MM-DD HH:mm:ss.SSS}</green> | "
|
|
"<level>{level: <8}</level> | "
|
|
"<cyan>{name}</cyan>:<cyan>{function}</cyan>:<cyan>{line}</cyan> - "
|
|
"<level>{message}</level>"
|
|
)
|
|
_logger.remove()
|
|
# 移除后重新添加sys.stderr, 目的: 控制台输出与文件日志内容和结构一致
|
|
_logger.add(sys.stderr, filter=self.__filter, format=format_str, enqueue=False)
|
|
_logger.add(
|
|
self.log_path_error,
|
|
filter=self.__filter,
|
|
format=format_str,
|
|
rotation="50MB",
|
|
encoding="utf-8",
|
|
enqueue=False,
|
|
compression="zip",
|
|
)
|
|
|
|
return _logger
|
|
|
|
|
|
# 初始化日志处理器
|
|
log_initializer = LoggerInitializer()
|
|
logger = log_initializer.init_log()
|