使用Python的PdfFileWriter()库生成水印PDF文件

要使用Python的PdfFileWriter()库生成带水印的PDF文件，可以按照以下步骤进行操作：

1. 安装依赖库：首先需要安装PyPDF2库，这是一个用于处理PDF文件的Python库。可以使用pip命令进行安装：pip install PyPDF2

2. 导入库：在Python脚本中导入PyPDF2库：from PyPDF2 import PdfReader, PdfWriter, PdfFileMerger

3. 打开PDF文件：使用PdfReader类的open()方法打开PDF文件，并创建一个PdfReader对象。例如：pdf = PdfReader('input.pdf')

4. 创建PdfWriter对象：使用PdfWriter类的构造函数创建一个PdfWriter对象。例如：pdf_writer = PdfWriter()

5. 添加水印：使用PdfWriter对象的addPage()方法将PdfReader对象的每一页添加到PdfWriter对象中。同时，可以使用PdfFileMerger类的merge()方法将水印文件合并到每一页的背景中。例如：

watermark = PdfReader('watermark.pdf')
watermark_page = watermark.getPage(0)
for page in pdf.pages:
    page.mergePage(watermark_page)
    pdf_writer.addPage(page)

6. 保存PDF文件：使用PdfWriter对象的write()方法将PdfWriter对象中的内容保存为新的PDF文件。例如：pdf_writer.write(open('output.pdf', 'wb'))

下面是一个完整的示例代码，用于生成带有水印的PDF文件：

from PyPDF2 import PdfReader, PdfWriter, PdfFileMerger

def add_watermark(input_file, watermark_file, output_file):
    pdf = PdfReader(input_file)
    pdf_writer = PdfWriter()
    
    watermark = PdfReader(watermark_file)
    watermark_page = watermark.getPage(0)
    
    for page in pdf.pages:
        page.mergePage(watermark_page)
        pdf_writer.addPage(page)
        
    pdf_writer.write(open(output_file, 'wb'))

add_watermark('input.pdf', 'watermark.pdf', 'output.pdf')

在上面的代码中，通过调用add_watermark()函数，将输入文件input.pdf和水印文件watermark.pdf作为参数传递，并指定输出文件为output.pdf。生成的output.pdf文件将包含原始PDF文件的每一页，并带有设置的水印。

请注意，这只是一个简单的示例，用于演示如何使用PdfFileWriter库生成带有水印的PDF文件。可以根据自己的需求进行修改和扩展。