Python-72 用pypdf2对自己的pdf文件加入watermark水印标记 2020-09-25

用pypdf2对自己的pdf文件加入watermark水印标记。
备注：需要设置被加印文件路径；输出路径；水印文本pdf文件路径。
具体代码与实现如下：

# pdf_watermarker.py

from PyPDF2 import PdfFileWriter, PdfFileReader

def create_watermark(input_pdf, output, watermark):
    watermark_obj = PdfFileReader(watermark)
    watermark_page = watermark_obj.getPage(0)

    pdf_reader = PdfFileReader(input_pdf)
    pdf_writer = PdfFileWriter()

    # Watermark all the pages
    for page in range(pdf_reader.getNumPages()):
        page = pdf_reader.getPage(page)
        page.mergePage(watermark_page)
        pdf_writer.addPage(page)

    with open(output, 'wb') as out:
        pdf_writer.write(out)

if __name__ == '__main__':
# input_pdf: the PDF file path to be watermarked 
# output: the path you want to save the watermarked version of the PDF
#watermark: a PDF that contains your watermark image or text

    create_watermark(
        input_pdf='C:/Users/Mr.R/Desktop/test1/merged.pdf',
        output='C:/Users/Mr.R/Desktop/test1/merged1.pdf',
        watermark='C:/Users/Mr.R/Desktop/test1/c.pdf')

加了水印后的文件如下：

image.png