liaozhaorun
9e7d4241c6
- 新增 moneyflow 资金流向数据同步模块 - 实现接口级速率限制配置（sync_config.py） - 更新流动性相关因子定义 - 添加非对称量化损失函数
83 lines
2.2 KiB
Python
83 lines
2.2 KiB
Python
"""配置管理模块
|
||
|
||
从环境变量加载应用配置，使用pydantic-settings进行类型验证
|
||
"""
|
||
|
||
import os
|
||
from pathlib import Path
|
||
from pydantic_settings import BaseSettings
|
||
from functools import lru_cache
|
||
from typing import Optional
|
||
|
||
|
||
# 获取项目根目录（config文件夹所在目录）
|
||
PROJECT_ROOT = Path(__file__).parent.parent.parent
|
||
CONFIG_DIR = PROJECT_ROOT / "config"
|
||
|
||
|
||
class Settings(BaseSettings):
|
||
"""应用配置类，从环境变量加载
|
||
|
||
所有配置项都会自动从环境变量读取（小写转大写）
|
||
例如：tushare_token 会读取 TUSHARE_TOKEN 环境变量
|
||
"""
|
||
|
||
# Tushare API 配置
|
||
tushare_token: str = ""
|
||
|
||
# 数据存储配置
|
||
root_path: str = "" # 项目根路径，默认自动检测
|
||
data_path: str = "data" # 数据存储路径，相对于 root_path
|
||
|
||
# 数据库配置（可选，用于未来扩展）
|
||
database_host: str = "localhost"
|
||
database_port: int = 5432
|
||
database_name: str = "prostock"
|
||
database_user: Optional[str] = None
|
||
database_password: Optional[str] = None
|
||
|
||
# Redis配置（可选，用于未来扩展）
|
||
redis_host: str = "localhost"
|
||
redis_port: int = 6379
|
||
redis_password: Optional[str] = None
|
||
|
||
# 应用配置
|
||
app_env: str = "development"
|
||
app_debug: bool = False
|
||
app_port: int = 8000
|
||
|
||
@property
|
||
def project_root(self) -> Path:
|
||
"""获取项目根路径。"""
|
||
if self.root_path:
|
||
return Path(self.root_path)
|
||
return PROJECT_ROOT
|
||
|
||
@property
|
||
def data_path_resolved(self) -> Path:
|
||
"""获取解析后的数据路径（绝对路径）。"""
|
||
path = Path(self.data_path)
|
||
if path.is_absolute():
|
||
return path
|
||
return self.project_root / path
|
||
|
||
class Config:
|
||
# 从 config/ 目录读取 .env.local 文件
|
||
env_file = str(CONFIG_DIR / ".env.local")
|
||
env_file_encoding = "utf-8"
|
||
case_sensitive = False
|
||
extra = "ignore" # 忽略 .env.local 中的额外变量
|
||
|
||
|
||
@lru_cache()
|
||
def get_settings() -> Settings:
|
||
"""获取配置单例
|
||
|
||
使用lru_cache确保配置只加载一次
|
||
"""
|
||
return Settings()
|
||
|
||
|
||
# 导出配置实例供全局使用
|
||
settings = get_settings()
|