diff --git a/management/server/services/knowledgebases/excel_parser.py b/management/server/services/knowledgebases/excel_parser.py
index cbc8168..500f718 100644
--- a/management/server/services/knowledgebases/excel_parser.py
+++ b/management/server/services/knowledgebases/excel_parser.py
@@ -1,32 +1,21 @@
import pandas as pd
-def parse_excel(file_path):
- # 读取所有工作表
- all_sheets = pd.read_excel(file_path, sheet_name=None) # 读取所有sheet
+def parse_excel(file_path):
+ # 读取所有工作表
+ all_sheets = pd.read_excel(file_path, sheet_name=None) # 读取所有sheet
+
+ blocks = []
+
+ for sheet_name, df in all_sheets.items():
+ df = df.fillna(method="ffill") # 填充合并的单元格
+ headers = df.columns.tolist()
+
+ for _, row in df.iterrows():
+ html_table = "".format("".join(f"| {col} | " for col in headers), "".join(f"{row[col]} | " for col in headers))
+ block = {"type": "table", "img_path": "", "table_caption": [f"Sheet: {sheet_name}"], "table_footnote": [], "table_body": f"{html_table}", "page_idx": 0}
+ blocks.append(block)
- blocks = []
-
- for sheet_name, df in all_sheets.items():
- df = df.fillna(method='ffill')#填充合并的单元格
- headers = df.columns.tolist()
-
- for _, row in df.iterrows():
- html_table = "".format(
- "".join(f"{col} | " for col in headers),
- "".join(f"{row[col]} | " for col in headers)
- )
- print(row['测试分类'])
- block = {
- "type": "table",
- "img_path": "",
- "table_caption": [f"Sheet: {sheet_name}"],
- "table_footnote": [],
- "table_body": f"{html_table}",
- "page_idx": 0
- }
- blocks.append(block)
-
return blocks