diff --git a/README.md b/README.md index c73ef33..92fdc05 100644 --- a/README.md +++ b/README.md @@ -15,15 +15,32 @@ pip install -r requirements.txt ## Tushare Token +**必须先设置Token才能运行!** + 需要在 tushare.pro 注册并获取token。 -设置方式: -1. 注册账号:https://tushare.pro/register -2. 获取token后在代码中设置,或设置环境变量: +设置方式(三选一): + +### 方式1:环境变量(推荐) ```bash 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 diff --git a/fetch_history.py b/fetch_history.py index 81608f0..879e66c 100644 --- a/fetch_history.py +++ b/fetch_history.py @@ -33,8 +33,27 @@ REQUEST_INTERVAL = 0.3 def setup_tushare(token=None): """初始化tushare""" - if token: - ts.set_token(token) + import os + + # 优先级:参数 > 环境变量 > 配置文件 + 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()