当前位置 博文首页 > Python实现给PDF添加水印的方法

    Python实现给PDF添加水印的方法

    作者:AXYZdong 时间:2021-02-02 12:10

    前言

    利用 PyPDF2 处理 PDF 文件,相关文档:https://pythonhosted.org/PyPDF2/

    本文针对 仅有 PDF 文件,而无相关 PDF 编辑器的情况下,给 PDF 添加水印。

    一、前期准备

    安装 PyPDF2 ,命令提示框输入:

    pip install PyPDF2

    新建 watermark.pdf 文件
    实际的水印,可以在此文件里修改水印文字的字体和位置。
    实现步骤: 新建 watermark.word ,【设计】 → \to → 【水印】【自定义水印】 → \to → 【文字水印】输入【文字】,修改【字体】、【自号】、【颜色】和【板式】

    在这里插入图片描述

    在这里插入图片描述

    效果

    在这里插入图片描述

    另存为 watermark.pdf

    watermark.pdf.py 文件放在同一目录下

    在这里插入图片描述

    二、代码实现

    // =============================================
    # @Time  : 2021-01-24
    # @Author : AXYZdong
    # @CSDN  : https://axyzdong.blog.csdn.net/
    # @FileName: PDFset.py
    # @Software: Python IDE
    // =============================================
    from PyPDF2 import PdfFileReader, PdfFileWriter
    
    def add_watermark(pdf_file_in, pdf_file_mark, pdf_file_out):
      """把水印添加到pdf中"""
      pdf_output = PdfFileWriter()
      input_stream = open(pdf_file_in, 'rb')
      pdf_input = PdfFileReader(input_stream, strict=False)
    
      # 获取PDF文件的页数
      pageNum = pdf_input.getNumPages()
    
      # 读入水印pdf文件
      pdf_watermark = PdfFileReader(open(pdf_file_mark, 'rb'), strict=False)
      # 给每一页打水印
      for i in range(pageNum):
        page = pdf_input.getPage(i)
        page.mergePage(pdf_watermark.getPage(0))
        page.compressContentStreams() # 压缩内容
        pdf_output.addPage(page)
      pdf_output.write(open(pdf_file_out, 'wb'))
    
    if __name__ == '__main__':
      pdf_file_in = '基于matlab的控制系统与仿真-1_AXYZdong的博客-CSDN博客.pdf'
      pdf_file_out = 'watermarked.pdf'
      pdf_file_mark = 'watermark.pdf'
      add_watermark(pdf_file_in, pdf_file_mark, pdf_file_out)
      
    

    效果

    在这里插入图片描述 

    三、其他方法

    // =============================================
    # @Time  : 2021-01-24
    # @Author : AXYZdong
    # @CSDN  : https://axyzdong.blog.csdn.net/
    # @FileName: PDFSet.py
    # @Software: Python IDE
    // =============================================
    from PyPDF2 import PdfFileReader, PdfFileWriter
    from reportlab.lib.units import cm
    from reportlab.pdfgen import canvas
    
    
    def create_watermark(content):
      """水印信息"""
      # 默认大小为21cm*29.7cm
      file_name = "mark.pdf"
      c = canvas.Canvas(file_name, pagesize=(30*cm, 30*cm))
      # 移动坐标原点(坐标系左下为(0,0))
      c.translate(10*cm, 5*cm)
    
      # 设置字体
      c.setFont("Helvetica", 30)
      # 指定描边的颜色
      c.setStrokeColorRGB(0, 1, 0)
      # 指定填充颜色
      c.setFillColorRGB(0, 1, 0)
      # 旋转45度,坐标系被旋转
      c.rotate(30)
      # 指定填充颜色
      c.setFillColorRGB(0, 0, 0, 0.1)
      # 设置透明度,1为不透明
      # c.setFillAlpha(0.1)
      # 画几个文本,注意坐标系旋转的影响
      for i in range(5):
        for j in range(10):
          a=10*(i-1)
          b=5*(j-2)
          c.drawString(a*cm, b*cm, content)
          c.setFillAlpha(0.1)
      # 关闭并保存pdf文件
      c.save()
      return file_name
    
    
    def add_watermark(pdf_file_in, pdf_file_mark, pdf_file_out):
      """把水印添加到pdf中"""
      pdf_output = PdfFileWriter()
      input_stream = open(pdf_file_in, 'rb')
      pdf_input = PdfFileReader(input_stream, strict=False)
    
      # 获取PDF文件的页数
      pageNum = pdf_input.getNumPages()
    
      # 读入水印pdf文件
      pdf_watermark = PdfFileReader(open(pdf_file_mark, 'rb'), strict=False)
      # 给每一页打水印
      for i in range(pageNum):
        page = pdf_input.getPage(i)
        page.mergePage(pdf_watermark.getPage(0))
        page.compressContentStreams() # 压缩内容
        pdf_output.addPage(page)
      pdf_output.write(open(pdf_file_out, 'wb'))
    
    if __name__ == '__main__':
      pdf_file_in = '基于matlab的控制系统与仿真-1_AXYZdong的博客-CSDN博客.pdf'
      pdf_file_out = 'watermarked.pdf'
      pdf_file_mark = create_watermark('CSDN@AXYZdong')
      add_watermark(pdf_file_in, pdf_file_mark, pdf_file_out)
    
    

    效果

    在这里插入图片描述

    四、总结

    •  Python通过PyPDF2实现PDF添加水印,实际上就是将两个PDF文件合并实现添加水印;
    • 第一种方法的水印文件是通过 word 另存为 PDF,第二种方法是自动创建一个 PDF 文件;
    • 第二种方法与第一种相比,水印的字体不能随意改变,不支持中文,省去了方法一中手动生成 PDF 水印的步骤;
    • 第一种方法可以任意修改水印的位置和字体。

    参考文献

    [1]:python pdf加水印
    [2]:Python中通过PyPDF2实现PDF添加水印

    js
    下一篇:没有了