mirror of https://github.com/Mai-with-u/MaiBot.git
优雅的日志系统！
UnCLAS-Prommer
parent
32c6d7ea55
commit
02eafe214b
12
bot.py
12
bot.py
|
|
@ -10,9 +10,9 @@ import uvicorn
|
|||
from dotenv import load_dotenv
|
||||
from nonebot.adapters.onebot.v11 import Adapter
|
||||
import platform
|
||||
from src.plugins.utils.logger_config import setup_logger
|
||||
from src.plugins.utils.logger_config import LogModule, LogClassification
|
||||
|
||||
from loguru import logger
|
||||
#from loguru import logger
|
||||
|
||||
# 配置日志格式
|
||||
|
||||
|
|
@ -102,7 +102,9 @@ def load_env():
|
|||
|
||||
|
||||
def load_logger():
|
||||
setup_logger()
|
||||
global logger
|
||||
log_module = LogModule()
|
||||
logger = log_module.setup_logger(LogClassification.BASE)
|
||||
|
||||
|
||||
def scan_provider(env_config: dict):
|
||||
|
|
@ -174,8 +176,6 @@ def raw_main():
|
|||
if platform.system().lower() != "windows":
|
||||
time.tzset()
|
||||
|
||||
# 配置日志
|
||||
load_logger()
|
||||
easter_egg()
|
||||
init_config()
|
||||
init_env()
|
||||
|
|
@ -207,6 +207,8 @@ def raw_main():
|
|||
|
||||
if __name__ == "__main__":
|
||||
try:
|
||||
# 配置日志，防止Exception无法使用logger
|
||||
load_logger()
|
||||
raw_main()
|
||||
|
||||
app = nonebot.get_asgi()
|
||||
|
|
|
|||
|
|
@ -29,10 +29,11 @@ from .utils_image import image_path_to_base64
|
|||
from .utils_user import get_user_nickname, get_user_cardname, get_groupname
|
||||
from .willing_manager import willing_manager # 导入意愿管理器
|
||||
from .message_base import UserInfo, GroupInfo, Seg
|
||||
from ..utils.logger_config import setup_logger, LogModule
|
||||
from ..utils.logger_config import LogClassification, LogModule
|
||||
|
||||
# 配置日志
|
||||
logger = setup_logger(LogModule.CHAT)
|
||||
log_module = LogModule()
|
||||
logger = log_module.setup_logger(LogClassification.CHAT)
|
||||
|
||||
|
||||
class ChatBot:
|
||||
|
|
|
|||
|
|
@ -9,7 +9,7 @@ from typing import Optional, Tuple
|
|||
from PIL import Image
|
||||
import io
|
||||
|
||||
from loguru import logger
|
||||
# from loguru import logger
|
||||
from nonebot import get_driver
|
||||
|
||||
from ...common.database import db
|
||||
|
|
@ -18,10 +18,11 @@ from ..chat.utils import get_embedding
|
|||
from ..chat.utils_image import ImageManager, image_path_to_base64
|
||||
from ..models.utils_model import LLM_request
|
||||
|
||||
from ..utils.logger_config import setup_logger, LogModule
|
||||
from ..utils.logger_config import LogClassification, LogModule
|
||||
|
||||
# 配置日志
|
||||
logger = setup_logger(LogModule.EMOJI)
|
||||
log_module = LogModule()
|
||||
logger = log_module.setup_logger(LogClassification.EMOJI)
|
||||
|
||||
driver = get_driver()
|
||||
config = driver.config
|
||||
|
|
|
|||
|
|
@ -19,10 +19,11 @@ from ..chat.utils import (
|
|||
)
|
||||
from ..models.utils_model import LLM_request
|
||||
|
||||
from ..utils.logger_config import setup_logger, LogModule
|
||||
from ..utils.logger_config import LogClassification, LogModule
|
||||
|
||||
# 配置日志
|
||||
logger = setup_logger(LogModule.MEMORY)
|
||||
log_module = LogModule()
|
||||
logger = log_module.setup_logger(LogClassification.MEMORY)
|
||||
|
||||
logger.info("初始化记忆系统")
|
||||
|
||||
|
|
|
|||
|
|
@ -1,71 +1,78 @@
|
|||
import sys
|
||||
from loguru import logger
|
||||
# from loguru import logger
|
||||
import loguru
|
||||
from enum import Enum
|
||||
|
||||
class LogModule(Enum):
|
||||
class LogClassification(Enum):
|
||||
BASE = "base"
|
||||
MEMORY = "memory"
|
||||
EMOJI = "emoji"
|
||||
CHAT = "chat"
|
||||
|
||||
def setup_logger(log_type: LogModule = LogModule.BASE):
|
||||
"""配置日志格式
|
||||
|
||||
Args:
|
||||
log_type: 日志类型，可选值：BASE(基础日志)、MEMORY(记忆系统日志)、EMOJI(表情包系统日志)
|
||||
"""
|
||||
# 移除默认的处理器
|
||||
logger.remove()
|
||||
|
||||
# 基础日志格式
|
||||
base_format = "<green>{time:HH:mm:ss}</green> | <level>{level: <8}</level> | <cyan>{name}</cyan>:<cyan>{function}</cyan>:<cyan>{line}</cyan> - <level>{message}</level>"
|
||||
|
||||
chat_format = "<green>{time:HH:mm:ss}</green> | <level>{level: <8}</level> | <cyan>{name}</cyan>:<cyan>{function}</cyan>:<cyan>{line}</cyan> - <level>{message}</level>"
|
||||
|
||||
# 记忆系统日志格式
|
||||
memory_format = "<green>{time:HH:mm}</green> | <level>{level: <8}</level> | <light-magenta>海马体</light-magenta> | <level>{message}</level>"
|
||||
|
||||
# 表情包系统日志格式
|
||||
emoji_format = "<green>{time:HH:mm}</green> | <level>{level: <8}</level> | <yellow>表情包</yellow> | <cyan>{function}</cyan>:<cyan>{line}</cyan> - <level>{message}</level>"
|
||||
# 根据日志类型选择日志格式和输出
|
||||
if log_type == LogModule.CHAT:
|
||||
logger.add(
|
||||
sys.stderr,
|
||||
format=chat_format,
|
||||
# level="INFO"
|
||||
)
|
||||
elif log_type == LogModule.MEMORY:
|
||||
# 同时输出到控制台和文件
|
||||
logger.add(
|
||||
sys.stderr,
|
||||
format=memory_format,
|
||||
# level="INFO"
|
||||
)
|
||||
logger.add(
|
||||
"logs/memory.log",
|
||||
format=memory_format,
|
||||
level="INFO",
|
||||
rotation="1 day",
|
||||
retention="7 days"
|
||||
)
|
||||
elif log_type == LogModule.EMOJI:
|
||||
logger.add(
|
||||
sys.stderr,
|
||||
format=emoji_format,
|
||||
# level="INFO"
|
||||
)
|
||||
logger.add(
|
||||
"logs/emoji.log",
|
||||
format=emoji_format,
|
||||
level="INFO",
|
||||
rotation="1 day",
|
||||
retention="7 days"
|
||||
)
|
||||
else: # BASE
|
||||
logger.add(
|
||||
sys.stderr,
|
||||
format=base_format,
|
||||
level="INFO"
|
||||
)
|
||||
|
||||
return logger
|
||||
class LogModule:
|
||||
logger = loguru.logger.opt()
|
||||
|
||||
def __init__(self):
|
||||
pass
|
||||
def setup_logger(self, log_type: LogClassification):
|
||||
"""配置日志格式
|
||||
|
||||
Args:
|
||||
log_type: 日志类型，可选值：BASE(基础日志)、MEMORY(记忆系统日志)、EMOJI(表情包系统日志)
|
||||
"""
|
||||
# 移除默认日志处理器
|
||||
self.logger.remove()
|
||||
|
||||
# 基础日志格式
|
||||
base_format = "<green>{time:HH:mm:ss}</green> | <level>{level: <8}</level> | <cyan>{name}</cyan>:<cyan>{function}</cyan>:<cyan>{line}</cyan> - <level>{message}</level>"
|
||||
|
||||
chat_format = "<green>{time:HH:mm:ss}</green> | <level>{level: <8}</level> | <cyan>{name}</cyan>:<cyan>{function}</cyan>:<cyan>{line}</cyan> - <level>{message}</level>"
|
||||
|
||||
# 记忆系统日志格式
|
||||
memory_format = "<green>{time:HH:mm}</green> | <level>{level: <8}</level> | <light-magenta>海马体</light-magenta> | <level>{message}</level>"
|
||||
|
||||
# 表情包系统日志格式
|
||||
emoji_format = "<green>{time:HH:mm}</green> | <level>{level: <8}</level> | <yellow>表情包</yellow> | <cyan>{function}</cyan>:<cyan>{line}</cyan> - <level>{message}</level>"
|
||||
# 根据日志类型选择日志格式和输出
|
||||
if log_type == LogClassification.CHAT:
|
||||
self.logger.add(
|
||||
sys.stderr,
|
||||
format=chat_format,
|
||||
# level="INFO"
|
||||
)
|
||||
elif log_type == LogClassification.MEMORY:
|
||||
|
||||
# 同时输出到控制台和文件
|
||||
self.logger.add(
|
||||
sys.stderr,
|
||||
format=memory_format,
|
||||
# level="INFO"
|
||||
)
|
||||
self.logger.add(
|
||||
"logs/memory.log",
|
||||
format=memory_format,
|
||||
level="INFO",
|
||||
rotation="1 day",
|
||||
retention="7 days"
|
||||
)
|
||||
elif log_type == LogClassification.EMOJI:
|
||||
self.logger.add(
|
||||
sys.stderr,
|
||||
format=emoji_format,
|
||||
# level="INFO"
|
||||
)
|
||||
self.logger.add(
|
||||
"logs/emoji.log",
|
||||
format=emoji_format,
|
||||
level="INFO",
|
||||
rotation="1 day",
|
||||
retention="7 days"
|
||||
)
|
||||
else: # BASE
|
||||
self.logger.add(
|
||||
sys.stderr,
|
||||
format=base_format,
|
||||
level="INFO"
|
||||
)
|
||||
|
||||
return self.logger
|
||||
|
|
|
|||
Loading…
Reference in New Issue