"fix:robust-column-detection"

sync_data.py

@@ -115,44 +115,128 @@ def get_stock_list() -> pd.DataFrame:
 
     # ETF (增加容错)
     try:
-        df_etf = ak.fund_etf_spot_em()[['代码', '名称']]
-        df_etf.columns = ['code', 'name']
-        df_etf['market'] = 'ETF'
-        all_lists.append(df_etf)
-        logger.info(f"ETF list fetched: {len(df_etf)} funds")
+        df_etf = ak.fund_etf_spot_em()
+        # 检查实际列名
+        code_cols = ['代码', 'code', 'fund_code', 'ETF代码']
+        name_cols = ['名称', 'name', 'fund_name', 'ETF名称']
+        
+        c_code = None
+        c_name = None
+        
+        for col in code_cols:
+            if col in df_etf.columns:
+                c_code = col
+                break
+        
+        for col in name_cols:
+            if col in df_etf.columns:
+                c_name = col
+                break
+        
+        if c_code and c_name:
+            df_etf = df_etf[[c_code, c_name]]
+            df_etf.columns = ['code', 'name']
+            df_etf['market'] = 'ETF'
+            all_lists.append(df_etf)
+            logger.info(f"ETF list fetched: {len(df_etf)} funds")
+        else:
+            logger.warning(f"Could not find code/name columns in ETF data. Available columns: {df_etf.columns.tolist()}")
     except Exception as e:
         logger.warning(f"ETF list fetch failed: {e}")
 
     # LOF
     try:
-        df_lof = ak.fund_lof_spot_em()[['代码', '名称']]
-        df_lof.columns = ['code', 'name']
-        df_lof['market'] = 'LOF'
-        all_lists.append(df_lof)
-        logger.info(f"LOF list fetched: {len(df_lof)} funds")
+        df_lof = ak.fund_lof_spot_em()
+        # 检查实际列名
+        code_cols = ['代码', 'code', 'fund_code', 'LOF代码']
+        name_cols = ['名称', 'name', 'fund_name', 'LOF名称']
+        
+        c_code = None
+        c_name = None
+        
+        for col in code_cols:
+            if col in df_lof.columns:
+                c_code = col
+                break
+        
+        for col in name_cols:
+            if col in df_lof.columns:
+                c_name = col
+                break
+        
+        if c_code and c_name:
+            df_lof = df_lof[[c_code, c_name]]
+            df_lof.columns = ['code', 'name']
+            df_lof['market'] = 'LOF'
+            all_lists.append(df_lof)
+            logger.info(f"LOF list fetched: {len(df_lof)} funds")
+        else:
+            logger.warning(f"Could not find code/name columns in LOF data. Available columns: {df_lof.columns.tolist()}")
     except Exception as e:
         logger.warning(f"LOF list fetch failed: {e}")
 
     # REITs
     try:
-        df_reits = ak.reits_realtime_em()[['代码', '名称']]
-        df_reits.columns = ['code', 'name']
-        df_reits['market'] = 'REITs'
-        all_lists.append(df_reits)
-        logger.info(f"REITs list fetched: {len(df_reits)} products")
+        df_reits = ak.reits_realtime_em()
+        # 检查实际列名
+        code_cols = ['代码', 'code', 'REITs代码']
+        name_cols = ['名称', 'name', 'REITs名称']
+        
+        c_code = None
+        c_name = None
+        
+        for col in code_cols:
+            if col in df_reits.columns:
+                c_code = col
+                break
+        
+        for col in name_cols:
+            if col in df_reits.columns:
+                c_name = col
+                break
+        
+        if c_code and c_name:
+            df_reits = df_reits[[c_code, c_name]]
+            df_reits.columns = ['code', 'name']
+            df_reits['market'] = 'REITs'
+            all_lists.append(df_reits)
+            logger.info(f"REITs list fetched: {len(df_reits)} products")
+        else:
+            logger.warning(f"Could not find code/name columns in REITs data. Available columns: {df_reits.columns.tolist()}")
     except Exception as e:
         logger.warning(f"REITs list fetch failed: {e}")
     
     # 可转债
     try:
         df_cb = ak.bond_zh_hs_cov_spot()
-        c_code = '代码' if '代码' in df_cb.columns else 'symbol'
-        c_name = '名称' if '名称' in df_cb.columns else 'name'
-        df_cb = df_cb[[c_code, c_name]]
-        df_cb.columns = ['code', 'name']
-        df_cb['market'] = '可转债'
-        all_lists.append(df_cb)
-        logger.info(f"Convertible bond list fetched: {len(df_cb)} bonds")
+        # 检查实际列名
+        logger.info(f"Convertible bond columns: {df_cb.columns.tolist()}")
+        
+        # 尝试找到正确的代码列名
+        code_cols = ['代码', 'symbol', 'bond_code', '代码', '转债代码', '代码']
+        name_cols = ['名称', 'name', 'bond_name', '转债名称', '名称']
+        
+        c_code = None
+        c_name = None
+        
+        for col in code_cols:
+            if col in df_cb.columns:
+                c_code = col
+                break
+        
+        for col in name_cols:
+            if col in df_cb.columns:
+                c_name = col
+                break
+        
+        if c_code and c_name:
+            df_cb = df_cb[[c_code, c_name]]
+            df_cb.columns = ['code', 'name']
+            df_cb['market'] = '可转债'
+            all_lists.append(df_cb)
+            logger.info(f"Convertible bond list fetched: {len(df_cb)} bonds")
+        else:
+            logger.warning(f"Could not find code/name columns in convertible bond data. Available columns: {df_cb.columns.tolist()}")
     except Exception as e:
         logger.warning(f"Convertible bond list fetch failed: {e}")