使用Python實現(xiàn)對PDF文件進(jìn)行密碼保護(hù)

更新時間：2024年01月29日 08:54:18 作者：Sitin濤哥

這篇文章主要為大家詳細(xì)了如何使用Python來實現(xiàn)PDF文件的密碼保護(hù),以確保只有授權(quán)的用戶可以訪問文檔,文中的示例代碼簡潔易懂,有需要的小伙伴可以參考一下

PDF（Portable Document Format）是一種流行的文檔格式，用于分享和存儲文檔。有時，為了保護(hù)文檔的安全性，我們需要在PDF文件上設(shè)置密碼。本文將介紹如何使用Python來實現(xiàn)PDF文件的密碼保護(hù)，以確保只有授權(quán)的用戶可以訪問文檔。將使用PyPDF2庫，這是一個用于操作PDF文件的強(qiáng)大工具。

安裝PyPDF2庫

首先，需要安裝PyPDF2庫?？梢允褂靡韵旅钍褂胮ip安裝PyPDF2：

pip install PyPDF2

密碼保護(hù)PDF文件

假設(shè)有一個名為"example.pdf"的PDF文件，希望將其設(shè)置為需要密碼才能打開。以下是實現(xiàn)這一目標(biāo)的步驟和示例代碼：

步驟1：導(dǎo)入PyPDF2庫

首先，導(dǎo)入PyPDF2庫：

import PyPDF2

步驟2：打開PDF文件并創(chuàng)建一個新的PDF寫入對象

需要打開要保護(hù)的PDF文件并創(chuàng)建一個新的PDF寫入對象。同時，可以指定一個輸出文件，其中將保存帶有密碼保護(hù)的PDF。

# 打開要保護(hù)的PDF文件
with open('example.pdf', 'rb') as pdf_file:
    # 創(chuàng)建PDF閱讀對象
    pdf_reader = PyPDF2.PdfFileReader(pdf_file)
    
    # 創(chuàng)建PDF寫入對象
    pdf_writer = PyPDF2.PdfFileWriter()

步驟3：設(shè)置密碼并添加PDF頁面

接下來，設(shè)置密碼并將PDF頁面添加到新的PDF寫入對象中。在這個示例中，將密碼設(shè)置為"mypassword"，但可以將其更改為喜歡的密碼。

    # 設(shè)置密碼
    pdf_writer.addPage(pdf_reader.getPage(0))
    pdf_writer.encrypt('mypassword')

步驟4：保存帶有密碼保護(hù)的PDF文件

最后，將新的PDF寫入對象保存到一個輸出文件中，以創(chuàng)建帶有密碼保護(hù)的PDF文件。

    # 保存帶有密碼保護(hù)的PDF文件
    with open('protected_example.pdf', 'wb') as protected_pdf_file:
        pdf_writer.write(protected_pdf_file)

現(xiàn)在，"protected_example.pdf"就是一個帶有密碼保護(hù)的PDF文件。只有知道密碼的用戶才能打開它。

使用密碼打開PDF文件

要打開帶有密碼保護(hù)的PDF文件，用戶需要提供正確的密碼。以下是如何使用Python和PyPDF2庫來打開受保護(hù)的PDF文件的示例代碼：

import PyPDF2

# 輸入密碼
password = input("請輸入密碼: ")

# 打開受保護(hù)的PDF文件
with open('protected_example.pdf', 'rb') as protected_pdf_file:
    # 創(chuàng)建PDF閱讀對象
    pdf_reader = PyPDF2.PdfFileReader(protected_pdf_file)
    
    # 解鎖PDF文件
    if pdf_reader.decrypt(password):
        # 打印PDF文件的總頁數(shù)
        print(f"總頁數(shù): {pdf_reader.numPages}")
    else:
        print("密碼錯誤，無法打開文件。")

在上述示例中，要求用戶輸入密碼，然后嘗試打開受保護(hù)的PDF文件。如果密碼正確，就可以訪問PDF文件的內(nèi)容，否則將顯示密碼錯誤的消息。

修改密碼

有時候，可能需要修改現(xiàn)有PDF文件的密碼。

以下是一個示例代碼，演示如何修改PDF文件的密碼：

import PyPDF2

# 輸入原始密碼和新密碼
original_password = input("請輸入原始密碼: ")
new_password = input("請輸入新密碼（按回車跳過）: ")

# 打開原始受保護(hù)的PDF文件
with open('protected_example.pdf', 'rb') as protected_pdf_file:
    # 創(chuàng)建PDF閱讀對象
    pdf_reader = PyPDF2.PdfFileReader(protected_pdf_file)
    
    # 解鎖PDF文件
    if pdf_reader.decrypt(original_password):
        # 創(chuàng)建PDF寫入對象
        pdf_writer = PyPDF2.PdfFileWriter()
        
        # 復(fù)制頁面到新文件
        for page_num in range(pdf_reader.numPages):
            page = pdf_reader.getPage(page_num)
            pdf_writer.addPage(page)
        
        # 設(shè)置新密碼（如果提供了新密碼）
        if new_password:
            pdf_writer.encrypt(new_password)
        
        # 保存新的PDF文件
        with open('modified_example.pdf', 'wb') as modified_pdf_file:
            pdf_writer.write(modified_pdf_file)
        
        print("PDF文件密碼已修改并保存為'modified_example.pdf'。")
    else:
        print("原始密碼錯誤，無法修改文件密碼。")

