liaozhaorun
e625a53162
- 添加项目规则文档（开发规范、安全规则、配置管理） - 实现数据模块核心功能（API 客户端、限流器、存储管理、配置加载） - 添加 .gitignore 和 .kilocodeignore 配置 - 配置环境变量模板 - 编写 daily 模块单元测试
34 lines
702 B
Python
34 lines
702 B
Python
"""Configuration management for data collection module."""
|
|
from pathlib import Path
|
|
from pydantic_settings import BaseSettings
|
|
|
|
|
|
class Config(BaseSettings):
|
|
"""Application configuration loaded from environment variables."""
|
|
|
|
# Tushare API token
|
|
tushare_token: str = ""
|
|
|
|
# Data storage path
|
|
data_path: Path = Path("./data")
|
|
|
|
# Rate limit: requests per minute
|
|
rate_limit: int = 100
|
|
|
|
# Thread pool size
|
|
threads: int = 2
|
|
|
|
class Config:
|
|
env_file = ".env.local"
|
|
env_file_encoding = "utf-8"
|
|
case_sensitive = False
|
|
|
|
|
|
# Global config instance
|
|
config = Config()
|
|
|
|
|
|
def get_config() -> Config:
|
|
"""Get configuration instance."""
|
|
return config
|