sanguo/sanguo_vnpy_v2
3
0
Fork 0
Code Issues 3 Pull Requests Actions Packages Projects Releases Wiki Activity

fix(data): index_downloader baostock error_code判断修(真实为'0'非'success',容两者)——mock与真实baostock分歧

Browse Source
This commit is contained in:
Claude Dev
2026-07-11 14:59:34 +08:00
parent 21dd30969f
commit 4ea740083a
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
sanguo_data/index_downloader.py
+3 -3
View File
@@ -24,7 +24,7 @@ def download_index(symbol: str, start_year: int, end_year: int, out_dir: str) ->
# 登录 baostock # 登录 baostock
lg = bs.login() lg = bs.login()
if lg.error_code != "success": if lg.error_code not in ("0", "success"):
raise RuntimeError(f"baostock 登录失败: {lg.error_msg}") raise RuntimeError(f"baostock 登录失败: {lg.error_msg}")
try: try:
@@ -44,13 +44,13 @@ def download_index(symbol: str, start_year: int, end_year: int, out_dir: str) ->
adjustfield="3" # 不复权 adjustfield="3" # 不复权
) )
if rs.error_code != "success": if rs.error_code not in ("0", "success"):
print(f"Warning: 下载 {symbol} {year} 失败: {rs.error_msg}") print(f"Warning: 下载 {symbol} {year} 失败: {rs.error_msg}")
continue continue
# 转换为 DataFrame # 转换为 DataFrame
data_list = [] data_list = []
while (rs.error_code == "success") and rs.next(): while (rs.error_code in ("0", "success")) and rs.next():
data_list.append(rs.get_row_data()) data_list.append(rs.get_row_data())
if not data_list: if not data_list: