使用 Python 为 PDF 添加水印

在现代数字化办公环境中，PDF 已成为一种广泛使用的文件格式，尤其是在需要保持文档格式时。为了保护文档内容，添加水印是一种常见的方法。本文将介绍如何使用 Python 为 PDF 文件添加水印，具体步骤将通过代码示例详细说明。

一、准备工作

要实现 PDF 水印的添加，我们将使用 Spire.PDF，这是一个强大的 PDF 操作库，能够轻松处理 PDF 文档。你可以从 Spire.PDF 官方网站下载和安装相关库。确保在你的环境中配置好 Python 和必要的依赖库。

二、实现步骤

1. 导入必要的库

首先，我们需要导入 Spire.PDF 相关的类。以下是基本的导入语句：

python 复制代码

from spire.pdf import *
from spire.pdf.common import *
import math

2. 创建 PdfDocument 对象

接下来，我们需要创建一个 PdfDocument 类的对象，并加载要处理的 PDF 文件。

python 复制代码

# 创建PdfDocument类的对象
doc = PdfDocument()

# 从指定路径加载PDF文档
doc.LoadFromFile("C:\\Users\\Administrator\\Desktop\\Input.pdf")

在这里，我们可以将路径替换为你的 PDF 文件的实际路径。

3. 设置水印参数

我们需要定义水印的文本内容和字体样式。在本例中，我们将使用"禁止复制"作为水印文本。

python 复制代码

# 为水印字体创建PdfTrueTypeFont类的对象
font = PdfTrueTypeFont("黑体", 48.0, 0, True)

# 指定水印文本
text = "禁 止 复 制"

字体可以根据你的需要进行调整，例如使用不同的文本样式或大小。

4. 测量文本尺寸

为了确保水印文本能够正确地显示在 PDF 中，我们需要测量文本的宽度和高度：

python 复制代码

# 测量文本的尺寸以确保正确定位
text_width = font.MeasureString(text).Width
text_height = font.MeasureString(text).Height

5. 遍历每一页

现在，我们将循环遍历 PDF 文件的每一页，并在每页上添加水印。

python 复制代码

# 循环遍历文档中的每一页
for i inrange(doc.Pages.Count):
# 获取当前页面
    page = doc.Pages.get_Item(i)

    # 保存当前画布状态
    state = page.Canvas.Save()
    
    # 计算页面的中心坐标
    x = page.Canvas.Size.Width / 2
    y = page.Canvas.Size.Height / 2

    # 将坐标系平移到中心，使页面的中心成为原点(0, 0)
    page.Canvas.TranslateTransform(x, y)

    # 将画布逆时针旋转45度以显示水印
    page.Canvas.RotateTransform(-45.0)

    # 设置水印的透明度
    page.Canvas.SetTransparency(0.4)
        
    # 使用负偏移量在中心位置绘制水印文本
    page.Canvas.DrawString(text, font, PdfBrushes.get_Blue(), PointF(-text_width / 2, -text_height / 2))
    
    # 恢复画布状态，以防止变换影响后续绘图
    page.Canvas.Restore(state)

在上述代码中，我们使用了以下步骤：

获取当前页面，保存画布状态。
计算页面的中心坐标，并将坐标系平移到中心。
旋转画布以逆时针显示水印。
设置透明度，使得水印不太显眼。
将水印文本绘制到页面中心。
恢复画布状态，保证之后的操作不受影响。

6. 保存修改后的 PDF

最后，我们将修改后的 PDF 文档保存到新的文件中：

python 复制代码

# 将修改后的文档保存到新的PDF文件
doc.SaveToFile("output/TextWatermark.pdf")
doc.Dispose()

这里，我们指定了新的输出文件路径，确保保存成功。

三、总结

通过上述步骤，我们成功地使用 Python 为 PDF 文件添加了水印。这种方法简单有效，适用于需要保护文档内容的场合。

在实际应用中，水印的样式和内容可以根据需要灵活调整。无论是企业文档、个人作品，还是仅仅为了防止未经授权的复制，通过水印都能够更好地保护你的知识产权。

如果你对 PDF 操作有更多的需求，Spire.PDF 还支持其他许多功能，比如合并、拆分 PDF 文档，提取文本和图像等。希望本文能为你在使用 Python 操作 PDF 提供一些实用的参考。