Python编程实现批量md5加密pdf文件

Python编程实现批量md5加密pdf文件
今天在某公众号看到一个关于pdf文件加密模块的介绍，突然有了一个想法，编写一个批量加密某目录下的所有pdf文件。结合之前学到的md5加密和文件查找等知识，实现这个想法。

运行环境：

OS：macOS Monterey Version 12.3.1（英文版）

IDE：PyCharm 2020.1

代码分享如下：

#_*_coding:utf-8_*_
# 作者      ：liuxiaowei
# 创建时间   ：4/24/22 11:38 AM
# 文件      ：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('请核实信息！！')

运行结果如下：

zsh: suspended  sudo python pdf加密.py
(venv) (base) liuxiaowei@MacBookAir 10个有趣的Python高级脚本 % sudo python pdf加密.py
请输入文件所在目录：/Users/liuxiaowei/Desktop
请输入加密文件的密码：
请再次输入加密文件的密码：
密码输入有误！！！              # 密码输入有误
请输入加密文件的密码：
请再次输入加密文件的密码：  
密码输入有误！！！
请输入加密文件的密码：      
请再次输入加密文件的密码：   # 密码输入正确
请输入用户名：lxw
/Users/liuxiaowei/Desktop/sql注入.pdf已经加密！！密码保存好！！！ # 当前目录下的pdf文件
/Users/liuxiaowei/Desktop/TS852D Specification.pdf已经加密！！密码保存好！！！
/Users/liuxiaowei/Desktop/md学习笔记/pdf/Windows PE权威指南.pdf已经加密！！密码保存好！！！ # 子目录下的pdf文件
/Users/liuxiaowei/Desktop/md学习笔记/pdf/PythonCodingRule.pdf已经加密！！密码保存好！！！

感兴趣的可以试试。欢迎分享，收藏，点赞。