python 加密pdf文件

在Python中加密PDF文件通常需要借助第三方库，如PyPDF2和reportlab等。以下是基本步骤：

1. **安装依赖库**：
首先需要安装`PyPDF2`库来处理PDF，以及`Crypto`或`cryptography`库用于加密。可以使用pip安装：

   pip install PyPDF2 cryptography

2. **创建加密对象**：
导入所需的模块并设置密码和加密选项：

   from PyPDF2 import PdfFileWriter, PdfFileReader
   from Crypto.Cipher import AES
   from Crypto.Util.Padding import pad, unpad
   from Crypto.Random import get_random_bytes

3. **读取PDF**：
使用PdfFileReader打开PDF，如果需要合并文档，则创建一个新的PdfFileWriter对象。

   reader = PdfFileReader(open('input.pdf', 'rb'))
   writer = PdfFileWriter()

4. **加密内容**：
- 创建AES加密器，并选择填充模式（比如PKCS7）。
- 获取随机数据作为初始化向量（IV），用于保证每次加密的数据都不一样。

   key = get_random_bytes(16)  # 16字节的AES密钥
   cipher = AES.new(key, AES.MODE_CBC)
   iv = cipher.iv

5. **加密每一页**：
对PDF的每个页面进行加密，将文本和图像转换为二进制，然后加密后再写回新的PDF。

   for page_num in range(reader.numPages):
       page = reader.getPage(page_num)
       encrypted_text = cipher.encrypt(page.extractText())
       decrypted_text = unpad(ciphertext=encrypted_text, mode=AES.MODE_CBC, length=16)
       page.mergePage(decrypted_text)
       writer.addPage(page)

6. **写入加密后的PDF**：
将加密后的PDF保存到新的文件中，并指定使用合适的压缩级别。

   with open('output.pdf', 'wb') as out_file:
       writer.write(out_file)