快捷導(dǎo)航

Python實現(xiàn)給PDF添加水印的方法

更新時間：2021年01月25日 16:30:51 作者：AXYZdong

這篇文章主要介紹了Python實現(xiàn)給PDF添加水印的方法，文中通過示例代碼介紹的非常詳細，對大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價值，需要的朋友們下面隨著小編來一起學(xué)習(xí)學(xué)習(xí)吧

前言

利用 PyPDF2 處理 PDF 文件，相關(guān)文檔：https://pythonhosted.org/PyPDF2/

本文針對僅有 PDF 文件，而無相關(guān) PDF 編輯器的情況下，給 PDF 添加水印。

一、前期準備

安裝 PyPDF2 ，命令提示框輸入:

pip install PyPDF2

新建 watermark.pdf 文件
實際的水印，可以在此文件里修改水印文字的字體和位置。
實現(xiàn)步驟：新建 watermark.word ，【設(shè)計】 → \to → 【水印】【自定義水印】 → \to → 【文字水印】輸入【文字】，修改【字體】、【自號】、【顏色】和【板式】

在這里插入圖片描述

效果

在這里插入圖片描述

另存為 watermark.pdf

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

在這里插入圖片描述

二、代碼實現(xiàn)

// =============================================
# @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文件的頁數(shù)
  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() # 壓縮內(nèi)容
    pdf_output.addPage(page)
  pdf_output.write(open(pdf_file_out, 'wb'))

if __name__ == '__main__':
  pdf_file_in = '基于matlab的控制系統(tǒng)與仿真-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)

  # 設(shè)置字體
  c.setFont("Helvetica", 30)
  # 指定描邊的顏色
  c.setStrokeColorRGB(0, 1, 0)
  # 指定填充顏色
  c.setFillColorRGB(0, 1, 0)
  # 旋轉(zhuǎn)45度,坐標系被旋轉(zhuǎn)
  c.rotate(30)
  # 指定填充顏色
  c.setFillColorRGB(0, 0, 0, 0.1)
  # 設(shè)置透明度,1為不透明
  # c.setFillAlpha(0.1)
  # 畫幾個文本,注意坐標系旋轉(zhuǎn)的影響
  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)
  # 關(guān)閉并保存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文件的頁數(shù)
  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() # 壓縮內(nèi)容
    pdf_output.addPage(page)
  pdf_output.write(open(pdf_file_out, 'wb'))

if __name__ == '__main__':
  pdf_file_in = '基于matlab的控制系統(tǒng)與仿真-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)

效果

在這里插入圖片描述