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

# 導(dǎo)入加密模塊
import pikepdf
# 導(dǎo)入操作模塊
import os
# 導(dǎo)入md5加密模塊
from hashlib import md5
# 此模塊實(shí)現(xiàn)輸入密碼不顯示
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}已經(jīng)加密！！密碼保存好！??！')


# 加密函數(shù)
def encrypt_md5(passwrd_str):
? ? new_md5 = md5() ? # 創(chuàng)建md5對(duì)象
? ? new_md5.update(passwrd_str.encode(encoding='utf-8'))
? ? # 返回信息摘要
? ? return ?new_md5.hexdigest()


# 加密函數(shù)，參數(shù)為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))

# 主函數(shù)
if __name__ == '__main__':
? ? Done = True
? ? path = input('請(qǐng)輸入文件所在目錄：')
? ? while Done:
? ? ? ? try:
? ? ? ? ? ? passwrd1 = getpass.getpass("請(qǐng)輸入加密文件的密碼：", stream=None)
? ? ? ? ? ? passwrd2 = getpass.getpass("請(qǐng)?jiān)俅屋斎爰用芪募拿艽a：", stream=None)
? ? ? ? ? ? if passwrd1 == passwrd2:
? ? ? ? ? ? ? ? username1 = input('請(qǐng)輸入用戶名：')
? ? ? ? ? ? ? ? search_pdf(path, passwrd1, username1)
? ? ? ? ? ? ? ? Done = False
? ? ? ? ? ? ? ? break
? ? ? ? ? ? else:
? ? ? ? ? ? ? ? print('密碼輸入有誤！！！')
? ? ? ? except:
? ? ? ? ? ? print('請(qǐng)核實(shí)信息??！')