使用python将pdf取消密码保护-密码轻松移除-轻松编辑

# coding:utf-8
import os
import PyPDF2

def get_reader(filename, password):
    try:
        old_file = open(filename, 'rb')
    except Exception as err:
        print('文件打开失败!' + str(err))
        return None
    # 创建读实例
    pdf_reader = PyPDF2.PdfReader(old_file, strict=False)
    # 解密操作
    if pdf_reader.is_encrypted:
        if password is None:
            print('%s文件被加密,需要密码!' % filename)
            return None
        else:
            if pdf_reader.decrypt(password) != 1:
                print('%s密码不正确!' % filename)
                return None
    if old_file in locals():
        old_file.close()
    return pdf_reader


def decrypt_pdf(filename, password, decrypted_filename=None):
    """
       将加密的文件及逆行解密,并生成一个无需密码pdf文件
       :param filename: 原先加密的pdf文件
       :param password: 对应的密码
       :param decrypted_filename: 解密之后的文件名
       :return:
       """
    # 生成一个Reader和Writer
    print('run  文件解密')
    pdf_reader = get_reader(filename, password)
    if pdf_reader is None:
        return
    if not pdf_reader.is_encrypted:
        print('文件没有被加密,无需操作!')
        return
    pdf_writer = PyPDF2.PdfWriter()
    pdf_writer.append_pages_from_reader(pdf_reader)
    if decrypted_filename is None:
        decrypted_filename = "".join(filename.split('.')[:-1]) + '_' + 'decrypted' + '.pdf'
    print('文件解密完成!')
    # 写入新文件
    pdf_writer.write(open(decrypted_filename, 'wb'))


decrypt_pdf(r'xxx.pdf', '666')

你可能感兴趣的:(python,pdf,开发语言)