"""
量化数据库配置文件

敏感信息（数据库密码、API Token）通过环境变量管理：
  - QUANT_DB_PASSWORD : 数据库密码
  - TUSHARE_TOKEN     : Tushare API Token

设置方式：
  Windows:  set QUANT_DB_PASSWORD=your_password && set TUSHARE_TOKEN=your_token
  Linux:    export QUANT_DB_PASSWORD=your_password && export TUSHARE_TOKEN=your_token
  也可创建 .env 文件（参考 .env.example）
"""
import os

# PostgreSQL Docker 连接配置
DB_CONFIG = {
    "host": os.environ.get("QUANT_DB_HOST", "192.168.27.11"),
    "port": int(os.environ.get("QUANT_DB_PORT", "5438")),
    "database": os.environ.get("QUANT_DB_NAME", "quant_db"),
    "user": os.environ.get("QUANT_DB_USER", "postgres"),
    "password": os.environ.get("QUANT_DB_PASSWORD", ""),
}

# Tushare API Token
TUSHARE_TOKEN = os.environ.get("TUSHARE_TOKEN", "")

# 批量导入参数
BATCH_SIZE = int(os.environ.get("QUANT_BATCH_SIZE", "5000"))  # 每批次插入行数
START_DATE = os.environ.get("QUANT_START_DATE", "2010-01-01")  # 数据起始日期
END_DATE = os.environ.get("QUANT_END_DATE", "2025-12-31")  # 数据结束日期