快捷導(dǎo)航

Python給PDF添加水印的代碼步驟

更新時(shí)間：2025年02月26日 10:29:32 作者：蠟筆小新星

在本教程中,我們將學(xué)習(xí)如何使用 Python 編程語(yǔ)言以及 PyPDF2 和 reportlab 庫(kù)來(lái)向 PDF 文檔中添加水印,水印通常用于標(biāo)記文檔的版權(quán)信息、保密級(jí)別或其他重要通知,需要的朋友可以參考下

步驟 1：安裝必要的庫(kù)

首先，確保你已經(jīng)安裝了 PyPDF2 和 reportlab 庫(kù)。如果尚未安裝，可以使用以下命令通過(guò) pip 安裝它們：

pip install PyPDF2 reportlab

步驟 2：準(zhǔn)備字體文件

由于我們要在水印中使用中文，因此需要準(zhǔn)備一個(gè)支持中文的字體文件，例如 SimHei.ttf（黑體）。確保這個(gè)字體文件與你的 Python 腳本位于同一目錄下，或者你知道它的完整路徑。

步驟 3：編寫 Python 腳本

接下來(lái)，我們編寫一個(gè) Python 腳本來(lái)生成水印并將其添加到現(xiàn)有的 PDF 文檔中。

import io
from PyPDF2 import PdfWriter, PdfReader
from reportlab.lib import pagesizes
from reportlab.lib.units import cm
from reportlab.pdfbase import pdfmetrics
from reportlab.pdfbase.ttfonts import TTFont
from reportlab.pdfgen import canvas

# 注冊(cè)字體，以便在 PDF 中使用中文
pdfmetrics.registerFont(TTFont('SimHei', 'SimHei.ttf'))

# 生成水印文件的函數(shù)
def create_water_mark(text):
    packet = io.BytesIO()
    # 創(chuàng)建一個(gè)新的 PDF 頁(yè)面作為水印
    my_canvas = canvas.Canvas(packet, pagesizes.A4)
    # 設(shè)置水印字體和大小
    my_canvas.setFont("SimHei", 20)
    # 設(shè)置填充色為黑色
    my_canvas.setFillColorRGB(0, 0, 0)
    # 設(shè)置透明度（0.0 完全透明，1.0 完全不透明）
    my_canvas.setFillAlpha(0.1)
    # 旋轉(zhuǎn)字體以增加水印效果
    my_canvas.rotate(30)
    # 在頁(yè)面上多次繪制水印文本，以形成水印圖案
    for i in range(3, 24, 10):
        for j in range(-5, 30, 5):
            my_canvas.drawString(i * cm, j * cm, text)
    my_canvas.save()
    packet.seek(0)
    # 返回水印頁(yè)面的 PdfReader 對(duì)象
    return PdfReader(packet)

# 將水印添加到 PDF 文件的函數(shù)
def add_watermark(input_pdf_path, output_pdf_path, watermark_text):
    # 創(chuàng)建水印
    watermark = create_water_mark(watermark_text)
    # 讀取輸入 PDF 文件
    pdf_reader = PdfReader(input_pdf_path)
    pdf_writer = PdfWriter()
    # 遍歷輸入 PDF 的每一頁(yè)，并添加水印
    for page in pdf_reader.pages:
        page.merge_page(watermark.pages[0])
        pdf_writer.add_page(page)
    # 將帶有水印的 PDF 寫入輸出文件
    with open(output_pdf_path, "wb") as output_pdf:
        pdf_writer.write(output_pdf)

# 主程序
if __name__ == '__main__':
    # 輸入和輸出 PDF 文件的路徑
    input_pdf = r"input.pdf"  # 替換為你的輸入 PDF 路徑
    output_pdf = r"output_with_watermark.pdf"  # 期望的輸出 PDF 路徑
    watermark_text = "添加水印測(cè)試文本"  # 水印文本內(nèi)容
    # 調(diào)用函數(shù)添加水印
    add_watermark(input_pdf, output_pdf, watermark_text)