fix: 优化Token配置方式,支持环境变量和配置文件

This commit is contained in:
2026-04-08 19:00:40 +08:00
parent f40d3aed81
commit 2857d8febb
2 changed files with 41 additions and 5 deletions
+20 -3
View File
@@ -15,15 +15,32 @@ pip install -r requirements.txt
## Tushare Token ## Tushare Token
**必须先设置Token才能运行!**
需要在 tushare.pro 注册并获取token。 需要在 tushare.pro 注册并获取token。
设置方式: 设置方式(三选一)
1. 注册账号:https://tushare.pro/register
2. 获取token后在代码中设置,或设置环境变量: ### 方式1:环境变量(推荐)
```bash ```bash
export TUSHARE_TOKEN=your_token_here export TUSHARE_TOKEN=your_token_here
python fetch_history.py
``` ```
### 方式2:配置文件
```bash
echo 'your_token_here' > config.txt
python fetch_history.py
```
### 方式3:代码中直接设置
```python
# 编辑 fetch_history.py,在 main() 函数中修改
pro = setup_tushare(token='your_token_here')
```
**注册地址**https://tushare.pro/register
## 运行 ## 运行
```bash ```bash
+21 -2
View File
@@ -33,8 +33,27 @@ REQUEST_INTERVAL = 0.3
def setup_tushare(token=None): def setup_tushare(token=None):
"""初始化tushare""" """初始化tushare"""
if token: import os
ts.set_token(token)
# 优先级:参数 > 环境变量 > 配置文件
if not token:
token = os.environ.get('TUSHARE_TOKEN', '')
if not token:
# 尝试从配置文件读取
config_file = BASE_DIR / 'config.txt'
if config_file.exists():
token = config_file.read_text().strip()
if not token:
print("错误:未设置 Tushare Token")
print("请通过以下方式之一设置:")
print(" 1. 设置环境变量: export TUSHARE_TOKEN=your_token")
print(" 2. 创建配置文件: echo 'your_token' > config.txt")
print(" 3. 注册地址: https://tushare.pro/register")
raise ValueError("缺少 Tushare Token")
ts.set_token(token)
return ts.pro_api() return ts.pro_api()