from spire.xls import *

def convert_multi_tables_to_excel(md_file_path, excel_file_path):
    # 初始化一個(gè)新的工作簿實(shí)例
    workbook = Workbook()
    
    # 清除由構(gòu)造函數(shù)創(chuàng)建的默認(rèn)空白工作表
    # 這確保我們的 Excel 文件只包含我們顯式創(chuàng)建的工作表
    workbook.Worksheets.Clear()
    
    try:
        # 逐行讀取 Markdown 文件內(nèi)容
        with open(md_file_path, 'r', encoding='utf-8') as file:
            lines = file.readlines()
    except FileNotFoundError:
        print(f"錯(cuò)誤：未找到文件 {md_file_path}。")
        return

    # --- 狀態(tài)變量 ---
    current_table_data = []  # 一個(gè)緩沖區(qū)，用于存儲(chǔ)當(dāng)前正在解析的表格的行數(shù)據(jù)
    sheet_count = 0          # 用于命名工作表的計(jì)數(shù)器（表 1、表 2 等）
    in_table = False         # 布爾標(biāo)志，用于跟蹤當(dāng)前是否處于表格塊內(nèi)部

    # --- 輔助函數(shù) ---
    def save_current_table():
        """
        獲取 current_table_data 中的數(shù)據(jù)，創(chuàng)建一個(gè)新的 Excel 工作表，
        填充數(shù)據(jù)，然后重置緩沖區(qū)。
        """
        nonlocal sheet_count, current_table_data, in_table
        
        if current_table_data:
            sheet_count += 1
            # 創(chuàng)建一個(gè)帶有描述性名稱的新工作表
            sheet_name = f"表 {sheet_count}"
            sheet = workbook.Worksheets.Add(sheet_name)
                        
            # 遍歷存儲(chǔ)的行并寫入 Excel
            for r_idx, row_data in enumerate(current_table_data):
                for c_idx, cell_val in enumerate(row_data):
                    # 訪問指定的單元格（行，列）
                    cell = sheet.Range[r_idx + 1, c_idx + 1]
                    cell.Text = cell_val
                    
                    # 對(duì)標(biāo)題行（索引為 0）應(yīng)用加粗格式
                    if r_idx == 0:
                        cell.Style.Font.IsBold = True
            
            # 自動(dòng)調(diào)整列寬以確保內(nèi)容可見
            sheet.AllocatedRange.AutoFitColumns()
            
            # 為下一個(gè)表格重置狀態(tài)
            current_table_data = []
            in_table = False

    # --- 解析循環(huán) ---
    for line in lines:
        line = line.strip()
        
        # 檢查該行是否代表一個(gè) Markdown 表格行（以 '|' 開頭和結(jié)尾）
        if line.startswith('|') and line.endswith('|'):
            in_table = True
            
            # 跳過分隔行（例如 |---|---|），它只包含管道符、破折號(hào)或冒號(hào)
            if all(c in '|- :' for c in line):
                continue
            
            # 解析單元格：移除外部管道符并按內(nèi)部管道符分割
            # 我們還去除了每個(gè)單元格內(nèi)容中的空白字符
            cells = [c.strip() for c in line[1:-1].split('|')]
            
            # 將解析后的行添加到我們的臨時(shí)緩沖區(qū)
            current_table_data.append(cells)
            
        else:
            # 如果該行不是表格行，檢查我們之前是否在表格中
            if in_table:
                # 在表格之后遇到了非表格行，因此表格已完成。
                # 觸發(fā)保存函數(shù)。
                save_current_table()

    # 邊界情況：如果文件結(jié)束時(shí)我們?nèi)栽诒砀駜?nèi)部，
    # 循環(huán)將在不觸發(fā)上面 'else' 塊的情況下結(jié)束。
    # 我們必須確保保存最后一個(gè)表格。
    if in_table:
        save_current_table()

    # 將填充好的工作簿保存到文件
    workbook.SaveToFile(excel_file_path, ExcelVersion.Version2016)
    workbook.Dispose()
    print(f"成功將 {sheet_count} 個(gè)表格轉(zhuǎn)換到 {excel_file_path}")

# 使用示例
convert_multi_tables_to_excel("example.md", "multi_sheet_output.xlsx")