From d0e5ddb678461eaa144c55d91cbff73b3e95f06e Mon Sep 17 00:00:00 2001 From: fish Date: Sun, 3 May 2026 20:35:18 +0800 Subject: [PATCH] =?UTF-8?q?=E5=B0=86=20TUSHARE=5FTOKEN=20=E5=86=99?= =?UTF-8?q?=E6=AD=BB=E5=88=B0=E4=BB=A3=E7=A0=81=E4=B8=AD=EF=BC=8C=E7=A7=BB?= =?UTF-8?q?=E9=99=A4=20.env=20=E5=BC=95=E7=94=A8?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- docker-compose.yml | 2 +- tushare/main.py | 10 +++------- tushare/src/fetcher.py | 8 +++----- 3 files changed, 7 insertions(+), 13 deletions(-) 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()