Py 自动化办公

PyPDF2 历史

正文开始之前，说一下 PyPDF2 的发展历史，PyPDF 的前身是 pyPDf

包在2005年发布，该包的最后一个版本发布于2010年，后来大约经过一年左右，名为 Phasit 的公司赞助 PyPdf 的一个分支后来命名为

PyPDF2，两个版本功能都基本一样，最大区别就是 PyPDF2 中加入了支持 Python3 特性；

PyPDF2 近期也没有再更新了，最近一个版本发布在2016年，但使用热度依然没有消退；虽然后面又出现了 PyPDF3、PyPDF4

等不同版本，但这些包并没有对 PyPDF2 功能向后完全兼容，用户受欢迎程度当然也不如 PyPDF2

PyPDF2 安装

与其它Python 库一样，安装可通过 pip 或 conda 工具

代码语言：txt复制

pip install pypdf2

PDF 信息提取

使用 PyPDF2 可以从 PDF 中提取到一些元数据和文本信息，对 PDF 有个大致了解

用 PyPDF2 能够提取的数据如下

作者；
创建者；
制作者；
Subject；
标题；
页数；

这里我下载了官网提供的 PDF 样本《Seige_of_Vicksburg_Sample_OCR》一共六页，作为测试数据

image-20210313230206113

代码语言：txt复制

from  PyPDF2 import PdfFileReader

代码语言：txt复制

# # pdf 文档

代码语言：txt复制

pdf_path = "D:/Data/自动化办公/PDF/Seige_of_Vicksburg_Sample_OCR.pdf"

代码语言：txt复制

with open(pdf_path,'rb') as f:

代码语言：txt复制

 pdf = PdfFileReader(f)

代码语言：txt复制

 infomation = pdf.getDocumentInfo()

代码语言：txt复制

 number_of_pages = pdf.getNumPages()

代码语言：txt复制

 txt = f'''{pdf_path} information:

代码语言：txt复制

 Author : {infomation.author},

代码语言：txt复制

 Creator : {infomation.creator},

代码语言：txt复制

 Producer : {infomation.producer},

代码语言：txt复制

 Subject : {infomation.subject},

代码语言：txt复制

 Title : {infomation.title},

代码语言：txt复制

 Number of pages : {number_of_pages}

代码语言：txt复制

'''

代码语言：txt复制

 print(txt)

下面为打印结果

代码语言：txt复制

D:/Data/自动化办公/PDF/Seige_of_Vicksburg_Sample_OCR.pdf information:

代码语言：txt复制

 Author : DSI,

代码语言：txt复制

 Creator : LuraDocument PDF Compressor Server 5.5.46.38,

代码语言：txt复制

 Producer : LuraDocument PDF v2.38,

代码语言：txt复制

 Subject : None,

代码语言：txt复制

 Title : Binder1.pdf,

代码语言：txt复制

 Number of pages : 6

在上面例子中用到了 PdfFileReader 类，用于与 pdf 文件交互；调用该类中的 getDocumentInfo() 方法返回一个

DocumentInformation 的实例，该实例中存储着我们需要的信息；对 reader 对象调用 getNumPages 方法也可以返回文档页数；

个人看法，这里面的数据也就页数有点价值，当批量统计时该方法很适用

PDF 页面旋转

PyPDF2 中 pdf 每一页都是以 page 对象存在，返回某一页的实例可通过 reader 对象中的 get_Page(page_index)

方法，其中 page_index 表示索引

对某一页旋转，有两种方式

rotateClockwise(90)，顺时针旋转90度；
rotateCounterClockwise(90)，逆时针旋转 90 度；

下面代码表示将目标 PDF 中第一页顺时针方向旋转 90 度，第二页以逆时针方向旋转 90 度，其它页位置角度不变；

代码语言：txt复制

from  PyPDF2 import PdfFileReader,PdfFileWriter

代码语言：txt复制

pdf_writer = PdfFileWriter()

代码语言：txt复制

pdf_reader = PdfFileReader(pdf_path)

代码语言：txt复制

# Rotate page 90 degrees to the right

代码语言：txt复制

page_1 = pdf_reader.getPage(0).rotateClockwise(90)

代码语言：txt复制

pdf_writer.addPage(page_1)

代码语言：txt复制

# Rotate page 90 degrees to the left

代码语言：txt复制

page_2 = pdf_reader.getPage(1).rotateCounterClockwise(90)

代码语言：txt复制

pdf_writer.addPage(page_2)

代码语言：txt复制

# 之后的正常写出

代码语言：txt复制

for i in range(2,pdf_reader.getNumPages()):

代码语言：txt复制

 pdf_writer.addPage(pdf_reader.getPage(i))

代码语言：txt复制

with open(pdf_path, 'wb') as fh:

代码语言：txt复制

 pdf_writer.write(fh)

结果如下

image-20210313232532349

代码中同时用到了PdfFileReader,PdfFileWriter 这两个类，页面旋转并不是在原有 PDF

基础上进行操作而是在内存处创建了一个新的PDF流对象，将操作后的每一页通过 addPage() 方法加入到这个对象中，之后将内存中的这个对象写入到文件中；

写到这里，说实话其实 页面旋转 这个功能没基本没什么作用，加在这里只是想充当一些字数，哈哈哈

单个 PDF 拆分成多个PDF

代码语言：txt复制

from  PyPDF2 import PdfFileReader,PdfFileWriter

代码语言：txt复制

# # pdf 文档

代码语言：txt复制

pdf_path = "D:/Data/自动化办公/PDF/Seige_of_Vicksburg_Sample_OCR.pdf"

代码语言：txt复制

save_path = 'D:/Data/自动化办公/PDF/'

代码语言：txt复制

# Split Pages of PDF

代码语言：txt复制

pdf_reader = PdfFileReader(pdf_path)

代码语言：txt复制

for i in range(0,pdf_reader.getNumPages()):

代码语言：txt复制

 pdf_writer = PdfFileWriter()

代码语言：txt复制

 pdf_writer.addPage(pdf_reader.getPage(i))

代码语言：txt复制

 # Every page write to a path

代码语言：txt复制

 with open(save_path '{}.pdf'.format(str(i)), 'wb') as fh:

代码语言：txt复制

 pdf_writer.write(fh)

代码语言：txt复制

 print('{} Save Sucessfully !n'.format(str(i)))

代码将 PDF 原文件中的每一页拆分到每一个PDF文件，其中文件名用页索引来命名；

image-20210313235957539

通过拆分也可以提取到 pdf 文件中固定页码范围，例如我只想提取 pdf 中的 2-5 页，其它部分不要，那么代码将写成下面形式

代码语言：txt复制

pdf_writer = PdfFileWriter()

代码语言：txt复制

pdf_reader = PdfFileReader(pdf_path)

代码语言：txt复制

for i in range(1,5):

代码语言：txt复制

 # pdf_writer = PdfFileWriter()

代码语言：txt复制

 pdf_writer.addPage(pdf_reader.getPage(i))

代码语言：txt复制

 # Every page write to a path

代码语言：txt复制

with open(save_path '2_5.pdf', 'wb') as fh:

代码语言：txt复制

 pdf_writer.write(fh)

多个 PDF 文件合并为单个

pdf 拆分与合并方向虽然相反，但用到的类、原理都是一样的

PdfFileReader读取每个pdf，并递归获取每一页page 对象， PdfFileWrite 新建一个流对象，把前面内存中读取到的 page

对象按顺序写入到这个流对象中，最后写入到磁盘文件

代码语言：javascript复制

···  
from PyPDF2 import PdfFileReader,PdfFileWriter
p1_pdf = "D:/Data/自动化办公/PDF/Seige_of_Vicksburg_Sample_OCR.pdf"  
p2_pdf = "D:/Data/自动化办公/PDF/Seige_of_Vicksburg_Sample_OCR.pdf"
merge_pdf = 'D:/Data/自动化办公/PDF/merge.pdf'
p1_reader = PdfFileReader(p1_pdf)  
p2_reader = PdfFileReader(p2_pdf)
merge = PdfFileWriter()

Write p1

代码语言：javascript复制

for i in range(0,p1_reader.getNumPages()):  
merge.addPage(p1_reader.getPage(i))

Write p2

代码语言：javascript复制

for j in range(0,p2_reader.getNumPages()):  
merge.addPage(p2_reader.getPage(j))

Write out

代码语言：javascript复制

with open(merge_pdf,'wb') as f:  
merge.write(f)  
···

运行结果如下

