Python处理docx：页眉与图片批注失效问题求助

问题根源 #

• 页眉批注失效：doc.paragraphs仅返回正文段落，页眉里的段落需要单独从页眉对象中获取，原代码没处理这部分。
• 图片批注失效：原代码对图片关联段落的判断逻辑不适用网络图片，且缺少正确的段落背景色设置函数实现。

完整修复代码 #

from docx import Document
from docx.oxml import OxmlElement
from docx.oxml.ns import qn

def set_paragraph_bg_color(paragraph, hex_color):
    """给段落设置背景色的核心函数"""
    # 获取段落的属性节点，不存在则创建
    pPr = paragraph._element.get_or_add_pPr()
    # 创建背景色设置的XML元素
    shd = OxmlElement('w:shd')
    shd.set(qn('w:fill'), hex_color)
    shd.set(qn('w:color'), 'auto')
    shd.set(qn('w:val'), 'clear')
    # 将背景色节点加入段落属性
    pPr.append(shd)

def annotate_images(doc):
    """批注包含图片的段落（兼容网络图片和本地嵌入图片）"""
    for paragraph in doc.paragraphs:
        for run in paragraph.runs:
            # 遍历run内的元素，找到图片对应的blip标签
            for elem in run._element.iter():
                if elem.tag.endswith('}blip'):
                    set_paragraph_bg_color(paragraph, 'ADD8E6')  # 浅蓝
                    break
            else:
                continue
            break

def process_docx(input_path, output_path):
    doc = Document(input_path)

    # 处理正文段落
    for paragraph in doc.paragraphs:
        if paragraph.style.name.startswith('Heading'):
            set_paragraph_bg_color(paragraph, 'FFFF00')  # 黄色
        elif paragraph.text.strip():  # 跳过空段落
            set_paragraph_bg_color(paragraph, 'D3D3D3')  # 浅灰

    # 处理页眉段落
    for section in doc.sections:
        header = section.header
        for paragraph in header.paragraphs:
            set_paragraph_bg_color(paragraph, 'FFFF00')  # 黄色

    # 处理图片段落
    annotate_images(doc)
    
    doc.save(output_path)

关键修复说明 #

1. 页眉处理：
   通过遍历文档的所有节（doc.sections），获取每个节的页眉对象（section.header），再遍历页眉内的段落设置背景色。
2. 图片处理：
   直接遍历段落中的run，查找Word图片对应的blip标签——不管是网络链接还是本地嵌入的图片，都会包含这个标签，找到后给对应段落设置背景色。
3. 背景色函数：
   按照Word的XML结构规范创建背景色节点，确保设置的背景色能正确写入文档。

内容的提问来源于stack exchange，提问作者sdd3242