使用Python的pdf()函数生成带水印的PDF文件的方法

要生成带水印的PDF文件，可以使用Python中的pdf()函数。pdf()函数是PyPDF2库中的一个方法，可以用于创建PDF文档并添加水印。

下面是一个使用pdf()函数生成带水印的PDF文件的示例代码：

from PyPDF2 import PdfFileWriter, PdfFileReader
from reportlab.lib.pagesizes import letter
from reportlab.pdfgen import canvas

def add_watermark(input_pdf, watermark_text, output_pdf):
    # 打开输入的PDF文件
    input_pdf_obj = PdfFileReader(open(input_pdf, "rb"))
    # 创建一个新的PDF对象
    output_pdf_obj = PdfFileWriter()

    # 获取输入PDF的页数
    num_pages = input_pdf_obj.getNumPages()

    # 加载水印文本
    watermark_canvas = canvas.Canvas("watermark.pdf", pagesize=letter)
    watermark_canvas.setFont("Helvetica", 60)
    watermark_canvas.setFillColorRGB(0.5, 0.5, 0.5)
    watermark_canvas.rotate(45)
    watermark_canvas.drawString(200, 0, watermark_text)
    watermark_canvas.save()

    # 打开水印文件
    watermark_obj = PdfFileReader(open("watermark.pdf", "rb"))

    # 逐页将输入PDF和水印合并
    for i in range(num_pages):
        page = input_pdf_obj.getPage(i)
        page.mergePage(watermark_obj.getPage(0))
        output_pdf_obj.addPage(page)

    # 将带水印的PDF保存
    with open(output_pdf, "wb") as output_file:
        output_pdf_obj.write(output_file)

    # 删除临时的水印文件
    watermark_obj.stream.close()
    watermark_obj.close()

    watermark_canvas.stream.close()
    watermark_canvas.save()

    # 打印完成的消息
    print("带水印的PDF文件已生成：" + output_pdf)

# 使用示例
if __name__ == "__main__":
    input_pdf_file = "sample.pdf"  # 输入的PDF文件名
    output_pdf_file = "sample_watermarked.pdf"  # 生成的带水印的PDF文件名
    watermark_text = "CONFIDENTIAL"  # 水印文本

    add_watermark(input_pdf_file, watermark_text, output_pdf_file)

以上代码中，我们首先导入所需的模块和函数。然后，定义了一个add_watermark()函数，该函数将输入的PDF文件与水印合并，并创建一个带水印的新PDF文件。

在add_watermark()函数中，我们首先打开输入的PDF文件，并创建一个新的PDF对象。然后，获取输入PDF的页数。

接下来，我们创建一个水印文本，并将其保存为一个临时的PDF文件。接着，我们打开水印文件，并逐页将输入PDF和水印合并。合并后的页面被添加到新的PDF对象中。

最后，我们保存带水印的PDF文件，并删除临时的水印文件。最终，打印出带水印的PDF文件已生成的消息。

在示例中，我们传入了输入的PDF文件名、水印文本和生成的带水印的PDF文件名。你可以根据自己的需求修改这些参数。

请确保你已经安装并导入了所需的库，比如PyPDF2、reportlab等。你可以使用pip install命令安装缺失的库。

通过运行以上代码，你就可以生成带有水印的PDF文件。