pip install python-docx

def is_heading_enhanced(paragraph):
    # 標(biāo)題特征：樣式名稱含關(guān)鍵詞 + 字體加粗 + 字體足夠大
    style_name = (paragraph.style.name or "").lower()
    if "標(biāo)題" in style_name or "heading" in style_name:
        return True
    
    try:
        font = paragraph.style.font
        return (font.size.pt >= 14 and font.bold) if font.size else False
    except:
        return False

def get_tables_under_heading(doc_path, target_title):
    from docx import Document
    doc = Document(doc_path)
    target_tables = []
    found = False

    for element in doc.element.body.iter():
        if element.tag.endswith('p'):  # 段落
            par = Paragraph(element, doc)
            if target_title in par.text and is_heading_enhanced(par):
                found = True
            elif found and is_heading_enhanced(par):  # 遇到新標(biāo)題停止
                break
        elif element.tag.endswith('tbl') and found:  # 表格
            table = Table(element, doc)
            data = [[cell.text.strip() for cell in row.cells] for row in table.rows]
            target_tables.append(data)
    
    return target_tables

def to_markdown(table_data):
    if not table_data:
        return ""
    headers = table_data[0]
    md = f"| {' | '.join(headers)} |\n"
    md += f"| {' | '.join(['---']*len(headers))} |\n"
    for row in table_data[1:]:
        md += f"| {' | '.join(row)} |\n"
    return md

def main():
    doc_path = "需求文檔_v2.0.docx"        # 替換為你的文檔路徑
    target_title = "功能權(quán)限矩陣"           # 替換為目標(biāo)標(biāo)題
    
    # 提取表格
    tables = get_tables_under_heading(doc_path, target_title)
    
    # 輸出Markdown
    for idx, table in enumerate(tables, 1):
        print(f"--- 表格{idx} ---\n")
        print(to_markdown(table))
        print("\n-------------------\n")

if __name__ == "__main__":
    main()

# 在 is_heading_enhanced 中，降低字體大小要求
return (font.size.pt >= 12 and font.bold) if font.size else False

# 移除遇到新標(biāo)題停止的邏輯
# 刪除以下代碼：
elif found and is_heading_enhanced(par): break

# 在提取單元格時(shí)去除換行符
cell.text.strip().replace('\n', ' ')

# 提取"接口參數(shù)表"后，直接生成測(cè)試用例模板
table_data = get_tables_under_heading("接口文檔.docx", "接口參數(shù)表")[0]
for row in table_data[1:]:
    param, type_, example = row
    print(f"測(cè)試點(diǎn)：{param}參數(shù)為{example}時(shí)，接口應(yīng)返回{type_}類(lèi)型")

# 將表格數(shù)據(jù)寫(xiě)入CSV文件供測(cè)試框架讀取
import csv

tables = get_tables_under_heading("測(cè)試數(shù)據(jù).docx", "登錄用例數(shù)據(jù)")
with open("test_data.csv", "w", newline="") as f:
    writer = csv.writer(f)
    for table in tables:
        writer.writerows(table)

def get_tables_under_heading(doc_path, target_title, level=1):
    # 新增參數(shù)：level表示標(biāo)題層級(jí)（如1級(jí)標(biāo)題）
    # 在is_heading_enhanced中增加層級(jí)判斷
    if f"heading{level}" in style_name:
        return True

def extract_cell(cell):
    # 處理合并單元格時(shí)的空值
    if cell.text.strip():
        return cell.text.strip()
    else:
        return "[合并單元格]"

import logging
logging.basicConfig(level=logging.INFO)

def main():
    try:
        tables = get_tables_under_heading(...)
    except FileNotFoundError:
        logging.error("文檔路徑錯(cuò)誤，請(qǐng)檢查文件是否存在！")