python pdf 去水印

import fitz  # PyMuPDF

def remove_watermark(input_pdf, output_pdf, watermark_text):
    # 打开输入的PDF文件
    doc = fitz.open(input_pdf)
    
    for page in doc:
        # 查找并删除指定的水印文本
        watermark_instances = page.searchFor(watermark_text)
        for inst in watermark_instances:
            highlight = page.addHighlightAnnot(inst)
            highlight.update()
            page.deleteObject(highlight)
    
    # 保存修改后的PDF文件
    doc.save(output_pdf)
    doc.close()

# 使用示例
input_pdf_path = 'input.pdf'
output_pdf_path = 'output.pdf'
watermark_to_remove = 'DRAFT'

remove_watermark(input_pdf_path, output_pdf_path, watermark_to_remove)