使用Python加密和解密PDF文件

更新時間：2025年03月03日 08:53:09 作者：蠟筆小新星

在日常工作和生活中,保護PDF文件的隱私和安全至關重要,Python提供了一些強大的庫,使得加密和解密PDF文件變得相對簡單,本文將詳細介紹如何使用PyPDF2庫來加密和解密PDF文件,需要的朋友可以參考下

一、安裝PyPDF2庫

在開始之前，你需要確保已經(jīng)安裝了PyPDF2庫。如果還沒有安裝，可以使用以下命令通過pip進行安裝：

pip install PyPDF2

二、加密PDF文件

加密PDF文件的過程涉及讀取原始PDF文件，將其內容復制到新的PDF對象中，然后應用加密。以下是一個完整的示例代碼：

import PyPDF2

def encrypt_pdf(input_file, output_file, password):
    # 創(chuàng)建PDF閱讀器對象，讀取輸入文件
    reader = PyPDF2.PdfReader(input_file)
    # 創(chuàng)建PDF寫入對象
    writer = PyPDF2.PdfWriter()

    # 將所有頁面從閱讀器對象添加到寫入對象
    for page in reader.pages:
        writer.add_page(page)

    # 打開輸出文件并寫入加密后的PDF內容
    with open(output_file, 'wb') as f:
        writer.encrypt(password)  # 應用加密
        writer.write(f)

# 使用示例
input_file = 'example.pdf'
output_file = 'encrypted_example.pdf'
password = 'your_password'

encrypt_pdf(input_file, output_file, password)

在上述代碼中，我們定義了一個名為encrypt_pdf的函數(shù)，它接受三個參數(shù)：輸入文件名、輸出文件名和密碼。函數(shù)內部，我們首先創(chuàng)建一個PdfReader對象來讀取原始PDF文件，然后創(chuàng)建一個PdfWriter對象來準備寫入新的加密PDF文件。通過遍歷PdfReader對象的頁面，我們將它們添加到PdfWriter對象中。最后，使用writer.encrypt(password)方法應用加密，并將加密后的內容寫入指定的輸出文件。

三、解密PDF文件

解密PDF文件的過程與加密類似，但多了一步解密操作。以下是解密PDF文件的示例代碼：

import PyPDF2

def decrypt_pdf(input_pdf, output_pdf, password):
    # 打開PDF文件并創(chuàng)建閱讀器對象
    with open(input_pdf, 'rb') as file:
        pdf_reader = PyPDF2.PdfReader(file)

        # 解密PDF文件
        pdf_reader.decrypt(password)

        # 創(chuàng)建PDF寫入對象
        pdf_writer = PyPDF2.PdfWriter()

        # 將所有頁面從閱讀器對象添加到寫入對象
        for page_num in range(len(pdf_reader.pages)):
            pdf_writer.add_page(pdf_reader.pages[page_num])

        # 打開輸出文件并寫入解密后的PDF內容
        with open(output_pdf, 'wb') as output:
            pdf_writer.write(output)

# 使用示例
decrypt_pdf('encrypted_example.pdf', 'decrypted_example.pdf', 'your_password')

在這個示例中，decrypt_pdf函數(shù)接受三個參數(shù)：輸入加密PDF文件名、輸出解密PDF文件名和密碼。函數(shù)內部，我們首先使用PdfReader對象讀取加密的PDF文件，并調用decrypt(password)方法進行解密。然后，創(chuàng)建一個PdfWriter對象，并將解密后的頁面添加到該對象中。最后，將解密后的內容寫入指定的輸出文件。