Python实现基于身份证后六位的PDF批量加密

import os
from PyPDF2 import PdfFileWriter, PdfFileReader
import pandas as pd

# 全局工作目录
os.chdir("D:\加密")

def main():
    # 1、找文件
    # 获取实习证明原文件的文件名
    rosterfile = os.listdir('.\原文件\\')
    # 获取信息中表的姓名与密码
    name,Password = getPassword()

    # 对每个文件进行加密并保存
    for prove in rosterfile:
        for Name in name:
            if Name in prove:
                i = name.index(Name)
                rosterLast = prove
                rosterNow = '.\文件加密版本\\' + prove
                password = Password[i]
                # 2、加密文件并输出
                encryptionPDF(rosterLast,rosterNow,password)


# 获取了表中姓名信息，并提取身份证后六位作为密码
def getPassword():
    password = []
    name = []
    # 获取Excel列，包括姓名、身份证号
    Name = pd.read_excel('.\名单.xlsx', usecols=[0],names=None)  # 读取项目名称列,不要列名
    Id = pd.read_excel('.\名单.xlsx', usecols=[1],names=None)  # 读取项目名称列,不要列名
    Name1 = Name.values.tolist()
    id = Id.values.tolist()
    # 提取身份证后六位作为密码
    for i in id:
        password.append(str(i[0])[12:18:1])
    # 提取姓名为字符串，原来为list
    for j in Name1:
        name.append(j[0])
    return name,password

# 加密文件并输出
# 变量说明：rosterLast-原PDF文件路径,rosterNow-加密后输出路径,password-加密密码
def encryptionPDF(rosterLast,rosterNow,password):
    pdf_reader = PdfFileReader(r'.\原文件\\%s'%(rosterLast))  # 输入你想要操作的pdf文档的位置/名称
    pdf_writer = PdfFileWriter()

    for page in range(pdf_reader.getNumPages()):
        pdf_writer.addPage(pdf_reader.getPage(page))

    pdf_writer.encrypt(password)  # 写入密码并输出到目的路径
    with open(rosterNow, 'wb') as out:
        pdf_writer.write(out)
    print("文件:%s----已完成加密,密码为：%s"%(rosterLast,password))

# 函数入口
if __name__ == '__main__':
    print("-----开始加密-----")
    main()
    print("-----加密完成-----")

# 文件操作
import os

# 操作PDF文件
from PyPDF2 import PdfFileWriter, PdfFileReader

# 操作数据
import pandas as pd

# 主函数，从这里我们可以看到我整体是如何实现此需求的
def main():

# 功能函数，用来获取文件密码的，然后使用姓名与文件名做匹配，达到本人文件为本人身份证后六位的需求。
def getPassword():

#  功能函数，本次编码的核心，前面都可以作为准备阶段，就是为本函数用来加密文件而提供读取、输出路径以及密码的。
def encryptionPDF(rosterLast,rosterNow,password):