tushare

main_force_selector.py

@@ -11,6 +11,7 @@ import pywencai
 from datetime import datetime, timedelta
 from typing import Dict, List, Tuple
 import time
+from http_timeout import call_with_timeout
 
 class MainForceStockSelector:
     """主力选股类"""
@@ -71,7 +72,7 @@ class MainForceStockSelector:
                 print(f"查询语句: {query[:100]}...")
                 
                 try:
-                    result = pywencai.get(query=query, loop=True)
+                    result = call_with_timeout(pywencai.get, timeout=30, query=query, loop=True)
                     
                     if result is None:
                         print(f"  ⚠️ 方案{i}返回None，尝试下一个方案")
@@ -387,4 +388,3 @@ class MainForceStockSelector:
 
 # 全局实例
 main_force_selector = MainForceStockSelector()
-