python加解密PDF文件

#解密
import PyPDF2

def decrypt_pdf(input_pdf, output_pdf, password):
    # 打开PDF文件
    with open(input_pdf, 'rb') as file:
        # 创建PDF写入对象
        pdf_writer = PyPDF2.PdfWriter()

        # 创建PDF阅读器对象
        pdf_reader = PyPDF2.PdfReader(file)

        # 解密PDF文件
        pdf_reader.decrypt(password)

        # 遍历每一页并添加到写入对象中
        for page_num in range(len(pdf_reader.pages)):
            pdf_writer.add_page(pdf_reader.pages[page_num])

        # 写入解密后的PDF文件
        with open(output_pdf, 'wb') as output:
            pdf_writer.write(output)

# 使用示例

decrypt_pdf('abc2.pdf', 'output_decrypted2.pdf', 'password')

#加密
import PyPDF2

def encrypt_pdf(input_pdf, output_pdf, password):
    # 打开PDF文件
    with open(input_pdf, 'rb') as file:
        # 创建PDF写入对象
        pdf_writer = PyPDF2.PdfWriter()

        # 创建PDF阅读器对象
        pdf_reader = PyPDF2.PdfReader(file)

        # 遍历每一页并添加到写入对象中
        for page_num in range(len(pdf_reader.pages)):
            pdf_writer.add_page(pdf_reader.pages[page_num])

        # 设置密码
        pdf_writer.encrypt(password)

        # 写入加密后的PDF文件
        with open(output_pdf, 'wb') as output:
            pdf_writer.write(output)

# 使用示例
encrypt_pdf('input.pdf', 'output_encrypted.pdf', 'my_password')