Python编程快速上手——PDF文件操作案例分析
本文实例讲述了PythonPDF文件操作。分享给大家供大家参考，具体如下：
题目如下：
- 利用第九章的os.walk()函数编写脚本，遍历文件夹中的所有pdf，用命令行提供的命令对这些PDF进行加密，用原来的文件名加上_encrypted.pdf后缀，保存每个加密的PDF。在删除原来的文件之前，尝试用程序读取并解密该文件，确保被正确加密
- 然后编写一个程序，找到文件夹中所有加密的PDF文件，利用提供的口令，创建pdf的解密拷贝，如果口令不对，程序应该打印一条消息，
并继续处理下一个pdf文件
思路如下：
-程序内函数1需要做以下事情：
找出文件夹中所有PDF文件
对PDF文件进行加密
保存加密的PDF文件
检验是否正确加密
删除源文件
-程序内函数2需要做以下事情：
遍历文件夹中所有带_encrypted后缀的PDF文件
利用提供的口令进行打开
能够正确打开，则进行口令拷贝保存到txt文件
不能正确打开输出到屏幕
-代码需要做以下事情：
导入os,PyPDF2,sys，send2trash
生成新文件夹用于保存加密PDF及拷贝文本
- 编写一个加密函数
函数内调用os.walk()遍历文件夹，文件名保存到列表
命令行参数sys.argv()提供加密口令
for循环进行文件加密和保存加密文件操作
decrypt进行解密，确保正确加密，并进行反馈
删除原有文件(send2trash) - 编写一个生成解密拷贝函数
os.walk()遍历，decrypt进行解密，反馈结果，生成密码拷贝txt
try-except进行decrypt控制，解密失败打印消息，continue继续
代码如下：
由于我的代码在命令行运行时提示找不到PyPDF2模块，所以sys.argv命令行参数用的字符串直接放入函数进行代替。
#!python3
importos,sys,PyPDF2,send2trash
os.makedirs(".\\NewPDF")
print("文件夹创建成功！")
path1=os.path.abspath(".\\NewPDF")
#文件加密函数
defdecryptFile(argv,p=os.path.abspath(".\\New")):
tagFloder='.\\PDF'
pdfList=[]
#当前目录下创建新文件夹
#os.makedirs(".\\NewPDF")
#遍历目标文件夹，将.pdf文件名添加到列表
forfoldername,subfolders,filenamesinos.walk(tagFloder):
forfilenameinfilenames:
iffilename.endswith('.pdf'):
pdfList.append(filename)
print('找到PDF文件：%s'%filename)
else:
continue
#对pdf文件进行加密)
foriinpdfList:
pdfFile=open(os.path.join(foldername,filename),'rb')
pdfReader=PyPDF2.PdfFileReader(pdfFile)
pdfWriter=PyPDF2.PdfFileWriter()
#拷贝
forpageNuminrange(pdfReader.numPages):
pdfWriter.addPage(pdfReader.getPage(pageNum))
newName=i[0:-4]#对文件名进行分割
pdfWriter.encrypt(argv)#口令加密
pdfResult=open("{0}\\{1}_encrypted.pdf".format(p,newName),"wb+")#创建新文件名对象
pdfWriter.write(pdfResult)#写入新文件
pdfReader2=PyPDF2.PdfFileReader(pdfResult,'rb')
#进行加密确认
ifpdfReader2.decrypt(argv):
print("正确加密！删除原文件中...")
try:
send2trash.send2trash(os.path.join(foldername,i))
except:
print("删除原文件：%s失败!"%i)
pdfResult.close()
print("Done!")
#口令拷贝函数
defcopyDcrypt(argv,p=os.path.abspath(".\\New")):
pdfList=[]
forfoldername,subfolders,filenamesinos.walk("."):
print("父文件夹：%s"%foldername)
forfilenameinfilenames:
iffilename.endswith(".pdf"):
pdfReader=PyPDF2.PdfFileReader(open(os.path.join(foldername,filename),'rb'))
ifpdfReader.isEncrypted:
pdfList.append(filename)
print('找到已加密PDF文件：%s'%filename)
else:
continue
foriinpdfList:
newName=i[0:-4]#对文件名进行分割
try:
ifpdfReader.decrypt(argv)==1:
copyFile=open("{0}\\{1}_PASSWORD.txt".format(p,newName),'w')
copyFile.write("Passwordis：%s"%argv)
copyFile.close()
print("口令正确！拷贝生成成功！")
else:
print("口令错误！")
except:
continue
#调用函数
decryptFile("ABCDEFG",path1)
copyDcrypt("ABCDEFG",path1)
运行结果：
- pycharm界面运行结果：
-原文件夹：
-新文件夹：
-拷贝文本文件：
更多Python相关内容感兴趣的读者可查看本站专题：《Python文件与目录操作技巧汇总》、《Python编码操作技巧总结》、《Python数据结构与算法教程》、《Python函数使用技巧总结》、《Python字符串操作技巧汇总》及《Python入门与进阶经典教程》
希望本文所述对大家Python程序设计有所帮助。
声明：本文内容来源于网络，版权归原作者所有，内容由互联网用户自发贡献自行上传，本网站不拥有所有权，未作人工编辑处理，也不承担相关法律责任。如果您发现有涉嫌版权的内容，欢迎发送邮件至：czq8825#qq.com（发邮件时，请将#更换为@）进行举报，并提供相关证据，一经查实，本站将立刻删除涉嫌侵权内容。