问python-docx:在后面插入一个段落

EN

from docx import Document

from docx.text.paragraph import Paragraph
from docx.oxml.xmlchemy import OxmlElement

def insert_paragraph_after(paragraph, text=None, style=None):
    """Insert a new paragraph after the given paragraph."""
    new_p = OxmlElement("w:p")
    paragraph._p.addnext(new_p)
    new_para = Paragraph(new_p, paragraph._parent)
    if text:
        new_para.add_run(text)
    if style is not None:
        new_para.style = style
    return new_para

def main():
    # Create a minimal document
    document = Document()
    p1 = document.add_paragraph("First Paragraph.")
    p2 = document.add_paragraph("Second Paragraph.")

    # Insert a paragraph wedged between p1 and p2
    insert_paragraph_after(p1, "Paragraph One And A Half.")

    # Test if the function succeeded
    document.save(r"D:\somepath\docx_para_after.docx")

if __name__ == "__main__":
    main()

def par_index(paragraph):
    "Get the index of the paragraph in the document"
    doc = paragraph._parent
    # the paragraphs elements are being generated on the fly,
    # they change all the time
    # so in order to index, we must use the elements
    l_elements = [p._element for p in doc.paragraphs]
    return l_elements.index(paragraph._element)


def insert_paragraph_after(paragraph, text, style=None):
    """
    Add a paragraph to a docx document, after this one.
    """
    doc = paragraph._parent
    i = par_index(paragraph) + 1 # next
    if i <= len(doc.paragraphs):
        # we find the next paragraph and we insert before:
        next_par = doc.paragraphs[i]
        new_par = next_par.insert_paragraph_before(text, style)
    else:
        # we reached the end, so we need to create a new one:
        new_par = parent.add_paragraph(text, style)
    return new_par

para1 = document.add_paragraph("Hello World")
para2 = document.add_paragraph("Testing!!")

p1 = para1._p
p1.addnext(para2._p)