python如何添加pdf水印

在Python中添加PDF水印可以通过使用第三方库来实现，如PyPDF2、reportlab和pdfrw。选择合适的库可以根据需求的复杂性和功能来决定。本文将详细介绍如何使用这些库来为PDF文件添加水印。

一、使用PyPDF2添加水印

PyPDF2是一个纯Python PDF库，用于处理PDF文件。它可以读取、合并、拆分和修改PDF文件。添加水印的过程主要涉及读取原始PDF和水印PDF，然后将水印PDF合并到原始PDF的每一页上。

安装PyPDF2

要使用PyPDF2，首先需要安装这个库。可以通过pip命令来安装：
```
pip install PyPDF2
```

读取PDF文件

使用PyPDF2读取需要添加水印的PDF文件和水印PDF文件。水印通常是一个单页PDF文件。

import PyPDF2
读取PDF文件
with open('original.pdf', 'rb') as original_file, open('watermark.pdf', 'rb') as watermark_file:
    original_pdf = PyPDF2.PdfReader(original_file)
    watermark_pdf = PyPDF2.PdfReader(watermark_file)
    watermark_page = watermark_pdf.pages[0]

合并PDF文件

将水印应用到每一页的过程中，逐页读取原始PDF并合并水印。

# 创建一个新的PDF对象
output_pdf = PyPDF2.PdfWriter()
遍历每一页，添加水印
for page_number in range(len(original_pdf.pages)):
    page = original_pdf.pages[page_number]
    page.merge_page(watermark_page)
    output_pdf.add_page(page)
写入新的PDF文件
with open('watermarked_output.pdf', 'wb') as output_file:
    output_pdf.write(output_file)

二、使用reportlab创建水印

Reportlab是一款用于生成PDF的库，支持创建复杂的PDF文件和图形。在创建水印时，可以使用Reportlab生成一个PDF文件，然后通过其他库（如PyPDF2）将其应用到目标PDF上。

安装reportlab

使用pip命令安装reportlab：
```
pip install reportlab
```

创建水印PDF

使用reportlab生成一个包含水印文本或图像的PDF文件。

from reportlab.pdfgen import canvas
def create_watermark(output_path, text):
    c = canvas.Canvas(output_path)
    c.setFont("Helvetica", 40)
    c.setFillColorRGB(0.5, 0.5, 0.5, alpha=0.3)  # 设置水印颜色和透明度
    c.drawString(100, 500, text)  # 设置水印位置
    c.save()
create_watermark('watermark.pdf', 'CONFIDENTIAL')

合并水印

使用PyPDF2将生成的水印PDF合并到目标PDF上，如上面的步骤所述。

三、使用pdfrw添加水印

pdfrw是一个用于读写PDF文件的轻量级库，支持对PDF文件的解析和修改。

安装pdfrw

通过pip安装pdfrw：
```
pip install pdfrw
```

添加水印

使用pdfrw可以实现类似于PyPDF2的水印合并功能。

from pdfrw import PdfReader, PdfWriter, PageMerge
def add_watermark(input_pdf, output_pdf, watermark_pdf):
    reader = PdfReader(input_pdf)
    watermark = PdfReader(watermark_pdf).pages[0]
    writer = PdfWriter()
    for page in reader.pages:
        merger = PageMerge(page)
        merger.add(watermark).render()
        writer.addpage(page)
    writer.write(output_pdf)
add_watermark('original.pdf', 'watermarked_output.pdf', 'watermark.pdf')