python使用pypdf2實現(xiàn)pdf文檔解密

更新時間：2023年12月25日 12:09:26 作者：不當(dāng)王多魚不改名

利用pypdf2完成pdf的解密,這里的事例是python3環(huán)境下的,當(dāng)然python2下也可以運行,只需要修改名稱即可,文中通過代碼示例給大家介紹的非常詳細(xì),需要的朋友可以參考下

前言

利用pypdf2完成pdf的解密，這里的事例是python3環(huán)境下的，當(dāng)然python2下也可以運行，只需要修改名稱即可，在代碼中有修改的地方，筆者也會指出，看完這篇你不會解密pdf的話?。?！那我就再寫一篇。

一、導(dǎo)入庫函數(shù)

這里用到的模塊有pypdf2、pycryptodome，如果沒有模塊的需要在對應(yīng)的虛擬環(huán)境中執(zhí)行以下的pip安裝命令

pip install pypdf2
pip install pycryptodome

二、主要函數(shù)解讀

1.引入庫

如果是python2的在引入函數(shù)方法的時候需要改一下，如果是python3的就直接復(fù)制下面代碼就好

from PyPDF2 import PdfReader #pdf的讀取方法
from PyPDF2 import PdfWriter #pdf的寫入方法
"""
如果是python2的將上面的PdfReader和PdfWriter改為
PdfFileReader和PDFfileWriter即可
"""
from Crypto.Cipher import AES #高加密的方法，要引入不然會報錯

2.自定義pdf讀取函數(shù)

python2需要修改的地方在代碼中已經(jīng)標(biāo)注，python3的直接復(fù)制下面代碼即可

def get_reader(filename, password): #讀取pdf的方法（自定義函數(shù)）
    try:
        old_file = open(filename, 'rb')
        print('解密開始...')
    except Exception as err:
        return print('文件打開失??！' + str(err))

    """如果是python2將PdfReader改為PdfFileReader"""
    pdf_reader = PdfReader(old_file, strict=False) #讀取pdf文件

    """如果是python2將is_encrypted改為isEncrypted"""
    if pdf_reader.is_encrypted: #解密操作（以下操作是自適應(yīng)，不會展示在終端中）
        if password is None:
            return print('文件被加密，需要密碼！--{}'.format(filename))
        else:
            if pdf_reader.decrypt(password) != 1:
                return print('密碼不正確！--{}'.format(filename))
    elif old_file in locals():
        old_file.close() #如果pdf文件已經(jīng)在本地了就關(guān)閉
    return pdf_reader #返回讀出pdf的值

3.自定義pdf生成函數(shù)

將加密的文件及逆行解密，并生成一個無需密碼pdf文件，下面是生成函數(shù)deception_pdf的基本參數(shù)

filename: 原先加密的pdf文件名稱

password: 原文件對應(yīng)的密碼

decrypted_filename: 解密之后的文件名

def deception_pdf(filename, password, decrypted_filename=None): #生成新pdf的方法（自定義函數(shù)）
    print('正在生成解密...')
    #得到傳入的文件名和密碼(這里不用做修改)
    pdf_reader = get_reader(filename, password) 
    if pdf_reader is None:
        return print("無內(nèi)容讀取")

    """如果是python2將is_encrypted改為isEncrypted"""
    elif not pdf_reader.is_encrypted:
        return print('文件沒有被加密，無需操作')

    """如果是python2將PdfWriter改為PdfFileWriter"""
    pdf_writer = PdfWriter() #寫pdf（記錄pdf內(nèi)容）

    """如果是python2將append_pages_from_reader改為appendPagesFromReader"""
    pdf_writer.append_pages_from_reader(pdf_reader)

    if decrypted_filename is None: #創(chuàng)建解密后的pdf文件和展示文件的路徑
        decrypted_filename = "".join(filename.split('.')[:-1]) + '_' + '已解密' + '.pdf'
        print("解密文件已生成:{}".format(decrypted_filename))
    # 寫入新文件
    pdf_writer.write(open(decrypted_filename, 'wb'))

三、完整源代碼

from PyPDF2 import PdfReader #pdf的讀取方法
from PyPDF2 import PdfWriter #pdf的寫入方法
from Crypto.Cipher import AES #高加密的方法，要引入不然會報錯

def get_reader(filename, password): #讀取pdf的方法（自定義函數(shù)）
    try:
        old_file = open(filename, 'rb')
        print('解密開始...')
    except Exception as err:
        return print('文件打開失??！' + str(err))

    #如果是python2將PdfReader改為PdfFileReader
    pdf_reader = PdfReader(old_file, strict=False) #讀取pdf文件

    # 如果是python2將is_encrypted改為isEncrypted
    if pdf_reader.is_encrypted: #解密操作（以下操作是自適應(yīng)，不會展示在終端中）
        if password is None:
            return print('文件被加密，需要密碼！--{}'.format(filename))
        else:
            if pdf_reader.decrypt(password) != 1:
                return print('密碼不正確！--{}'.format(filename))
    elif old_file in locals():
        old_file.close() #如果pdf文件已經(jīng)在本地了就關(guān)閉
    return pdf_reader #返回讀出pdf的值
    
def deception_pdf(filename, password, decrypted_filename=None): #生成新pdf的方法（自定義函數(shù)）
    print('正在生成解密...')
    pdf_reader = get_reader(filename, password) #得到傳入的文件名，和密碼（如果密碼沒有可以不填）
    if pdf_reader is None:
        return print("無內(nèi)容讀取")

    # 如果是python2將is_encrypted改為isEncrypted
    elif not pdf_reader.is_encrypted:
        return print('文件沒有被加密，無需操作')

    # 如果是python2將PdfWriter改為PdfFileWriter
    pdf_writer = PdfWriter() #寫pdf（記錄pdf內(nèi)容）

    #如果是python2將append_pages_from_reader改為appendPagesFromReader
    pdf_writer.append_pages_from_reader(pdf_reader)

    if decrypted_filename is None: #創(chuàng)建解密后的pdf文件和展示文件的路徑
        decrypted_filename = "".join(filename.split('.')[:-1]) + '_' + '已解密' + '.pdf'
        print("解密文件已生成:{}".format(decrypted_filename))
    # 寫入新文件
    pdf_writer.write(open(decrypted_filename, 'wb'))

if __name__ == '__main__':
    # 逗號前面的為需要解密的pdf文件路徑，后面的''里面為密碼，如果不知道就用不填
    deception_pdf(r'E:\考研\(zhòng)301數(shù)學(xué)一\數(shù)學(xué)書\數(shù)學(xué)書\2024kaoyanmath1.pdf', '')

使用方法：

我們要先得到需要解密的文件路徑，如下看到我們的文件路徑為（?E:\考研\(zhòng)301數(shù)學(xué)一\數(shù)學(xué)書\數(shù)學(xué)書\2024kaoyanmath1.pdf）復(fù)制后，再黏貼到代碼中的指定位置即可

在這里插入圖片描述