Python加密pdf文件

# 文件      ：pdf加密.py
# IDE      ：PyCharm

# 导入加密模块
import pikepdf
# 导入操作模块
import os
# 导入md5加密模块
from hashlib import md5
# 此模块实现输入密码不显示
import getpass

# 搜索pdf文件,加密
def search_pdf(path, passwrd, username):
    for pdfFile in os.listdir(path):
        pdfPath = os.path.join(path, pdfFile)
        if os.path.isdir(pdfPath):
            search_pdf(pdfPath, passwrd, username)

        elif '.pdf' in pdfPath:

            encry_pdf(pdfPath, encrypt_md5(passwrd), username)
            print(f'{pdfPath}已经加密！！密码保存好！！！')


# 加密函数
def encrypt_md5(passwrd_str):
    new_md5 = md5()   # 创建md5对象
    new_md5.update(passwrd_str.encode(encoding='utf-8'))
    # 返回信息摘要
    return  new_md5.hexdigest()


# 加密函数，参数为pdf文件名，密码
def encry_pdf(pdfname, passwrd, username):
    # 打开被加密的文件，允许覆盖
    pdf = pikepdf.open(pdfname, allow_overwriting_input=True)
    # 保存加密文件
    pdf.save(pdfname, encryption=pikepdf.Encryption(owner=passwrd,user=username, R=4))

# 主函数
if __name__ == '__main__':
    Done = True
    path = input('请输入文件所在目录：')
    while Done:
        try:
            passwrd1 = getpass.getpass("请输入加密文件的密码：", stream=None)
            passwrd2 = getpass.getpass("请再次输入加密文件的密码：", stream=None)
            if passwrd1 == passwrd2:
                username1 = input('请输入用户名：')
                search_pdf(path, passwrd1, username1)
                Done = False
                break
            else:
                print('密码输入有误！！！')
        except:
            print('请核实信息！！')