16 添加水印

在上一篇中，我们学习了如何将一个PDF文件拆分成多个部分。接下来，我们将深入探讨如何在PDF文件中添加水印。这一过程将帮助你保护你的文件，也可以对文件进行标记以便于管理和阅读。

理解水印的概念

水印是一种常见的文档处理技术，它可以在PDF文件的每一页上添加文本或图像，以显示版权信息、机密性或说明性信息。添加水印可以让你更好地保护自己的知识产权，尤其是在分发文件时。

使用Python的PyPDF2库添加水印

为了在PDF文件中添加水印，我们将使用PyPDF2库。首先，你需要确保已经安装了该库。如果没有安装，可以使用以下命令安装：

pip install PyPDF2

接下来，我们将通过一个简单的示例来展示如何在PDF中添加水印。

示例：给PDF文件添加文本水印

假设我们有一个名为original.pdf的PDF文档，我们想在其每一页上添加一个水印“版权所有 © 2023”。

步骤 1: 创建水印PDF

首先，我们需要创建一个包含水印文本的PDF文件。我们可以使用reportlab库来实现这一点。安装reportlab：

pip install reportlab

然后，我们创建一个水印PDF，如下：

from reportlab.lib.pagesizes import letter
from reportlab.pdfgen import canvas

def create_watermark(output_path):
    c = canvas.Canvas(output_path, pagesize=letter)
    c.setFont("Helvetica", 40)
    c.setFillColorRGB(0.8, 0.8, 0.8)  # 灰色
    c.drawString(100, 500, "版权所有 © 2023")
    c.save()

create_watermark("watermark.pdf")

这段代码会生成一个名为watermark.pdf的PDF文件，其中包含我们的水印。

步骤 2: 将水印应用到原始PDF

接下来，我们将水印应用到原始的original.pdf文件。如下所示：

from PyPDF2 import PdfReader, PdfWriter

def add_watermark(original_pdf_path, watermark_pdf_path, output_pdf_path):
    with open(original_pdf_path, "rb") as original_file, open(watermark_pdf_path, "rb") as watermark_file:
        original_pdf = PdfReader(original_file)
        watermark_pdf = PdfReader(watermark_file)
        writer = PdfWriter()

        watermark_page = watermark_pdf.pages[0]

        for page in original_pdf.pages:
            page.merge_page(watermark_page)
            writer.add_page(page)

        with open(output_pdf_path, "wb") as output_file:
            writer.write(output_file)

add_watermark("original.pdf", "watermark.pdf", "watermarked_output.pdf")

这段代码将读取原始PDF和水印PDF，并为原始PDF的每一页添加水印，最终输出为watermarked_output.pdf。

结果

在运行以上代码之后，你将会在当前目录下找到watermarked_output.pdf文件，里面的每一页都添加了你所指定的水印文字。你可以使用任何PDF阅读器打开该文件查看效果。

总结

在这一篇中，我们学习了如何使用Python的PyPDF2和reportlab库来在PDF文档中添加水印。这种方法不仅简单有效，还能帮助我们在文件中添加重要信息。

在下一篇教程中，我们将讨论如何在PDF中添加注释和使用技巧，这会进一步增强我们对PDF文档的管理能力，确保我们的信息更易于理解和交流。

如果你有任何问题或想法，请随时在下方留言，我们将共同学习和探讨！