使用 PyPDF2 加密和解密 PDF

PyPDF2 是一个 Python 库，可以用于处理 PDF 文件。它支持 PDF 文件的读取、写入和修改，包括加密和解密 PDF 文件。在本文中，我们将介绍如何使用 PyPDF2 来加密和解密 PDF 文件。

环境准备

在使用 PyPDF2 之前，我们需要先安装它。可以使用 pip 命令来安装 PyPDF2：

pip install PyPDF2

加密 PDF 文件

使用 PyPDF2 加密 PDF 文件非常简单。我们只需要创建一个 PdfFileWriter 对象，然后设置密码即可。以下是一个示例代码：

import PyPDF2

# 打开一个 PDF 文件
file = open('example.pdf', 'rb')
pdf_reader = PyPDF2.PdfFileReader(file)

# 创建一个 PdfFileWriter 对象
pdf_writer = PyPDF2.PdfFileWriter()

# 将 PDF 文件的每个页面都写入 PdfFileWriter 对象中
for page_num in range(pdf_reader.numPages):
    pdf_writer.addPage(pdf_reader.getPage(page_num))

# 设置密码
pdf_writer.encrypt('password')

# 将加密后的 PDF 文件保存到磁盘上
output = open('encrypted.pdf', 'wb')
pdf_writer.write(output)
output.close()
file.close()

上面的代码中，我们打开一个 PDF 文件，然后将每个页面写入一个 PdfFileWriter 对象中，最后设置了一个密码，将加密后的 PDF 文件保存到磁盘上。

解密 PDF 文件

使用 PyPDF2 解密 PDF 文件同样很简单。我们只需要打开一个加密的 PDF 文件，然后输入正确的密码即可。以下是一个示例代码：

import PyPDF2

# 打开一个加密的 PDF 文件
file = open('encrypted.pdf', 'rb')
pdf_reader = PyPDF2.PdfFileReader(file)

# 检查 PDF 文件是否加密
if pdf_reader.isEncrypted:
    # 输入密码
    password = input('Enter password:')
    pdf_reader.decrypt(password)

# 获取 PDF 文件的第一页，并将其保存到磁盘上
page = pdf_reader.getPage(0)
output = open('decrypted.pdf', 'wb')
pdf_writer = PyPDF2.PdfFileWriter()
pdf_writer.addPage(page)
pdf_writer.write(output)
output.close()
file.close()

上面的代码中，我们打开了一个加密的 PDF 文件，然后检查该文件是否加密。如果已经加密，就输入密码来解密。接着，我们获取了 PDF 文件的第一页，将其保存到磁盘上，并且写入了一个新的 PdfFileWriter 对象中。

总结

PyPDF2 是一个强大的 Python 库，可以用于处理 PDF 文件。在本文中，我们介绍了如何使用 PyPDF2 来加密和解密 PDF 文件。我们希望这篇文章能够帮助您更好地处理 PDF 文件。