代码语言：javascript复制

image-20210314002536754

PDF 添加水印

在今天列举的这么多功能中，我想这个功能是最有用，批量添加水印主要用到 page 对象中的 margePage()

方法，通过将两个页面合并来达到添加水印的效果

因为 PyPDF2 只能操作 pdf 对象，因此在添加水印之前，需要将准备添加的水印存放到一个 pdf 文件中

代码语言：javascript复制

···  
from PyPDF2 import PdfFileReader,PdfFileWriter  
watermark = 'D:/Data/自动化办公/PDF/watermark.pdf'  
input_pdf = 'D:/Data/自动化办公/PDF/merge.pdf'  
output = 'D:/Data/自动化办公/PDF/merge_watermark.pdf'
watermark_obj = PdfFileReader(watermark)  
watermark_page = watermark_obj.getPage(0)
pdf_reader = PdfFileReader(input_pdf)  
pdf_writer = PdfFileWriter()

Watermark all the pages

代码语言：javascript复制

for page in range(pdf_reader.getNumPages()):  
page = pdf_reader.getPage(page)  
page.mergePage(watermark_page)  
pdf_writer.addPage(page)
with open(output, 'wb') as out:  
pdf_writer.write(out)  
···

效果如下，从左到右，依次为原图、水印、添加水印后的原图

从左到右，依次是原图、水印、加完水印的效果图

上面效果不好是因为制作水印时没有考虑到页面布局问题，所以合并时出现一部分缺失；

用以上代码添加水印的好处是，可以对 pdf 指定页田间水印，比如说只对奇数页添加偶数页不管，不但灵活性强而且高效，当然也可以对多个文件进行批量操作

PDF加密解密

pdf加密

对一份 pdf 文件，如果我们不想让其他人能够读取里面的内容，可以通过 pypdf2

对它设置密码，如果只是单个文件的话，建议最好自己找个工具受手动操作一下会高效一点，但若是多个文件，非常建议用下面方法

代码语言：javascript复制

···  
watermark = 'D:/Data/自动化办公/PDF/Seige_of_Vicksburg_Sample_OCR.pdf'  
input_pdf = 'D:/Data/自动化办公/PDF/merge.pdf'  
output = 'D:/Data/自动化办公/PDF/merge_watermark1.pdf'
watermark_obj = PdfFileReader(watermark)  
watermark_page = watermark_obj.getPage(0)
pdf_reader = PdfFileReader(input_pdf)  
pdf_writer = PdfFileWriter()

Watermark all the pages

代码语言：javascript复制

for page in range(pdf_reader.getNumPages()):  
page = pdf_reader.getPage(page)  
page.mergePage(watermark_page)  
pdf_writer.addPage(page)  
pdf_writer.encrypt(user_pwd='123456',  
use_128bit=True)  
with open(output, 'wb') as out:  
pdf_writer.write(out)  
···
image-20210314092935806

主要用到 encrypt 函数，需要注意三个参数

user_pwd，str，用户密码，用来限制打开读取文件；
owner_pwd，str，比用户密码更高一级，提供时可让打开文件不受任何限制，不指定时默认owner_pwd 与 user_pwd 相同；
use_128bit 布尔值，用来表示是否使用128位作为密码，False 时代表用 40 位密码，默认为True；

pdf解密

解密是在读取文件时用的，用到 decrypt() 函数

代码语言：txt复制

from PyPDF2 import PdfFileWriter, PdfFileReader

代码语言：txt复制

input_pdf='reportlab-encrypted.pdf'

代码语言：txt复制

output_pdf='reportlab.pdf'

代码语言：txt复制

password='twofish'

代码语言：txt复制

pdf_writer = PdfFileWriter()

代码语言：txt复制

pdf_reader = PdfFileReader(input_pdf)

代码语言：txt复制

pdf_reader = pdf_reader.decrypt(password)

代码语言：txt复制

for page in range(pdf_reader.getNumPages()):

代码语言：txt复制

 pdf_writer.addPage(pdf_reader.getPage(page))

代码语言：txt复制

with open(output_pdf, 'wb') as fh:

代码语言：txt复制

 pdf_writer.write(fh)

上面例子中解密原理是通过将一个加密文件进行读取，并写入到一个非加密 pdf 中

image pdf watermark

0 人点赞