快捷導(dǎo)航

使用Python加密和解密PDF文件

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

在日常工作和生活中,保護(hù)PDF文件的隱私和安全至關(guān)重要,Python提供了一些強(qiáng)大的庫(kù),使得加密和解密PDF文件變得相對(duì)簡(jiǎn)單,本文將詳細(xì)介紹如何使用PyPDF2庫(kù)來(lái)加密和解密PDF文件,需要的朋友可以參考下

一、安裝PyPDF2庫(kù)

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

pip install PyPDF2

二、加密PDF文件

加密PDF文件的過(guò)程涉及讀取原始PDF文件，將其內(nèi)容復(fù)制到新的PDF對(duì)象中，然后應(yīng)用加密。以下是一個(gè)完整的示例代碼：

import PyPDF2

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

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

    # 打開(kāi)輸出文件并寫(xiě)入加密后的PDF內(nèi)容
    with open(output_file, 'wb') as f:
        writer.encrypt(password)  # 應(yīng)用加密
        writer.write(f)

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

encrypt_pdf(input_file, output_file, password)

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

三、解密PDF文件

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

import PyPDF2

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

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

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

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

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

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

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