216 lines
6.8 KiB
Python
216 lines
6.8 KiB
Python
|
"""
|
||
|
|
Config loader (local-only).
|
||
|
|
|
||
|
|
This project is fully localized: all sensitive configuration should come from
|
||
|
|
`backend_api_python/.env` (or OS environment variables).
|
||
|
|
|
||
|
|
We keep the return shape compatible with the old PHP `loadConfig`:
|
||
|
|
flat keys like `openrouter.api_key` become nested dicts like:
|
||
|
|
{
|
||
|
|
"openrouter": {"api_key": "..."}
|
||
|
|
}
|
||
|
|
"""
|
||
|
|
from typing import Dict, Any, Optional, List, Tuple
|
||
|
|
import os
|
||
|
|
|
||
|
|
from app.utils.logger import get_logger
|
||
|
|
|
||
|
|
logger = get_logger(__name__)
|
||
|
|
|
||
|
|
# 配置缓存
|
||
|
|
_config_cache: Optional[Dict[str, Any]] = None
|
||
|
|
|
||
|
|
|
||
|
|
def load_addon_config() -> Dict[str, Any]:
|
||
|
|
"""
|
||
|
|
Build config from environment variables (.env / OS env).
|
||
|
|
|
||
|
|
NOTE: We intentionally do NOT load secrets from the database.
|
||
|
|
|
||
|
|
Returns:
|
||
|
|
Nested config dict (PHP-compatible shape)
|
||
|
|
"""
|
||
|
|
global _config_cache
|
||
|
|
|
||
|
|
# 如果缓存存在，直接返回
|
||
|
|
if _config_cache is not None:
|
||
|
|
return _config_cache
|
||
|
|
|
||
|
|
config: Dict[str, Any] = {}
|
||
|
|
|
||
|
|
def set_nested(cfg: Dict[str, Any], dotted_key: str, value: Any) -> None:
|
||
|
|
keys = dotted_key.split('.')
|
||
|
|
ref = cfg
|
||
|
|
for i, k in enumerate(keys):
|
||
|
|
if i == len(keys) - 1:
|
||
|
|
ref[k] = value
|
||
|
|
else:
|
||
|
|
if k not in ref or not isinstance(ref[k], dict):
|
||
|
|
ref[k] = {}
|
||
|
|
ref = ref[k]
|
||
|
|
|
||
|
|
def env_get(name: str) -> Optional[str]:
|
||
|
|
val = os.getenv(name)
|
||
|
|
if val is None:
|
||
|
|
return None
|
||
|
|
val = str(val).strip()
|
||
|
|
return val if val != '' else None
|
||
|
|
|
||
|
|
# Map env vars to PHP-style dotted keys.
|
||
|
|
mappings: List[Tuple[str, str, str]] = [
|
||
|
|
# internal
|
||
|
|
('INTERNAL_API_KEY', 'internal_api.key', 'string'),
|
||
|
|
|
||
|
|
# OpenRouter / LLM
|
||
|
|
('OPENROUTER_API_KEY', 'openrouter.api_key', 'string'),
|
||
|
|
('OPENROUTER_API_URL', 'openrouter.api_url', 'string'),
|
||
|
|
('OPENROUTER_MODEL', 'openrouter.model', 'string'),
|
||
|
|
('OPENROUTER_TEMPERATURE', 'openrouter.temperature', 'float'),
|
||
|
|
('OPENROUTER_MAX_TOKENS', 'openrouter.max_tokens', 'int'),
|
||
|
|
('OPENROUTER_TIMEOUT', 'openrouter.timeout', 'int'),
|
||
|
|
('OPENROUTER_CONNECT_TIMEOUT', 'openrouter.connect_timeout', 'int'),
|
||
|
|
('AI_MODELS_JSON', 'ai.models', 'json'),
|
||
|
|
|
||
|
|
# Market
|
||
|
|
('MARKET_TYPES_JSON', 'market.types', 'json'),
|
||
|
|
('TRADING_SUPPORTED_SYMBOLS_JSON', 'trading.supported_symbols', 'json'),
|
||
|
|
|
||
|
|
# App
|
||
|
|
('CORS_ORIGINS', 'app.cors_origins', 'string'),
|
||
|
|
('RATE_LIMIT', 'app.rate_limit', 'int'),
|
||
|
|
('ENABLE_CACHE', 'app.enable_cache', 'bool'),
|
||
|
|
('ENABLE_REQUEST_LOG', 'app.enable_request_log', 'bool'),
|
||
|
|
('ENABLE_AI_ANALYSIS', 'app.enable_ai_analysis', 'bool'),
|
||
|
|
|
||
|
|
# Data source common
|
||
|
|
('DATA_SOURCE_TIMEOUT', 'data_source.timeout', 'int'),
|
||
|
|
('DATA_SOURCE_RETRY', 'data_source.retry_count', 'int'),
|
||
|
|
('DATA_SOURCE_RETRY_BACKOFF', 'data_source.retry_backoff', 'float'),
|
||
|
|
|
||
|
|
# Finnhub
|
||
|
|
('FINNHUB_API_KEY', 'finnhub.api_key', 'string'),
|
||
|
|
('FINNHUB_TIMEOUT', 'finnhub.timeout', 'int'),
|
||
|
|
('FINNHUB_RATE_LIMIT', 'finnhub.rate_limit', 'int'),
|
||
|
|
|
||
|
|
# CCXT
|
||
|
|
('CCXT_DEFAULT_EXCHANGE', 'ccxt.default_exchange', 'string'),
|
||
|
|
('CCXT_TIMEOUT', 'ccxt.timeout', 'int'),
|
||
|
|
('CCXT_PROXY', 'ccxt.proxy', 'string'),
|
||
|
|
|
||
|
|
# Other sources
|
||
|
|
('YFINANCE_TIMEOUT', 'yfinance.timeout', 'int'),
|
||
|
|
('AKSHARE_TIMEOUT', 'akshare.timeout', 'int'),
|
||
|
|
('TIINGO_API_KEY', 'tiingo.api_key', 'string'),
|
||
|
|
('TIINGO_TIMEOUT', 'tiingo.timeout', 'int'),
|
||
|
|
|
||
|
|
# Search (Google CSE / Bing)
|
||
|
|
('SEARCH_PROVIDER', 'search.provider', 'string'),
|
||
|
|
('SEARCH_MAX_RESULTS', 'search.max_results', 'int'),
|
||
|
|
('SEARCH_GOOGLE_API_KEY', 'search.google.api_key', 'string'),
|
||
|
|
('SEARCH_GOOGLE_CX', 'search.google.cx', 'string'),
|
||
|
|
('SEARCH_BING_API_KEY', 'search.bing.api_key', 'string'),
|
||
|
|
]
|
||
|
|
|
||
|
|
for env_name, dotted_key, value_type in mappings:
|
||
|
|
raw = env_get(env_name)
|
||
|
|
if raw is None:
|
||
|
|
continue
|
||
|
|
try:
|
||
|
|
value = _convert_config_value(raw, value_type)
|
||
|
|
set_nested(config, dotted_key, value)
|
||
|
|
except Exception as e:
|
||
|
|
logger.warning(f"Config env parse failed: {env_name} -> {dotted_key}: {e}")
|
||
|
|
|
||
|
|
_config_cache = config
|
||
|
|
return config
|
||
|
|
|
||
|
|
|
||
|
|
def _convert_config_value(value: str, value_type: str) -> Any:
|
||
|
|
"""
|
||
|
|
根据类型转换配置值（与PHP端convertConfigValue方法保持一致）
|
||
|
|
|
||
|
|
Args:
|
||
|
|
value: 配置值字符串（可能为None）
|
||
|
|
value_type: 配置类型
|
||
|
|
|
||
|
|
Returns:
|
||
|
|
转换后的配置值
|
||
|
|
"""
|
||
|
|
# 处理 None 或空值
|
||
|
|
if value is None or value == '':
|
||
|
|
if value_type == 'int':
|
||
|
|
return 0
|
||
|
|
elif value_type == 'float':
|
||
|
|
return 0.0
|
||
|
|
elif value_type == 'bool':
|
||
|
|
return False
|
||
|
|
elif value_type == 'json':
|
||
|
|
return {}
|
||
|
|
else:
|
||
|
|
return ''
|
||
|
|
|
||
|
|
try:
|
||
|
|
if value_type == 'int':
|
||
|
|
return int(value)
|
||
|
|
elif value_type == 'float':
|
||
|
|
return float(value)
|
||
|
|
elif value_type == 'bool':
|
||
|
|
return bool(value) or value == '1' or value == 'true' or value == 'True'
|
||
|
|
elif value_type == 'json':
|
||
|
|
import json
|
||
|
|
try:
|
||
|
|
return json.loads(value) if value else {}
|
||
|
|
except (json.JSONDecodeError, TypeError):
|
||
|
|
return {}
|
||
|
|
else:
|
||
|
|
return str(value) if value is not None else ''
|
||
|
|
except (ValueError, TypeError) as e:
|
||
|
|
logger.warning(f"Config value type conversion failed: value={value}, type={value_type}, error={str(e)}")
|
||
|
|
# 转换失败时返回默认值
|
||
|
|
if value_type == 'int':
|
||
|
|
return 0
|
||
|
|
elif value_type == 'float':
|
||
|
|
return 0.0
|
||
|
|
elif value_type == 'bool':
|
||
|
|
return False
|
||
|
|
elif value_type == 'json':
|
||
|
|
return {}
|
||
|
|
else:
|
||
|
|
return str(value) if value is not None else ''
|
||
|
|
|
||
|
|
|
||
|
|
def get_internal_api_key() -> Optional[str]:
|
||
|
|
"""
|
||
|
|
获取内部API密钥（优先从环境变量读取）
|
||
|
|
|
||
|
|
Returns:
|
||
|
|
内部API密钥，如果未配置则返回None
|
||
|
|
"""
|
||
|
|
try:
|
||
|
|
env_val = os.getenv('INTERNAL_API_KEY', '').strip()
|
||
|
|
if env_val:
|
||
|
|
return env_val
|
||
|
|
|
||
|
|
config = load_addon_config()
|
||
|
|
api_key = config.get('internal_api', {}).get('key')
|
||
|
|
|
||
|
|
if api_key:
|
||
|
|
logger.debug(f"Loaded INTERNAL_API_KEY from env-config shape, length: {len(api_key)}")
|
||
|
|
else:
|
||
|
|
logger.warning("Missing INTERNAL_API_KEY (env).")
|
||
|
|
|
||
|
|
return api_key
|
||
|
|
except Exception as e:
|
||
|
|
logger.error(f"Failed to load internal API key: {str(e)}")
|
||
|
|
return None
|
||
|
|
|
||
|
|
|
||
|
|
def clear_config_cache():
|
||
|
|
"""
|
||
|
|
清除配置缓存（配置更新后调用）
|
||
|
|
"""
|
||
|
|
global _config_cache
|
||
|
|
_config_cache = None
|
||
|
|
logger.debug("Addon config cache cleared")
|
||
|
|
|