from spire.doc import *
from spire.doc.common import *

def insert_document_at_paragraph(target_doc_path: str, source_doc_path: str, insert_after_paragraph_index: int, output_path: str):
    """
    在一個(gè)目標(biāo)文檔的指定段落之后插入源文檔的所有內(nèi)容。

    Args:
        target_doc_path (str): 目標(biāo)文檔的路徑。
        source_doc_path (str): 源文檔的路徑。
        insert_after_paragraph_index (int): 插入點(diǎn)：目標(biāo)文檔中某個(gè)節(jié)的段落索引，內(nèi)容將在此段落之后插入。
        output_path (str): 合并后文檔的保存路徑。
    """
    target_document = Document()
    target_document.LoadFromFile(target_doc_path, FileFormat.Docx)

    source_document = Document()
    source_document.LoadFromFile(source_doc_path, FileFormat.Docx)

    print(f"開(kāi)始在 '{target_doc_path}' 的第 {insert_after_paragraph_index} 段之后插入 '{source_doc_path}' 的內(nèi)容。")

    # 假設(shè)只處理目標(biāo)文檔的第一個(gè)節(jié)
    target_section = target_document.Sections[0] 
    
    # 找到目標(biāo)插入點(diǎn)所在的段落
    if insert_after_paragraph_index < 0 or insert_after_paragraph_index >= target_section.Paragraphs.Count:
        print(f"  - 警告: 指定的段落索引 {insert_after_paragraph_index} 超出范圍，將追加到文檔末尾。")
        insert_after_paragraph_index = target_section.Paragraphs.Count - 1 # 調(diào)整為最后一個(gè)段落

    # 獲取插入點(diǎn)段落
    insert_paragraph = target_section.Paragraphs.get_Item(insert_after_paragraph_index)
    
    # 獲取插入點(diǎn)段落的父級(jí)Body，以及該段落在Body中的索引
    body = insert_paragraph.OwnerTextBody
    insert_index_in_body = body.ChildObjects.IndexOf(insert_paragraph)

    # 遍歷源文檔的所有節(jié)和其內(nèi)容，并插入到目標(biāo)文檔的指定位置
    for source_section_index in range(source_document.Sections.Count):
        source_sec = source_document.Sections.get_Item(source_section_index)
        for child_obj_index in range(source_sec.Body.ChildObjects.Count):
            # 克隆源文檔的每個(gè)子對(duì)象（段落、表格等）
            cloned_obj = source_sec.Body.ChildObjects.get_Item(child_obj_index).Clone()
            # 插入到目標(biāo)文檔的Body中
            body.ChildObjects.Insert(insert_index_in_body + 1, cloned_obj)
            insert_index_in_body += 1 # 每次插入后更新插入索引

    target_document.SaveToFile(output_path, FileFormat.Docx)
    target_document.Close()
    source_document.Close()
    print(f"內(nèi)容插入完成，結(jié)果保存到: {output_path}")

# 示例用法
if __name__ == "__main__":
    # 創(chuàng)建一個(gè)目標(biāo)文檔
    doc = Document()
    doc.Sections[0].Paragraphs.Add().AppendText("這是目標(biāo)文檔的第一段。")
    doc.Sections[0].Paragraphs.Add().AppendText("這是目標(biāo)文檔的第二段，我們將在其后插入內(nèi)容。")
    doc.Sections[0].Paragraphs.Add().AppendText("這是目標(biāo)文檔的第三段。")
    doc.SaveToFile("target_doc.docx", FileFormat.Docx)
    doc.Close()

    # 創(chuàng)建一個(gè)源文檔
    doc = Document()
    doc.Sections[0].Paragraphs.Add().AppendText("這是源文檔的第一段（將被插入）。")
    doc.Sections[0].Paragraphs.Add().AppendText("這是源文檔的第二段（將被插入）。")
    doc.SaveToFile("source_doc.docx", FileFormat.Docx)
    doc.Close()

    target_file = "target_doc.docx"
    source_file = "source_doc.docx"
    output_file_insert = "merged_output_insert.docx"
    
    # 在目標(biāo)文檔的第二個(gè)段落（索引為1）之后插入源文檔內(nèi)容
    insert_document_at_paragraph(target_file, source_file, 1, output_file_insert)