From 4ea740083aa52044b0500a5114954a2330a951c8 Mon Sep 17 00:00:00 2001 From: claude_dev Date: Sat, 11 Jul 2026 14:59:34 +0800 Subject: [PATCH] =?UTF-8?q?fix(data):=20index=5Fdownloader=20baostock=20er?= =?UTF-8?q?ror=5Fcode=E5=88=A4=E6=96=AD=E4=BF=AE(=E7=9C=9F=E5=AE=9E?= =?UTF-8?q?=E4=B8=BA'0'=E9=9D=9E'success',=E5=AE=B9=E4=B8=A4=E8=80=85)?= =?UTF-8?q?=E2=80=94=E2=80=94mock=E4=B8=8E=E7=9C=9F=E5=AE=9Ebaostock?= =?UTF-8?q?=E5=88=86=E6=AD=A7?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- sanguo_data/index_downloader.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/sanguo_data/index_downloader.py b/sanguo_data/index_downloader.py index 691b13c..d4b24b2 100644 --- a/sanguo_data/index_downloader.py +++ b/sanguo_data/index_downloader.py @@ -24,7 +24,7 @@ def download_index(symbol: str, start_year: int, end_year: int, out_dir: str) -> # 登录 baostock lg = bs.login() - if lg.error_code != "success": + if lg.error_code not in ("0", "success"): raise RuntimeError(f"baostock 登录失败: {lg.error_msg}") try: @@ -44,13 +44,13 @@ def download_index(symbol: str, start_year: int, end_year: int, out_dir: str) -> adjustfield="3" # 不复权 ) - if rs.error_code != "success": + if rs.error_code not in ("0", "success"): print(f"Warning: 下载 {symbol} {year} 失败: {rs.error_msg}") continue # 转换为 DataFrame 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()) if not data_list: