diff --git a/docker-compose.yml b/docker-compose.yml index 6dc925e..560c770 100644 --- a/docker-compose.yml +++ b/docker-compose.yml @@ -17,7 +17,7 @@ services: tushare: build: ./tushare container_name: trade-tushare - env_file: ./tushare/.env + # token 已写死在代码中,无需 env_file environment: - DATABASE_URL=postgresql://trade:trade@postgres:5432/futures depends_on: diff --git a/tushare/main.py b/tushare/main.py index dec1fd3..2e7360f 100644 --- a/tushare/main.py +++ b/tushare/main.py @@ -1,16 +1,12 @@ -import os import sys import tushare as ts +TUSHARE_TOKEN = "76efd8465f9f2591aa42a385268e06acf6b80b7a15be2267ad2281b7" + def main() -> int: - token = os.environ.get("TUSHARE_TOKEN") - if not token: - print("[ERROR] 未设置 TUSHARE_TOKEN 环境变量", file=sys.stderr) - return 1 - - ts.set_token(token) + ts.set_token(TUSHARE_TOKEN) pro = ts.pro_api() df = pro.trade_cal(exchange="SHFE", start_date="20260101", end_date="20260110") diff --git a/tushare/src/fetcher.py b/tushare/src/fetcher.py index b1bc7d9..9122bd0 100644 --- a/tushare/src/fetcher.py +++ b/tushare/src/fetcher.py @@ -1,15 +1,13 @@ -import os from typing import Optional import pandas as pd import tushare as ts +TUSHARE_TOKEN = "76efd8465f9f2591aa42a385268e06acf6b80b7a15be2267ad2281b7" + def _init_api(): - token = os.environ.get("TUSHARE_TOKEN") - if not token: - raise RuntimeError("TUSHARE_TOKEN 环境变量未设置") - ts.set_token(token) + ts.set_token(TUSHARE_TOKEN) return ts.pro_api()