Python中使用PDFDocument()生成带水印的PDF文档的方法

要使用Python中的PDFDocument()生成带水印的PDF文档，您需要使用适当的库来处理PDF文件。 在以下示例中，我们将使用PyPDF2库来生成带水印的PDF文档。

首先，您需要安装PyPDF2库。 可以使用以下命令在Python中安装它：

pip install PyPDF2

接下来，我们将按照以下步骤生成带水印的PDF文档：

1. 导入所需的库：

from PyPDF2 import PdfReader, PdfWriter
from PyPDF2.pdf import PageObject
from reportlab.pdfgen import canvas
from io import BytesIO

2. 创建一个水印PDF文档。 您可以使用ReportLab库中的canvas方法来创建水印。 在这个例子中，我们将创建一个带有“水印”文本的PDF页面，文本将处于45度角并且透明度设置为0.3。 您可以根据需要更改文本和样式：

def create_watermark():
    packet = BytesIO()
    # 创建水印页面
    c = canvas.Canvas(packet)
    c.setFont("Helvetica", 60)
    c.rotate(45)
    c.setFillColorRGB(0.5, 0.5, 0.5, 0.3)  # 设置颜色和透明度
    c.drawString(100, 200, "水印")
    c.save()
    packet.seek(0)
    watermark = PdfReader(packet)
    return watermark.getPage(0)

3. 创建一个带有水印的PDF文档。 在这个例子中，我们将使用一个现有的PDF文件作为输入，并将水印添加到每个页面上。 您可以根据需要更改输入和输出文件的路径：

def add_watermark(input_file, output_file):
    watermark = create_watermark()

    # 打开输入文件和输出文件
    input_pdf = PdfReader(input_file)
    output_pdf = PdfWriter()

    # 遍历输入PDF的每个页面并添加水印
    for page in input_pdf.pages:
        page_obj = PageObject.createBlankPage(None, page.mediaBox.getWidth(), page.mediaBox.getHeight())
        page_obj.mergePage(page)
        page_obj.mergePage(watermark)
        output_pdf.addPage(page_obj)

    # 保存输出PDF
    with open(output_file, "wb") as f:
        output_pdf.write(f)

    print("带水印的PDF文档已生成。")

4. 在主程序中调用add_watermark函数，并提供输入和输出文件的路径：

if __name__ == "__main__":
    input_file = "input.pdf"
    output_file = "output.pdf"
    add_watermark(input_file, output_file)

在这个例子中，我们假设已经有一个名为input.pdf的PDF文件作为输入。 然后，我们创建了一个名为output.pdf的带有水印的PDF文档。

请注意，这只是一个简单的示例来演示如何使用PyPDF2库生成带水印的PDF文档。 根据您的具体需求，您可能需要进行进一步的自定义和调整。