python - 加密的 PDF 不会写入磁盘

Question

我有一些加密 pdf 的代码，由于某种原因，一旦我加密 pdf，书写就会挂起。如果我注释掉 pdf_writer.encrypt(password) 就没有问题，并且内容写入正确。我尝试以最高权限运行脚本并修改文件夹选项，但也许我错过了一些简单的东西。

with open(filename, 'rb') as pdf_file:
    pdf_reader = PyPDF2.PdfFileReader(pdf_file)
    pdf_writer = PyPDF2.PdfFileWriter()

    for page_number in range(pdf_reader.numPages):
        pdf_writer.addPage(pdf_reader.getPage(page_number))

    pdf_writer.encrypt(password)
    filename_encrypted = filename.parents[0] / f "{filename.stem}_encrypted.pdf"

    with open(filename_encrypted, 'wb') as pdf_file_encrypted:
        pdf_writer.write(pdf_file_encrypted)

任何帮助将不胜感激。

Answer 1

我不明白为什么 PyPDF2 不起作用，所以我只是尝试了另一个名为 pikepdf 的模块，一切正常。无论如何，这实际上看起来是一个比 PyPDF2 更好的模块。

import pikepdf


filename = pathlib.Path(r"path\to\pdf\test.pdf")
password = 'password'

with Pdf.open(filename) as pdf:
    filename_encrypted = filename.parents[0] / f"{filename.stem}_encrypted.pdf"
    pdf.save(filename_encrypted, encryption=Encryption(user=password, owner=password))