fix: 优化Token配置方式,支持环境变量和配置文件
This commit is contained in:
23
README.md
23
README.md
@@ -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
|
||||||
|
|||||||
@@ -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()
|
||||||
|
|
||||||
|
|
||||||
|
|||||||
Reference in New Issue
Block a user