春河晴andClaude
fdc098d0db
- 修复异常处理链，使用from语法保留原始异常 - 格式化代码以符合项目规范 - 优化导入模块的顺序 🤖 Generated with [Claude Code](https://claude.ai/code) Co-Authored-By: Claude <noreply@anthropic.com>
89 lines
2.9 KiB
Python
89 lines
2.9 KiB
Python
import sys
|
|
import loguru
|
|
from enum import Enum
|
|
|
|
|
|
class LogClassification(Enum):
|
|
BASE = "base"
|
|
MEMORY = "memory"
|
|
EMOJI = "emoji"
|
|
CHAT = "chat"
|
|
PBUILDER = "promptbuilder"
|
|
|
|
|
|
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> | "
|
|
" d<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>"
|
|
)
|
|
|
|
promptbuilder_format = (
|
|
"<green>{time:HH:mm}</green> | <level>{level: <8}</level> | <yellow>Prompt</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.PBUILDER:
|
|
self.logger.add(
|
|
sys.stderr,
|
|
format=promptbuilder_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
|