config.py•2.02 kB
"""
配置管理模块
"""
import os
import logging
from typing import Optional
from dotenv import load_dotenv
# 加载环境变量
load_dotenv()
# 日志配置
LOG_LEVEL = os.getenv('LOG_LEVEL', 'INFO').upper()
_LEVELS = {
'CRITICAL': logging.CRITICAL,
'ERROR': logging.ERROR,
'WARNING': logging.WARNING,
'INFO': logging.INFO,
'DEBUG': logging.DEBUG,
}
# 配置日志
# 在stdio模式下，日志必须输出到stderr或文件，不能干扰stdout
TRANSPORT_TYPE = os.getenv('TRANSPORT_TYPE', 'stdio')
if TRANSPORT_TYPE == 'stdio':
# stdio模式：日志输出到stderr
import sys
logging.basicConfig(
level=_LEVELS.get(LOG_LEVEL, logging.INFO),
format='%(asctime)s - %(name)s - %(levelname)s - %(message)s',
handlers=[
logging.StreamHandler(sys.stderr) # 明确输出到stderr
]
)
else:
# HTTP模式：可以输出到stdout
logging.basicConfig(
level=_LEVELS.get(LOG_LEVEL, logging.INFO),
format='%(asctime)s - %(name)s - %(levelname)s - %(message)s',
handlers=[
logging.StreamHandler()
]
)
class Config:
"""配置类"""
# MCP服务配置
MCP_SERVER_NAME: str = os.getenv('MCP_SERVER_NAME', 'ziwei_mcp')
MCP_SERVER_VERSION: str = '1.0.0'
# 默认语言
DEFAULT_LANGUAGE: str = os.getenv('DEFAULT_LANGUAGE', 'zh-CN')
# 传输层配置
TRANSPORT_TYPE: str = os.getenv('TRANSPORT_TYPE', 'stdio')
# HTTP传输配置
HTTP_HOST: str = os.getenv('HTTP_HOST', '0.0.0.0')
HTTP_PORT: int = int(os.getenv('HTTP_PORT', '8080'))
HTTP_API_KEY: str = os.getenv('HTTP_API_KEY', '')
# WebSocket传输配置（预留）
WS_HOST: str = os.getenv('WS_HOST', '0.0.0.0')
WS_PORT: int = int(os.getenv('WS_PORT', '8081'))
@classmethod
def get_logger(cls, name: str) -> logging.Logger:
"""获取logger实例"""
return logging.getLogger(name)
# 全局配置实例
config = Config()