最全總結(jié)！聊聊 Python 操作PDF的幾種方法

以下文章來源于早起Python ，作者陳熹

前言

本文主要涉及：

·  os 模塊綜合應(yīng)用

·  glob 模塊綜合應(yīng)用

·  PyPDF2 模塊操作

基本操作

PyPDF2 導(dǎo)入模塊的代碼常常是：

from PyPDF2 import PdfFileReader, PdfFileWriter

這里導(dǎo)入了兩個(gè)方法：

·  PdfFileReader 可以理解為讀取器

·  PdfFileWriter可以理解為寫入器

接下來通過幾個(gè)案例進(jìn)一步認(rèn)識這兩個(gè)工具的奇妙之處，用到的示例文件是5個(gè)****的pdf

每個(gè)****的PDF都由兩頁組成：

合并

第一個(gè)工作是將5個(gè)****pdf合并成10頁。這里讀取器和寫入器應(yīng)該怎么配合呢？

邏輯如下：

1、讀取器將所有pdf讀取一遍

2、讀取器將讀取的內(nèi)容交給寫入器

3、寫入器統(tǒng)一輸出到一個(gè)新pdf

這里還有一個(gè)重要的知識點(diǎn)：讀取器只能將讀取的內(nèi)容一頁一頁交給寫入器。

因此，邏輯中第1步和第2步實(shí)際上不是彼此獨(dú)立的步驟，而是讀取器讀取完一個(gè)pdf后，就將這個(gè)pdf全部頁循環(huán)一遍，挨頁交給寫入器。最后等讀取工作全部結(jié)束后再輸出。

看一下代碼可以讓思路更清楚：

from PyPDF2 import PdfFileReader, PdfFileWriterpath = 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)

由于全部內(nèi)容都需要交給同一個(gè)寫入器最后一起輸出，所以寫入器的初始化一定是在循環(huán)體之外的.

如果在循環(huán)體內(nèi)則會(huì)變成每次訪問讀取一個(gè)pdf就生成一個(gè)新的寫入器，這樣每一個(gè)讀取器交給寫入器的內(nèi)容就會(huì)被反復(fù)覆蓋，無法實(shí)現(xiàn)我們的合并需求!

循環(huán)體開頭的代碼：

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

目的就是每次循環(huán)讀取一個(gè)新的pdf文件交給讀取器進(jìn)行后續(xù)操作。實(shí)際上這種寫法不是很提倡，由于各pdf命名恰好很規(guī)則，所以可以直接人為指定數(shù)字進(jìn)行循環(huán)。更好的方法是用 glob 模塊：

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

代碼中 pdf_reader.getNumPages(): 能夠獲取讀取器的頁數(shù)，配合range就能遍歷讀取器的所有頁。

pdf_writer.addPage(pdf_reader.getPage(page))能夠?qū)?dāng)前頁交給寫入器。

最后，用with新建一個(gè)pdf并由寫入器的 pdf_writer.write(out)方法輸出即可

拆分

如果明白了合并操作中讀取器和寫入器的配合，那么拆分就很好理解了，這里我們以拆分INV1.pdf為2個(gè)單獨(dú)的pdf文檔為例，同樣也先來捋一捋邏輯：

1、讀取器讀取PDF文檔

2、讀取器一頁一頁交給寫入器

3、寫入器每獲取一頁就立即輸出

通過這個(gè)代碼邏輯我們也可以明白，寫入器初始化和輸出的位置一定都在讀取PDF循環(huán)每一頁的循環(huán)體內(nèi)，而不是在循環(huán)體外

代碼很簡單：

from PyPDF2 import PdfFileReader, PdfFileWriterpath = r'C:\Users\xxx'pdf_reader = PdfFileReader(path + '\INV1.pdf')for page in range(pdf_reader.getNumPages()):# 遍歷到每一頁挨個(gè)生成寫入器    pdf_writer = PdfFileWriter()    pdf_writer.addPage(pdf_reader.getPage(page))# 寫入器被添加一頁后立即輸出產(chǎn)生pdfwith open(path + '\INV1-{}.pdf'.format(page + 1), 'wb') as out:        pdf_writer.write(out)

水印

本次的工作是將下圖作為水印添加到INV1.pdf中

首先是準(zhǔn)備工作，將需要作為水印的圖片插入word中調(diào)整合適位置后保存為PDF文件。然后就可以碼代碼了，需要額外用到copy模塊，具體解釋見下圖：

就是把讀取器和寫入器初始化，并且把水印PDF頁先讀取好備用,核心代碼稍微比較難理解：

加水印本質(zhì)上就是把水印PDF頁和需要加水印的每一頁都合并一遍

由于需要加水印的PDF可能有很多頁，而水印PDF只有一頁，因此如果直接把水印PDF拿來合并，可以抽象理解成加完第一頁，水印PDF頁就沒有了。

因此不能直接拿來合并，而要把水印PDF頁不斷copy出來成新的一頁備用new_page，再運(yùn)用.mergePage方法完成跟每一頁合并，把合并后的頁交給寫入器待最后統(tǒng)一輸出!

關(guān)于.mergePage的使用：出現(xiàn)在下面的頁.mergePage(出現(xiàn)在上面的頁)，最后效果如圖：

加密

加密很簡單，只需要記?。骸讣用苁轻槍懭肫骷用堋?/p>

因此只需要在相關(guān)操作完成后調(diào)用pdf_writer.encrypt(密碼)

以單個(gè)PDF的加密為例：

當(dāng)然除了對PDF的合并、拆分、加密、水印，我們還可以使用Python結(jié)合Excel和Word實(shí)現(xiàn)更多的自動(dòng)化需求，這些就留給讀者自己開發(fā)。