Python 自动化：处理 PDF（笔记）

import PyPDF2 # 导入 PyPDF2 模块处理 PDF

pdfFileObj = open('meetingminutes.pdf', 'rb') # 把目标文件作为二进制文件打开
pdfReader = PyPDF2.PdfFileReader(pdfFileObj) # 用模块的 PdfFileReader 方法把打开的文件变成一个 PDF 文件对象
print(pdfReader.numPages) # 显示 PDF 文件对象有多少页
pageObj = pdfReader.getPage(0) # 把 PDF 文件的第一页变成一个 PDF 页面对象
print(pageObj.extractText()) # 把这个 PDF 页面对象的文字抽取出来
pdfFileObj.close() # 关闭目标文件

pdfReader = PyPDF2.PdfFileReader(open('encrypted.pdf', 'rb')) # 把目标文件作为二进制文件打开，用模块的 PdfFileReader 方法把打开的文件变成一个 PDF 文件对象
print(pdfReader.isEncrypted) # 检查 PDF 文件是否已经加密
pageObj = pdfReader.getPage(0) # 尝试获得第一页，加密的状态下会出错
pdfReader = PyPDF2.PdfFileReader(open('encrypted.pdf', 'rb')) # 再次把目标文件作为二进制文件打开，用模块的 PdfFileReader 方法把打开的文件变成一个 PDF 文件对象
pdfReader.decrypt('rosebud') # 使用密码 rosebud 解密解密 PDF 对象；并不能直接去掉 PDF 文件本身的密码
pageObj = pdfReader.getPage(0) # 获得第一页
print(pageObj.extractText()) # 把这个 PDF 页面对象的文字抽取出来
pdfFileObj.close() # 关闭目标文件

pdf1File = open('meetingminutes.pdf', 'rb') # 把第一个目标文件作为二进制文件打开
pdf2File = open('meetingminutes2.pdf', 'rb') # 把第二个目标文件作为二进制文件打开
pdf1Reader = PyPDF2.PdfFileReader(pdf1File) # 用模块的 PdfFileReader 方法把打开的文件变成一个 PDF 文件对象
pdf2Reader = PyPDF2.PdfFileReader(pdf2File) # 用模块的 PdfFileReader 方法把打开的文件变成一个 PDF 文件对象
pdfWriter = PyPDF2.PdfFileWriter() # 用模块的 PdfFileWriter 方法创建一个新的 PDF 文件对象

for pageNum in range(pdf1Reader.numPages): # 在第一个 PDF 文件对象的页数范围内
    pageObj = pdf1Reader.getPage(pageNum) # 遍历每一个 PDF 文件对象的每一页
    pdfWriter.addPage(pageObj) # 把这一页新增到 pdfWriter 文件对象当中。只能新增页面，无法从中间插入

for pageNum in range(pdf2Reader.numPages): # 在第二个 PDF 文件对象的页数范围内
    pageObj = pdf2Reader.getPage(pageNum) # 遍历每一个 PDF 文件对象的每一页
    pdfWriter.addPage(pageObj) # 把这一页新增到 pdfWriter 文件对象当中。只能新增页面，无法从中间插入

pdfOutputFile = open('combinedminutes.pdf', 'wb') # 在文件系统新建一个指定文件名的文件
pdfWriter.write(pdfOutputFile) # 把 pdfWriter 文件对象写入目标文件当中
pdfOutputFile.close() # 关闭文件
pdf1File.close() # 关闭文件
pdf2File.close() # 关闭文件

minutesFile = open('meetingminutes.pdf', 'rb') # 把目标文件作为二进制文件打开
pdfReader = PyPDF2.PdfFileReader(minutesFile) # 用模块的 PdfFileReader 方法把打开的文件变成一个 PDF 文件对象
page = pdfReader.getPage(0) # 把文件对象的第一页作为一个页面
page.rotateClockwise(90) # 把第一页顺时针旋转 90 度；只能是 90 180 270
pdfWriter = PyPDF2.PdfFileWriter() # 用模块的 PdfFileWriter 方法创建一个新的 PDF 文件对象
pdfWriter.addPage(page) # 把这一页新增到 pdfWriter 文件对象当中
resultPdfFile = open('rotatedPage.pdf', 'wb') # 在文件系统新建一个指定文件名的文件
pdfWriter.write(resultPdfFile) # 把 pdfWriter 文件对象写入目标文件当中
resultPdfFile.close() # 关闭文件
minutesFile.close() # 关闭文件

minutesFile = open('meetingminutes.pdf', 'rb') # 把目标文件作为二进制文件打开
pdfReader = PyPDF2.PdfFileReader(minutesFile) # 用模块的 PdfFileReader 方法把打开的文件变成一个 PDF 文件对象
minutesFirstPage = pdfReader.getPage(0) # 把文件对象的第一页作为一个页面
pdfWatermarkReader = PyPDF2.PdfFileReader(open('watermark.pdf', 'rb')) # 把水印目标文件作为二进制文件打开，用模块的 PdfFileReader 方法把打开的文件变成一个 PDF 文件对象
minutesFirstPage.mergePage(pdfWatermarkReader.getPage(0)) # 把原始文件的第一页与水印文件的第一页合并，水印文件在上层
pdfWriter = PyPDF2.PdfFileWriter() # 用模块的 PdfFileWriter 方法创建一个新的 PDF 文件对象
pdfWriter.addPage(minutesFirstPage) # 把合并后的这一页新增到 pdfWriter 文件对象当中

for pageNum in range(1, pdfReader.numPages): # 对剩下的页面不做处理，新增加到新的 PDF 文件对象当中
    pageObj = pdfReader.getPage(pageNum) # 遍历第二页到最后一页，把原始文件的每一页都作为一个页面对象
    pdfWriter.addPage(pageObj) # 把每一个页面对象新增到 pdfWriter 文件对象当中

resultPdfFile = open('watermarkedCover.pdf', 'wb') # 在文件系统新建一个指定文件名的文件
pdfWriter.write(resultPdfFile) # 把 resultPdfFile 文件对象写入目标文件当中
minutesFile.close() # 关闭文件
resultPdfFile.close() # 关闭文件

pdfFile = open('meetingminutes.pdf', 'rb') # 把目标文件作为二进制文件打开
pdfReader = PyPDF2.PdfFileReader(pdfFile) # 用模块的 PdfFileReader 方法把打开的文件变成一个 PDF 文件对象
pdfWriter = PyPDF2.PdfFileWriter() # 用模块的 PdfFileWriter 方法创建一个新的 PDF 文件对象
for pageNum in range(pdfReader.numPages): # 在 PDF 文件对象的页数范围内
    pdfWriter.addPage(pdfReader.getPage(pageNum)) # 遍历每一个 PDF 文件对象的每一页，并新增到一个新的文件当中

pdfWriter.encrypt('swordfish') # 把新的 PDF 文件对象用密码 swordfish 加密。PDF 有两个密码，一个是查看的密码，一个是拥有者密码，分别对应 encrypt() 的两个参数。如果只指定一个，则是两者都是
resultPdfFile = open('encryptedminutes.pdf', 'wb') # 在文件系统新建一个指定文件名的文件
pdfWriter.write(resultPdfFile) # 把 resultPdfFile 文件对象写入目标文件当中
resultPdfFile.close() # 关闭文件
pdfFile.close() # 关闭文件