【Python】給PDF添加水印

Author：AXYZdong 自動化專業 工科男
有一點思考，有一點想法，有一點理性！
定個小小目標，努力成為習慣！在最美的年華遇見更好的自己！
CSDN@AXYZdong，CSDN首發，AXYZdong原創
唯一博客更新的地址為： 👉 AXYZdong的博客 👈

前言

利用 PyPDF2 處理 PDF 檔案，相關檔案：https://pythonhosted.org/PyPDF2/

本文針對 僅有 PDF 檔案，而無相關 PDF 編輯器的情況下，給 PDF 添加水印，

一、前期準備

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

pip install PyPDF2

• 新建 watermark.pdf 檔案
  實際的水印，可以在此檔案里修改水印文字的字體和位置，
  實作步驟：

1. 新建 watermark.word ，【設計】 → \to → 【水印】
2. 【自定義水印】 → \to → 【文字水印】
3. 輸入【文字】，修改【字體】、【自號】、【顏色】和【板式】

• 效果
  

• 另存為 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添加水印

??本次的分享就到這里

好書不厭百回讀，熟讀自知其中意，讓學習成為習慣，用知識改變命運，讓博客見證成長，用行動證明努力，
如果我的博客對你有幫助、如果你喜歡我的博客內容，請 “點贊” “評論” “收藏” 一鍵三連哦！
聽說 👉 點贊 👈 的人運氣不會太差，每一天都會元氣滿滿呦！^ _ ^ ?? ?? ??
碼字不易，大家的支持就是我堅持下去的動力，點贊后不要忘了👉 關注 👈我哦！
更多精彩內容請前往 AXYZdong的博客

如果以上內容有任何錯誤或者不準確的地方，歡迎在下面 👇 留個言，或者你有更好的想法，歡迎一起交流學習~~~