117 lines
3.4 KiB
Python
117 lines
3.4 KiB
Python
|
import logging
|
|||
|
|
import logging.config
|
|||
|
|
import os
|
|||
|
|
from datetime import datetime
|
|||
|
|
|
|||
|
|
|
|||
|
|
def setup_logging(
|
|||
|
|
level: str = None,
|
|||
|
|
log_file: str = None,
|
|||
|
|
enable_file_logging: bool = False
|
|||
|
|
):
|
|||
|
|
"""
|
|||
|
|
设置日志配置
|
|||
|
|
|
|||
|
|
Args:
|
|||
|
|
level: 日志级别 (DEBUG, INFO, WARNING, ERROR, CRITICAL)
|
|||
|
|
log_file: 日志文件名
|
|||
|
|
enable_file_logging: 是否启用文件日志
|
|||
|
|
"""
|
|||
|
|
# 如果没有指定级别，从环境变量获取，默认INFO
|
|||
|
|
if level is None:
|
|||
|
|
level = os.getenv('LOG_LEVEL', 'INFO').upper()
|
|||
|
|
|
|||
|
|
# 默认日志格式
|
|||
|
|
log_format = '%(asctime)s - %(name)s - %(levelname)s - %(message)s'
|
|||
|
|
date_format = '%Y-%m-%d %H:%M:%S'
|
|||
|
|
|
|||
|
|
# 基本配置
|
|||
|
|
config = {
|
|||
|
|
'version': 1,
|
|||
|
|
'disable_existing_loggers': False,
|
|||
|
|
'formatters': {
|
|||
|
|
'standard': {
|
|||
|
|
'format': log_format,
|
|||
|
|
'datefmt': date_format
|
|||
|
|
},
|
|||
|
|
'detailed': {
|
|||
|
|
'format': '%(asctime)s - %(name)s - %(levelname)s - %(module)s - %(funcName)s - %(lineno)d - %(message)s',
|
|||
|
|
'datefmt': date_format
|
|||
|
|
}
|
|||
|
|
},
|
|||
|
|
'handlers': {
|
|||
|
|
'console': {
|
|||
|
|
'level': level,
|
|||
|
|
'class': 'logging.StreamHandler',
|
|||
|
|
'formatter': 'standard',
|
|||
|
|
'stream': 'ext://sys.stdout'
|
|||
|
|
}
|
|||
|
|
},
|
|||
|
|
'root': {
|
|||
|
|
'level': level,
|
|||
|
|
'handlers': ['console']
|
|||
|
|
},
|
|||
|
|
'loggers': {
|
|||
|
|
# 针对特定模块的日志配置
|
|||
|
|
'memory_integration': {
|
|||
|
|
'level': level,
|
|||
|
|
'handlers': ['console'],
|
|||
|
|
'propagate': False
|
|||
|
|
},
|
|||
|
|
'chat_service': {
|
|||
|
|
'level': level,
|
|||
|
|
'handlers': ['console'],
|
|||
|
|
'propagate': False
|
|||
|
|
},
|
|||
|
|
'rag_pipeline': {
|
|||
|
|
'level': level,
|
|||
|
|
'handlers': ['console'],
|
|||
|
|
'propagate': False
|
|||
|
|
},
|
|||
|
|
'uvicorn': {
|
|||
|
|
'level': 'INFO',
|
|||
|
|
'handlers': ['console'],
|
|||
|
|
'propagate': False
|
|||
|
|
},
|
|||
|
|
'fastapi': {
|
|||
|
|
'level': 'INFO',
|
|||
|
|
'handlers': ['console'],
|
|||
|
|
'propagate': False
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
# 如果启用文件日志
|
|||
|
|
if enable_file_logging:
|
|||
|
|
if log_file is None:
|
|||
|
|
# 默认日志文件名
|
|||
|
|
log_dir = 'logs'
|
|||
|
|
os.makedirs(log_dir, exist_ok=True)
|
|||
|
|
log_file = os.path.join(
|
|||
|
|
log_dir,
|
|||
|
|
f"app_{datetime.now().strftime('%Y%m%d')}.log"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
# 添加文件处理器
|
|||
|
|
config['handlers']['file'] = {
|
|||
|
|
'level': 'DEBUG', # 文件日志记录所有级别
|
|||
|
|
'class': 'logging.handlers.RotatingFileHandler',
|
|||
|
|
'filename': log_file,
|
|||
|
|
'maxBytes': 10485760, # 10MB
|
|||
|
|
'backupCount': 5,
|
|||
|
|
'formatter': 'detailed',
|
|||
|
|
'encoding': 'utf-8'
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
# 为所有logger添加文件处理器
|
|||
|
|
for logger_name in config['loggers']:
|
|||
|
|
config['loggers'][logger_name]['handlers'].append('file')
|
|||
|
|
config['root']['handlers'].append('file')
|
|||
|
|
|
|||
|
|
# 应用配置
|
|||
|
|
logging.config.dictConfig(config)
|
|||
|
|
|
|||
|
|
|
|||
|
|
def get_logger(name: str) -> logging.Logger:
|
|||
|
|
"""获取指定名称的logger"""
|
|||
|
|
return logging.getLogger(name)
|