123 lines
4.3 KiB
Python
123 lines
4.3 KiB
Python
|
"""
|
|||
|
|
FastAPI 应用配置管理
|
|||
|
|
支持环境变量配置，生产级配置管理
|
|||
|
|
"""
|
|||
|
|
|
|||
|
|
import os
|
|||
|
|
from pathlib import Path
|
|||
|
|
from typing import Optional
|
|||
|
|
import logging
|
|||
|
|
|
|||
|
|
try:
|
|||
|
|
from dotenv import load_dotenv
|
|||
|
|
except Exception: # pragma: no cover
|
|||
|
|
load_dotenv = None
|
|||
|
|
|
|||
|
|
# 项目根目录
|
|||
|
|
BASE_DIR = Path(__file__).resolve().parent.parent.parent
|
|||
|
|
|
|||
|
|
# 加载 .env（不覆盖已存在的系统环境变量）
|
|||
|
|
_dotenv_path = BASE_DIR / ".env"
|
|||
|
|
if load_dotenv is not None and _dotenv_path.exists():
|
|||
|
|
load_dotenv(dotenv_path=_dotenv_path, override=False)
|
|||
|
|
|
|||
|
|
# 环境变量
|
|||
|
|
ENVIRONMENT = os.getenv('ENV', 'development')
|
|||
|
|
DEBUG = os.getenv('DEBUG', 'False').lower() == 'true'
|
|||
|
|
|
|||
|
|
class Settings:
|
|||
|
|
"""应用配置类"""
|
|||
|
|
|
|||
|
|
# FastAPI 基础配置
|
|||
|
|
APP_TITLE = "时间序列数据分析系统"
|
|||
|
|
APP_DESCRIPTION = "支持多格式数据上传、AI增强分析、多语言报告生成"
|
|||
|
|
APP_VERSION = "2.0.0"
|
|||
|
|
|
|||
|
|
# API 暴露模式
|
|||
|
|
# - full: 暴露 v1 + v2（默认）
|
|||
|
|
# - v2: 仅暴露 v2 分析接口 + 基础状态接口（禁用 v1 上传/文件/图片接口）
|
|||
|
|
API_MODE = os.getenv('API_MODE', 'full').strip().lower()
|
|||
|
|
|
|||
|
|
# 服务器配置
|
|||
|
|
HOST = os.getenv('HOST', '0.0.0.0')
|
|||
|
|
PORT = int(os.getenv('PORT', 60201))
|
|||
|
|
RELOAD = DEBUG
|
|||
|
|
|
|||
|
|
# CORS 配置
|
|||
|
|
CORS_ORIGINS = os.getenv('CORS_ORIGINS', '*').split(',')
|
|||
|
|
CORS_ALLOW_CREDENTIALS = True
|
|||
|
|
CORS_ALLOW_METHODS = ['*']
|
|||
|
|
CORS_ALLOW_HEADERS = ['*']
|
|||
|
|
|
|||
|
|
# 文件上传配置
|
|||
|
|
UPLOAD_DIR = Path(os.getenv('UPLOAD_DIR', BASE_DIR / 'uploads'))
|
|||
|
|
UPLOAD_DIR.mkdir(exist_ok=True)
|
|||
|
|
|
|||
|
|
MAX_UPLOAD_SIZE = int(os.getenv('MAX_UPLOAD_SIZE', 16 * 1024 * 1024)) # 16MB
|
|||
|
|
ALLOWED_EXTENSIONS = {'csv'}
|
|||
|
|
|
|||
|
|
# 临时文件配置
|
|||
|
|
TEMP_DIR = Path(os.getenv('TEMP_DIR', BASE_DIR / 'temp'))
|
|||
|
|
TEMP_DIR.mkdir(exist_ok=True)
|
|||
|
|
|
|||
|
|
# 字体配置
|
|||
|
|
FONTS_DIR = Path(os.getenv('FONTS_DIR', BASE_DIR / 'resource' / 'fonts'))
|
|||
|
|
FONTS_DIR.mkdir(parents=True, exist_ok=True)
|
|||
|
|
|
|||
|
|
# API 配置 (阿里云千问)
|
|||
|
|
API_KEY = os.getenv('MY_API_KEY', '')
|
|||
|
|
API_BASE = os.getenv('MY_API_BASE', 'https://dashscope.aliyuncs.com/compatible-mode/v1')
|
|||
|
|
API_MODEL = os.getenv('MY_MODEL', 'qwen-turbo')
|
|||
|
|
API_TIMEOUT = int(os.getenv('API_TIMEOUT', 30))
|
|||
|
|
|
|||
|
|
# 分析配置
|
|||
|
|
LANGUAGE_DEFAULT = os.getenv('LANGUAGE_DEFAULT', 'zh')
|
|||
|
|
ANALYSIS_TIMEOUT = int(os.getenv('ANALYSIS_TIMEOUT', 300)) # 5分钟
|
|||
|
|
|
|||
|
|
# 日志配置
|
|||
|
|
LOG_LEVEL = os.getenv('LOG_LEVEL', 'INFO' if not DEBUG else 'DEBUG')
|
|||
|
|
LOG_DIR = Path(os.getenv('LOG_DIR', BASE_DIR / 'logs'))
|
|||
|
|
LOG_DIR.mkdir(exist_ok=True)
|
|||
|
|
|
|||
|
|
# 内存管理
|
|||
|
|
MAX_MEMORY_MB = int(os.getenv('MAX_MEMORY_MB', 500))
|
|||
|
|
|
|||
|
|
# v2 (OSS URL) 配置
|
|||
|
|
# 允许的域名白名单（逗号分隔）。为空时表示不启用域名白名单（仍会做私网/环回 IP 拦截）。
|
|||
|
|
V2_ALLOWED_HOSTS = [h.strip() for h in os.getenv('V2_ALLOWED_HOSTS', '').split(',') if h.strip()]
|
|||
|
|
# 是否允许 http（默认仅 https）
|
|||
|
|
V2_ALLOW_HTTP = os.getenv('V2_ALLOW_HTTP', 'False').lower() == 'true'
|
|||
|
|
# 是否允许私网/环回地址（仅用于本地开发/冒烟；生产建议保持 False）
|
|||
|
|
V2_ALLOW_PRIVATE_NETWORKS = os.getenv('V2_ALLOW_PRIVATE_NETWORKS', 'False').lower() == 'true'
|
|||
|
|
# 下载超时（秒）。requests 支持 (connect, read)，这里统一使用 read 超时。
|
|||
|
|
V2_DOWNLOAD_TIMEOUT_SECONDS = float(os.getenv('V2_DOWNLOAD_TIMEOUT_SECONDS', 30))
|
|||
|
|
V2_CONNECT_TIMEOUT_SECONDS = float(os.getenv('V2_CONNECT_TIMEOUT_SECONDS', 5))
|
|||
|
|
|
|||
|
|
@classmethod
|
|||
|
|
def get_upload_path(cls, filename: str) -> Path:
|
|||
|
|
"""获取上传文件的完整路径"""
|
|||
|
|
return cls.UPLOAD_DIR / filename
|
|||
|
|
|
|||
|
|
@classmethod
|
|||
|
|
def get_temp_path(cls, filename: str) -> Path:
|
|||
|
|
"""获取临时文件的完整路径"""
|
|||
|
|
return cls.TEMP_DIR / filename
|
|||
|
|
|
|||
|
|
# 日志配置
|
|||
|
|
def setup_logging():
|
|||
|
|
"""设置日志系统"""
|
|||
|
|
logging.basicConfig(
|
|||
|
|
level=Settings.LOG_LEVEL,
|
|||
|
|
format='%(asctime)s - %(name)s - %(levelname)s - %(message)s',
|
|||
|
|
handlers=[
|
|||
|
|
logging.FileHandler(Settings.LOG_DIR / 'app.log'),
|
|||
|
|
logging.StreamHandler()
|
|||
|
|
]
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
# 创建全局配置实例
|
|||
|
|
settings = Settings()
|
|||
|
|
|
|||
|
|
# 启用日志
|
|||
|
|
setup_logging()
|