最全总结！聊聊 Python 操做PDF的几种方法（合并、拆分、水印、加密）

1、前言

你们好，有关Python操做PDF的案例以前已经写过一个👉PDF批量合并，这个案例初衷只是给你们提供一个便利的脚本，并无太多讲解原理,其中涉及的就是PDF处理很实用的模块PyPDF2，本文就好好剖析一下这个模块，主要将涉及

• os 模块综合应用

• glob 模块综合应用

• PyPDF2 模块操做

2、基本操做

PyPDF2 导入模块的代码经常是：

from PyPDF2 import PdfFileReader, PdfFileWriter

这里导入了两个方法：

• PdfFileReader 能够理解为读取器

• PdfFileWriter能够理解为写入器

接下来经过几个案例进一步认识这两个工具的奇妙之处，用到的示例文件是5个发票的pdf每一个发票的PDF都由两页组成：

3、合并

第一个工做是将5个发票pdf合并成10页。这里读取器和写入器应该怎么配合呢？

逻辑以下：

1. 读取器将全部pdf读取一遍

2. 读取器将读取的内容交给写入器

3. 写入器统一输出到一个新pdf

这里还有一个重要的知识点：读取器只能将读取的内容一页一页交给写入器。

所以，逻辑中第1步和第2步实际上不是彼此独立的步骤，而是读取器读取完一个pdf后，就将这个pdf所有页循环一遍，挨页交给写入器。最后等读取工做所有结束后再输出。

看一下代码可让思路更清楚：

from PyPDF2 import PdfFileReader, PdfFileWriter

path = r'C:\Users\xxxxxx'
pdf_writer = PdfFileWriter()

for i in range(1, 6):
    pdf_reader = PdfFileReader(path + '/INV{}.pdf'.format(i))
    for page in range(pdf_reader.getNumPages()):
        pdf_writer.addPage(pdf_reader.getPage(page))

with open(path + r'\合并PDF\merge.pdf', 'wb') as out:
    pdf_writer.write(out)

因为所有内容都须要交给同一个写入器最后一块儿输出，因此写入器的初始化必定是在循环体以外的.

若是在循环体内则会变成每次访问读取一个pdf就生成一个新的写入器，这样每个读取器交给写入器的内容就会被反复覆盖，没法实现咱们的合并需求!

循环体开头的代码：

for i in range(1, 6):
    pdf_reader = PdfFileReader(path + '/INV{}.pdf'.format(i))

目的就是每次循环读取一个新的pdf文件交给读取器进行后续操做。实际上这种写法不是很提倡，因为各pdf命名刚好很规则，因此能够直接人为指定数字进行循环。更好的方法是用 glob 模块：

import glob
for file in glob.glob(path + '/*.pdf'):
    pdf_reader = PdfFileReader(path)

代码中 pdf_reader.getNumPages(): 可以获取读取器的页数，配合range就能遍历读取器的全部页。

pdf_writer.addPage(pdf_reader.getPage(page))可以将当前页交给写入器。

最后，用with新建一个pdf并由写入器的 pdf_writer.write(out)方法输出便可

4、拆分

若是明白了合并操做中读取器和写入器的配合，那么拆分就很好理解了，这里咱们以拆分INV1.pdf为2个单独的pdf文档为例，一样也先来捋一捋逻辑：

1. 读取器读取PDF文档

2. 读取器一页一页交给写入器

3. 写入器每获取一页就当即输出

经过这个代码逻辑咱们也能够明白，写入器初始化和输出的位置必定都在读取PDF循环每一页的循环体内，而不是在循环体外

代码很简单：

from PyPDF2 import PdfFileReader, PdfFileWriter
path = r'C:\Users\xxx'
pdf_reader = PdfFileReader(path + '\INV1.pdf')

for page in range(pdf_reader.getNumPages()):
    # 遍历到每一页挨个生成写入器
    pdf_writer = PdfFileWriter()
    pdf_writer.addPage(pdf_reader.getPage(page))
    # 写入器被添加一页后当即输出产生pdf
    with open(path + '\INV1-{}.pdf'.format(page + 1), 'wb') as out:
        pdf_writer.write(out)

5、水印

本次的工做是将下图做为水印添加到INV1.pdf中首先是准备工做，将须要做为水印的图片插入word中调整合适位置后保存为PDF文件。而后就能够码代码了，须要额外用到copy模块，具体解释见下图：就是把读取器和写入器初始化，而且把水印PDF页先读取好备用,核心代码稍微比较难理解：加水印本质上就是把水印PDF页和须要加水印的每一页都合并一遍

因为须要加水印的PDF可能有不少页，而水印PDF只有一页，所以若是直接把水印PDF拿来合并，能够抽象理解成加完第一页，水印PDF页就没有了。

所以不能直接拿来合并，而要把水印PDF页不断copy出来成新的一页备用new_page，再运用.mergePage方法完成跟每一页合并，把合并后的页交给写入器待最后统一输出!

关于.mergePage的使用：出如今下面的页.mergePage(出如今上面的页)，最后效果如图：

6、加密

加密很简单，只须要记住：「加密是针对写入器加密」

所以只须要在相关操做完成后调用pdf_writer.encrypt(密码)

以单个PDF的加密为例：

写在最后

固然除了对PDF的合并、拆分、加密、水印，咱们还可使用Python结合Excel和Word实现更多的自动化需求，这些就留给读者本身开发。

最后仍是但愿你们可以理解Python办公自动化的一个核心就是批量操做-解放双手，让复杂的工做自动化！

今天的文章就到这里，原创不易，若是喜欢的话请给我一波三连支持吧(在看、转发、留言)

-END-

本文分享自微信公众号 - 早起Python（zaoqi-python）。
若有侵权，请联系 support@oschina.cn 删除。
本文参与“OSC源创计划”，欢迎正在阅读的你也加入，一块儿分享。