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

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

在本教程中,我們將學習如何使用 Python 編程語言以及 PyPDF2 和 reportlab 庫來向 PDF 文檔中添加水印,水印通常用于標記文檔的版權信息、保密級別或其他重要通知,需要的朋友可以參考下

步驟 1：安裝必要的庫

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

pip install PyPDF2 reportlab

步驟 2：準備字體文件

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

步驟 3：編寫 Python 腳本

接下來，我們編寫一個 Python 腳本來生成水印并將其添加到現(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

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

# 生成水印文件的函數(shù)
def create_water_mark(text):
    packet = io.BytesIO()
    # 創(chuàng)建一個新的 PDF 頁面作為水印
    my_canvas = canvas.Canvas(packet, pagesizes.A4)
    # 設置水印字體和大小
    my_canvas.setFont("SimHei", 20)
    # 設置填充色為黑色
    my_canvas.setFillColorRGB(0, 0, 0)
    # 設置透明度（0.0 完全透明，1.0 完全不透明）
    my_canvas.setFillAlpha(0.1)
    # 旋轉字體以增加水印效果
    my_canvas.rotate(30)
    # 在頁面上多次繪制水印文本，以形成水印圖案
    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)
    # 返回水印頁面的 PdfReader 對象
    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 的每一頁，并添加水印
    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 = "添加水印測試文本"  # 水印文本內(nèi)容
    # 調用函數(shù)添加水印
    add_watermark(input_pdf, output_pdf, watermark_text)

步驟 4：運行腳本

將上述腳本保存為一個 .py 文件，并確保 SimHei.ttf 字體文件與腳本位于同一目錄下。然后，在命令行中運行該腳本：

python your_script_name.py

將 your_script_name.py 替換為你保存腳本的文件名。

步驟 5：檢查結果

腳本運行完成后，檢查指定的輸出目錄，你應該會看到一個名為 output_with_watermark.pdf 的新文件。打開它，你將看到每一頁都添加了指定的水印文本。

注意事項

確保 SimHei.ttf 字體文件與你的腳本在同一目錄下，或者正確指定其路徑。
reportlab 庫用于生成水印頁面，而 PyPDF2 庫用于讀取和寫入 PDF 文件以及合并頁面。
水印的透明度和旋轉角度可以根據(jù)需要進行調整。
本教程中的代碼是一個基本示例，可能需要根據(jù)具體情況進行修改和優(yōu)化。

到此這篇關于Python給PDF添加水印的代碼步驟的文章就介紹到這了,更多相關Python PDF添加水印內(nèi)容請搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關文章希望大家以后多多支持腳本之家！

您可能感興趣的文章:

Pandas之drop_duplicates:去除重復項方法
下面小編就為大家分享一篇Pandas之drop_duplicates:去除重復項方法，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧
2018-04-04
Python如何存儲數(shù)據(jù)到json文件
這篇文章主要介紹了Python如何存儲數(shù)據(jù)到json文件,文中通過示例代碼介紹的非常詳細，對大家的學習或者工作具有一定的參考學習價值,需要的朋友可以參考下
2020-03-03
Python機器學習應用之基于線性判別模型的分類篇詳解
線性判別分析(Linear?Discriminant?Analysis,?LDA)是一種監(jiān)督學習的降維方法，也就是說數(shù)據(jù)集的每個樣本是有類別輸出。和之前介紹的機器學習降維之主成分分析(PCA)方法不同，PCA是不考慮樣本類別輸出的無監(jiān)督學習方法
2022-01-01
python高階函數(shù)map（）和reduce（）實例解析
這篇文章主要介紹了python高階函數(shù)map（）和reduce（）實例解析,文中通過示例代碼介紹的非常詳細，對大家的學習或者工作具有一定的參考學習價值,需要的朋友可以參考下
2020-03-03
python第三方模塊xmltodict庫優(yōu)雅處理xml格式為json
這篇文章主要為大家介紹了python第三方模塊xmltodict庫優(yōu)雅處理xml格式為json實例探究,有需要的朋友可以借鑒參考下,希望能夠有所幫助,祝大家多多進步,早日升職加薪
2024-01-01
最簡化ChatGLM2-6b小白部署教程【windows系統(tǒng)，16G內(nèi)存即可，對顯卡無要求，CPU運行】
這篇文章主要介紹了最簡化ChatGLM2-6b小白部署的方法,可以在windows系統(tǒng)環(huán)境下，16G內(nèi)存且對顯卡無要求,并以CPU模式運行,方便實現(xiàn)個人PC機環(huán)境下的部署體驗,需要的朋友可以參考下
2023-07-07
Python析構函數(shù)__del__定義原理解析
這篇文章主要介紹了Python析構函數(shù)__del__定義原理解析,文中通過示例代碼介紹的非常詳細，對大家的學習或者工作具有一定的參考學習價值,需要的朋友可以參考下
2020-11-11
公眾號接入chatGPT的詳細教程附Python源碼
這篇文章主要介紹了公眾號接入chatGPT教程附Python源碼,這里需要大家準備一個域名，一臺服務器和一個公眾號，本文結合實例代碼給大家介紹的非常詳細，需要的朋友可以參考下
2023-02-02
Python實現(xiàn)創(chuàng)建詞云的示例詳解
詞云一般是根據(jù)輸入的大量詞語生成的,如果某個詞語出現(xiàn)的次數(shù)越多,那么相應的大小就會越大,本文將利用wordcloud模塊實現(xiàn)詞云生成,需要的可以參考下
2023-10-10
使用Python的turtle模塊繪制櫻花樹的代碼示例
Python的turtle模塊是一個直觀的圖形化編程工具,讓用戶通過控制海龜在屏幕上的移動來繪制各種形狀和圖案,在接下來的文章中,我將通過一個生動的例子——繪制一幅櫻花樹圖畫——來深入探討turtle模塊的實用性,需要的朋友可以參考下
2024-04-04