在Python中使用 PyPDF4 将水印添加到 PDF

在 Python 中，可以使用 PyPDF4 库来对 PDF 文件进行处理。其中一个功能就是将水印添加到 PDF 文件中。

安装 PyPDF4

要使用 PyPDF4，首先需要安装它。可以通过 pip 安装它：

pip install PyPDF4

添加水印

下面是一个示例代码，演示如何将一个文本水印添加到一个 PDF 文件中：

import PyPDF4

# 打开 PDF 文件
input_file = open('input_file.pdf', 'rb')
input_pdf = PyPDF4.PdfFileReader(input_file)

# 准备输出 PDF 文件
output_file = open('output_file.pdf', 'wb')
output_pdf = PyPDF4.PdfFileWriter()

# 准备水印文字
text = 'Confidential'

# 将水印添加到每一页
for i in range(input_pdf.getNumPages()):
    # 获取页面对象
    page = input_pdf.getPage(i)

    # 创建 PDF 文字对象
    watermark = PyPDF4.pdf.TextStringObject(text)

    # 创建 PDF 文字元素，设置透明度和旋转角度
    text_element = PyPDF4.pdf.SimplePDFTextElement(
        x=page.mediaBox.getUpperRight_x() - 40,
        y=page.mediaBox.getUpperRight_y() - 40,
        text=watermark,
        font_size=24.0,
        font_color=(1, 0, 0),  # 红色
        font_name='Helvetica-Bold',
        angle=45,
        alpha=0.3
    )

    # 将文字元素添加到页面
    page.addElement(text_element)

    # 将页面添加到输出 PDF 文件中
    output_pdf.addPage(page)

# 输出 PDF 文件
output_pdf.write(output_file)

# 关闭文件
input_file.close()
output_file.close()

代码中，首先打开一个输入的 PDF 文件，并创建一个输出的 PDF 输出流。然后遍历输入 PDF 文件的每一页，将指定的文本水印添加到每一页的右上角。

可以通过更改 text_element 中的属性来自定义水印的样式和位置。例如，font_size 属性可以更改水印的字体大小。font_name 属性可以更改水印的字体。alpha 属性可以更改水印的透明度。angle 属性可以更改水印的旋转角度。

结论

这个示例演示了使用 PyPDF4 将水印添加到 PDF 文件的基本步骤。可以通过自定义 text_element 的属性来实现水印的各种样式，并将水印添加到指定位置。如果需要更复杂的操作，可以参考 PyPDF4 文档，了解更多功能。