fix:数据库接入方式改为在系统环境变量中储存敏感信息。

2026-07-31 21:30:21 +08:00
parent 6deda5155a
commit b868e63e40
4 changed files with 61 additions and 14 deletions
@@ -1,19 +1,30 @@
 """
 量化数据库配置文件
+
+敏感信息（数据库密码、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": "192.168.27.15",
-    "port": 12345,
-    "database": "quant_db",
-    "user": "postgres",
-    "password": "postgres",  # 请修改为实际密码
+    "host": os.environ.get("QUANT_DB_HOST", "192.168.27.15"),
+    "port": int(os.environ.get("QUANT_DB_PORT", "12345")),
+    "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 = "你的token"  # 请替换为你的tushare token
+TUSHARE_TOKEN = os.environ.get("TUSHARE_TOKEN", "")

 # 批量导入参数
-BATCH_SIZE = 5000  # 每批次插入行数
-START_DATE = "2010-01-01"  # 数据起始日期
-END_DATE = "2025-12-31"  # 数据结束日期
+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")  # 数据结束日期