实用脚本！利用 Python 对 PDF 进行加密、解密操作，代码拿走就用！

相关导读

????01｜Python批量合并PDF

????02｜PDF合并、拆分、水印、加密

????03｜Python提取PDF表格

????04｜Python提取PDF指定内容

大家好，我是早起。

本文将分享如何利用 Python 对 PDF 进行加密和解密操作，主要利用到之前多次介绍过的PyPDF2 模块。

PDF 加密

在之前的文章PDF合并、拆分、水印、加密中简单提到过加密一个 PDF 文件的方法，我们先拿自己随意的一个PDF 文件试一下：

from PyPDF2 import PdfFileWriter, PdfFileReader
path = r'C:\xxx' # 这里填写目标 PDF 所在的路径

pdf_reader = PdfFileReader(path + r'\test.pdf')
pdf_writer = PdfFileWriter()

for page in range(pdf_reader.getNumPages()):
    pdf_writer.addPage(pdf_reader.getPage(page))
pdf_writer.encrypt('a123') # 设置密码为 a123
with open(path + r'\test.pdf', 'wb') as out:
    pdf_writer.write(out)

上面代码的运行逻辑是：实例化一个 PDF 写入器和读取器，读取器读取完目标 PDF 文件后，一页一页交给写入器，然后对写入器设置密码并输出。看一下运行结果：

可见这个 PDF 文件成功设置上了密码，如果有多个PDF需要加密，可以写一个简单的循环利用上述脚本批量处理，此处不再展开说明。

PDF 已知密码解密

如果知道密码的情况下，想直接取消 PDF 的加密，可以用 .decrypt，解密的过程需要读取器和写入器共同配合。但区别于加密 .encrypt，.decrypt 是针对读取器进行解密的，而不是写入器

from PyPDF2 import PdfFileWriter, PdfFileReader
path = r'C:\xxx'

pdf_reader = PdfFileReader(path + r'\test.pdf')
pdf_reader.decrypt('a123') #
pdf_writer = PdfFileWriter()

for page in range(pdf_reader.getNumPages()):
    pdf_writer.addPage(pdf_reader.getPage(page))
with open(path + r'\test.pdf', 'wb') as out:
    pdf_writer.write(out)

上面代码的运行逻辑是：首先用读取器读取加密文件，然后直接在读取器上使用 .decrypt 进行解密，逐页传到刚实例化的写入器后统一输出，就完成了解密

PDF 暴力解密

暴力破解，其实就是通过手上已经有的密码库，或者完全通过数字、字母、符号的穷举，将可能的密码都放进去，逐个尝试直到成功。

「目前暴力破解只适用于已知密码位数少，由简单的数字、字母构成」

假设今天要破解的密码就是 a123，已知密码 4 位且由数字和小写字母组成。破解又分为两种情况：

1. 手上有密码本

第一种情况，知道大概密码，可以整理成一个 password.txt ，保证可能的密码一定在其中：

首先读取 .txt 文件获取其中所有密码（示例文件只有 9 个密码）：

passw = []
path = r'C:\Scientific Research\Python'
file = open(path + r'\password.txt')
for line in file.readlines():
    passw.append(line.strip())
print(passw)
file.close()

然后就可以用密码本的密码做暴力破解：

from PyPDF2 import PdfFileReader

passw = []
path = r'C:\xxx'
file = open(path + r'\password.txt')
for line in file.readlines():
    passw.append(line.strip())
file.close()

path = r'C:\xxx'
pdf_reader = PdfFileReader(path + r'\test).pdf')

for i in passw:
    if pdf_reader.decrypt(i):
        print(f'破解成功，密码为{i}')
    else:
        print(f'破解不成功，密码{i}错误')

2. 手术无密码本，完全穷举

首先介绍 itertools，可以构建含需要组合的生成器供后续迭代：

import itertools
mylist = ("".join(x) for x in itertools.product("0123456789abcdef", repeat=4))
print(next(mylist))

利用这个方法就可以用 while 循环尝试暴力破解（为了减少时间，生成器用"abc123"生成组合）：

import itertools
from PyPDF2 import PdfFileReader

mylist = ("".join(x) for x in itertools.product("123abc", repeat=4))
path = r'C:\xxx'
pdf_reader = PdfFileReader(path + r'\test.pdf')

while True:
    i = next(mylist)
    if pdf_reader.decrypt(i):
        print(f'破解成功，密码为{i}')
        break
    else:
        print(f'破解不成功，密码{i}错误')

可以看到最后也成功破解了，但需要注意：但即使是这么简单的组合也耗费了一点时间，如果复杂的密码破解难度和破解时间就指数级增长了，暴力的穷举法就不一定适用，可以根据需求具体调整规则，缩小范围再进行尝试。

-END-

如果你希望学习更多Python办公自动化案例，可以点击下方图片查看！

文末赠送 5 本《Python数据分析从小白到专家》主要内容涵盖Python语法及数据分析方法。第1章主要介绍数据分析的概念，第2章主要做一些准备工作，手把手带读者搭建Python环境，第3章介绍Python编程基础。第4章到第7章介绍使用Python进行简单数据分析的基础库，并介绍使用正则表达式处理数据的方法。第8章到第13章属于进阶内容，但也是Python数据分析的基础，结合机器学习介绍一些常见的用于数据分析的机器学习算法及常用的数学模型。

点击查看书籍详情????

赠书规则：点赞本文 + 给本文留言，然后集赞数排名前 3 + 2 名优质留言的朋友，可获得包邮赠书。

开奖时间：5 月 20 日 20:00

注意：没有意义的留言将不会被选中，无需发送截图给我，开奖时以我这本截图为准，如若中奖我会回复留言，中奖者24小时内，必须添加小助理微信（sshs321）领取，逾期不候！