pycryptodome 加密PDF

对于使用 pycryptodome 进行 PDF 加密的方法，可以先使用 PyPDF2 库打开 PDF 文件，然后使用 pycryptodome 进行加密，并将加密后的结果写回到文件中。以下是一个示例代码：

from Crypto.PublicKey import RSA
from Crypto.Random import get_random_bytes
from Crypto.Cipher import AES, PKCS1_OAEP
import PyPDF2

# 加载 PDF 文件
pdf_path = "path/to/pdf/file"
pdf = open(pdf_path, "rb")
pdf_reader = PyPDF2.PdfFileReader(pdf)
pdf_writer = PyPDF2.PdfFileWriter()

# 加载加密密钥
key = get_random_bytes(16)
rsa_key = RSA.generate(2048)
cipher_rsa = PKCS1_OAEP.new(rsa_key)

# 加密 PDF 文件
for i in range(pdf_reader.getNumPages()):
    page = pdf_reader.getPage(i)
    pdf_writer.addPage(page)
    
pdf_writer.encrypt(key)

# 将密钥加密并写入 PDF 文件
with open(pdf_path, "wb") as pdf_output:
    cipher_aes = AES.new(key, AES.MODE_EAX)
    ciphertext_aes, tag = cipher_aes.encrypt_and_digest(key)
    public_key = rsa_key.publickey().export_key()
    cipher_rsa = PKCS1_OAEP.new(RSA.import_key(public_key))
    pdf_output.write(cipher_rsa.encrypt(ciphertext_aes))
    pdf_writer.write(pdf_output)

在以上代码中，我们使用 `get_random_bytes` 函数生成了一个长度为 16 的随机字节数组作为加密密钥。然后，我们使用 `RSA.generate` 函数生成一个 2048 位的 RSA 密钥对，并使用公钥加密密钥。接下来，我们使用 `pdf_writer.encrypt` 函数将 PDF 文件加密。最后，我们将加密后的密钥使用公钥进行加密，并将加密后的密钥一起写入到 PDF 文件中。注意，以上代码中的密钥长度是 16 字节，这只是举个例子，实际使用时应该根据需要调整密钥长度。