新增一键运行功能

basic.py

@@ -1,9 +1,10 @@
 import tushare as ts
-from sqlalchemy import create_engine
+
+from stock_config import TUSHARE_TOKEN, get_engine
 
 # 1. 配置参数
 # 在 Tushare 官网个人主页获取你的 token
-TOKEN = '0fad3cf498757089e2630028455d5bbe1637475788bcdaa1f2175e93'
+TOKEN = TUSHARE_TOKEN
 # 数据库连接配置 (格式: mysql+pymysql://用户名:密码@主机地址:端口/数据库名)
 # DB_CONN_STR = 'mysql+pymysql://yangfan:aA%8023321088@8.159.129.156:10836/level?charset=utf8'
 
@@ -25,18 +26,12 @@ def sync_stock_basic():
 
     if df.empty:
         print("未获取到数据，请检查接口权限或Token设置。")
-        return
+        return False
 
     # 4. 写入数据库
     try:
         # 创建数据库引擎
-        engine = create_engine(
-        "mysql+pymysql://yangfan@8.159.129.156:10836/level?charset=utf8mb4",
-        connect_args={
-            "password": "aA%8023321088"
-        },
-        pool_pre_ping=True
-    )
+        engine = get_engine()
 
         # 将数据写入数据库表 'stock_basic'
         # if_exists='replace': 如果表存在则替换（适合全量更新基础信息）
@@ -44,10 +39,12 @@ def sync_stock_basic():
         df.to_sql('stock_basic', engine, if_exists='replace', index=False)
 
         print(f"成功将 {len(df)} 条股票基础信息写入数据库表 [stock_basic]。")
+        return True
 
     except Exception as e:
         print(f"写入数据库失败: {e}")
+        return False
 
 
 if __name__ == "__main__":
-    sync_stock_basic()
+    sync_stock_basic()