這個示例首先要求用戶輸入原始密碼和（可選的）新密碼。然后，它嘗試解鎖原始PDF文件，創(chuàng)建一個新的PDF寫入對象，并復(fù)制頁面到新文件。如果提供了新密碼，它將設(shè)置新密碼并保存新的PDF文件。

刪除密碼

如果有一個受密碼保護(hù)的PDF文件，而不再需要密碼保護(hù)，可以使用Python來刪除密碼。以下是一個示例代碼：

import PyPDF2

# 輸入密碼
password = input("請輸入密碼: ")

# 打開受保護(hù)的PDF文件
with open('protected_example.pdf', 'rb') as protected_pdf_file:
    # 創(chuàng)建PDF閱讀對象
    pdf_reader = PyPDF2.PdfFileReader(protected_pdf_file)
    
    # 解鎖PDF文件
    if pdf_reader.decrypt(password):
        # 創(chuàng)建PDF寫入對象
        pdf_writer = PyPDF2.PdfFileWriter()
        
        # 復(fù)制頁面到新文件
        for page_num in range(pdf_reader.numPages):
            page = pdf_reader.getPage(page_num)
            pdf_writer.addPage(page)
        
        # 保存新的PDF文件（不設(shè)置密碼）
        with open('unprotected_example.pdf', 'wb') as unprotected_pdf_file:
            pdf_writer.write(unprotected_pdf_file)
        
        print("密碼已刪除，PDF文件保存為'unprotected_example.pdf'。")
    else:
        print("密碼錯誤，無法刪除密碼。")

這個示例代碼首先要求用戶輸入密碼，然后嘗試解鎖受保護(hù)的PDF文件。如果密碼正確，它將創(chuàng)建一個新的PDF寫入對象，并復(fù)制頁面到新文件，但不設(shè)置密碼。最終，它將保存一個沒有密碼的新PDF文件。

批量處理多個PDF文件

如果需要批量處理多個PDF文件，并對它們進(jìn)行密碼保護(hù)、密碼修改或密碼刪除，可以編寫一個腳本來自動處理它們。

以下是一個示例代碼，演示如何批量處理多個PDF文件：

import PyPDF2
import os

# 指定PDF文件所在文件夾路徑
pdf_folder = 'pdf_files'

# 輸入密碼（或原始密碼和新密碼）
password = input("請輸入密碼（或原始密碼和新密碼，以逗號分隔）: ")

# 將輸入的密碼拆分為原始密碼和新密碼
passwords = password.split(',')

# 獲取PDF文件夾中的所有PDF文件
pdf_files = [f for f in os.listdir(pdf_folder) if f.endswith('.pdf')]

for pdf_file in pdf_files:
    pdf_path = os.path.join(pdf_folder, pdf_file)
    
    # 打開PDF文件
    with open(pdf_path, 'rb') as pdf_file:
        # 創(chuàng)建PDF閱讀對象
        pdf_reader = PyPDF2.PdfFileReader(pdf_file)
        
        # 解鎖PDF文件
        if pdf_reader.decrypt(passwords[0]):
            # 創(chuàng)建PDF寫入對象
            pdf_writer = PyPDF2.PdfFileWriter()
            
            # 復(fù)制頁面到新文件
            for page_num in range(pdf_reader.numPages):
                page = pdf_reader.getPage(page_num)
                pdf_writer.addPage(page)
            
            # 設(shè)置新密碼（如果提供了新密碼）
            if len(passwords) > 1:
                pdf_writer.encrypt(passwords[1])
            
            # 保存新的PDF文件
            output_path = os.path.join(pdf_folder, f'unlocked_{pdf_file}')
            with open(output_path, 'wb') as output_pdf_file:
                pdf_writer.write(output_pdf_file)
            
            print(f"已處理文件：{pdf_file}，保存為：{output_path}")
        else:
            print(f"密碼錯誤，無法處理文件：{pdf_file}")

這個示例代碼首先要求用戶輸入密碼（或原始密碼和新密碼，以逗號分隔）。然后，它獲取指定文件夾中的所有PDF文件，并循環(huán)處理每個文件。對于每個文件，它嘗試解鎖PDF文件，創(chuàng)建一個新的PDF寫入對象，并復(fù)制頁面到新文件。如果提供了新密碼，它將設(shè)置新密碼并保存新的PDF文件。

總結(jié)

本文介紹了如何使用Python和PyPDF2庫來實現(xiàn)PDF文件的密碼保護(hù)、密碼修改和密碼刪除。這些操作可以更好地管理和保護(hù)PDF文件，確保只有授權(quán)的用戶可以訪問它們。無論是單個文件還是批量處理多個文件，Python提供了強(qiáng)大的工具來處理PDF文件的密碼保護(hù)。希望本文能幫助大家更好地利用Python來處理PDF文件的安全性。

到此這篇關(guān)于使用Python實現(xiàn)對PDF文件進(jìn)行密碼保護(hù)的文章就介紹到這了,更多相關(guān)Python PDF密碼保護(hù)內(nèi)容請搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家！

您可能感興趣的文章: