使用python批量给文件添加水印

from PyPDF2 import PdfReader, PdfWriter
import os

result = os.listdir('input')   # 批量获取文件

reader1 = PdfReader('files/MYWater.pdf')   # 准备水印页
water = reader1.getPage(0)

writer = PdfWriter()                    # 创建一个空的pdf

for x in result:
    reader2 = PdfReader('input/' + x)   # 给未加水印文件的文件路径
    for i in range(reader2.getNumPages()):
        page = reader2.getPage(i)  # 得到需添加水印的每一页
        page.mergePage(water)  # 给每一页添加水印
        writer.addPage(page)  # 将添加水印后的文件放入新的pdf文件
writer.write('out/' + x)      # 保存新的pdf文件