# coding:utf-8
import os
from PyPDF2 import PdfReader
from PyPDF2 import PdfWriter
def decrypt_pdf(filename, password, decrypted_filename=None):
"""
将加密的文件及逆行解密, 并生成一个无需密码pdf文件
:param filename: 原先加密的pdf文件
:param password: 对应的密码
:param decrypted_filename: 解密之后的文件名
:return:
"""
# 生成一个Reader和Writer
print('run jiemi')
pdf_reader = get_reader(filename, password)
if pdf_reader is None:
return
if not pdf_reader.is_encrypted:
print('文件没有被加密, 无需操作!')
return
pdf_writer = PdfWriter()
pdf_writer.append_pages_from_reader(pdf_reader)
if decrypted_filename is None:
decrypted_filename = "".join(filename.split('.')[:-1]) + '_' + 'decrypted' + '.pdf'
# 写入新文件
pdf_writer.write(open(decrypted_filename, 'wb'))
def get_reader(filename, password):
try:
old_file = open(filename, 'rb')
print('run jiemi1')
except Exception as err:
print('文件打开失败!' + str(err))
return None
# 创建读实例
pdf_reader = 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
decrypt_pdf(r'0627.pdf', '')