python批量添加水印pdf

你可以使用以下代码来实现Python批量添加水印到PDF文件的功能：

from PyPDF2 import PdfReader, PdfWriter

def add_watermark(watermark_file, pdf_files):
    # 读取水印文件，并获取含有水印的页
    watermark_reader = PdfReader(watermark_file)
    watermark_page = watermark_reader.pages[0]

    # 遍历所有PDF文件
    for pdf_file in pdf_files:
        # 创建一个写缓存，用于缓存合并后的结果页
        pdf_writer = PdfWriter()

        # 读取PDF文件
        pdf_reader = PdfReader(pdf_file)

        # 遍历PDF的所有页，并将水印页合并到当前页上
        for pdf_page in pdf_reader.pages:
            pdf_page.merge_page(watermark_page)
            pdf_writer.add_page(pdf_page)

        # 将合并后的PDF文件写入到指定地址
        with open("_水印".join(os.path.splitext(pdf_file)), "wb") as result_path:
            pdf_writer.write(result_path)

# 获取指定目录下所有PDF文件
pdf_files = get_pdf_files(input_path)

# 批量添加水印到PDF文件
add_watermark(watermark_file, pdf_files)