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

# 導入加密模塊
import pikepdf
# 導入操作模塊
import os
# 導入md5加密模塊
from hashlib import md5
# 此模塊實現(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對象
? ? 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('請輸入文件所在目錄：')
? ? 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('請核實信息！！')