电竞比分网-中国电竞赛事及体育赛事平台

分享

Python辦公自動(dòng)化|批量合并PDF,拿來(lái)就用

 liqualife 2020-05-07

第一時(shí)間接收最新Python干貨!

早起Python推薦搜索
Python
Pandas
數(shù)據(jù)分析

系列導(dǎo)讀

Python辦公自動(dòng)化|從Word到Excel
Python辦公自動(dòng)化|從Excel到Word

大家好,今天分享一個(gè)實(shí)用的辦公腳本:將多個(gè)PDF合并為一個(gè)PDF,例如我手上現(xiàn)在有如下3個(gè)PDF分冊(cè),需要整合成一個(gè)完整的PDF

如果換成你操作的話,是不是打開百度搜索:PDF合并,然后去第三方網(wǎng)站操作,可能會(huì)收費(fèi)不說(shuō)還擔(dān)心文件泄漏,現(xiàn)在有請(qǐng)Python出場(chǎng),簡(jiǎn)單快速,光速合并,拿走就用!

首先導(dǎo)入需要的庫(kù)和路徑設(shè)置

import os
from PyPDF2 import PdfFileReader, PdfFileWriter
if __name__ == '__main__':
    # 設(shè)置存放多個(gè)pdf文件的文件夾
    dir_path = r'C:\Scientific Research\Knowladge\Ophthalmology\Chinese Ophthalmology' 
    # 目標(biāo)文件的名字
    file_name = '中華眼科學(xué)(第3版)合并版.pdf' 

接著獲取所有pdf文件的絕對(duì)路徑,這里需要利用os庫(kù)中的os.walk遍歷文件和os.path.join拼接路徑

for dirpath, dirs, files in os.walk(dir_path):
    print(dirpath)
    print(files)
# 結(jié)果返回當(dāng)前路徑、當(dāng)前路徑下文件夾,并以列表返回所有文件

建議直接將需要合并的pdf放在一個(gè)文件夾,這樣就無(wú)需再對(duì)文件后綴進(jìn)行判斷,包裝成函數(shù)后如下:

def GetFileName(dir_path):
    file_list = [os.path.join(dirpath, filesname) \
                 for dirpath, dirs, files in os.walk(dir_path) \
                 for filesname in files]
    return file_list
調(diào)用該函數(shù)的結(jié)果


現(xiàn)在建立合并PDF的函數(shù)

def MergePDF(dir_path, file_name):
    # 實(shí)例化寫入對(duì)象
    output = PdfFileWriter()
    outputPages = 0
    # 調(diào)用上一個(gè)函數(shù)獲取全部文件的絕對(duì)路徑
    file_list = GetFileName(dir_path)

    for pdf_file in file_list:
        print('文件:%s' % pdf_file.split('\\')[-1], end=' ')

        # 讀取PDF文件
        input = PdfFileReader(open(pdf_file, 'rb'))
        # 獲得源PDF文件中頁(yè)面總數(shù)
        pageCount = input.getNumPages()
        outputPages += pageCount
        print('頁(yè)數(shù):%d' % pageCount)
        # 分別將page添加到輸出output中
        for iPage in range(pageCount):
            output.addPage(input.getPage(iPage))
    print('\n合并后的總頁(yè)數(shù):%d' % outputPages)
    # 寫入到目標(biāo)PDF文件
    print('PDF文件正在合并,請(qǐng)稍等......')
    with open(os.path.join(dir_path, file_name), 'wb'as outputfile:
        # 注意這里的寫法和正常的上下文文件寫入是相反的
        output.write(outputfile)
    print('PDF文件合并完成')

可以看到雖然待合并的PDF文件比較大,但是依舊快速的合并成功!


最后附上完整代碼,只需將代碼中PDF的路徑和文件名修改即可使用!

import os
from PyPDF2 import PdfFileReader, PdfFileWriter

def GetFileName(dir_path):
    file_list = [os.path.join(dirpath, filesname) \
                 for dirpath, dirs, files in os.walk(dir_path) \
                 for filesname in files]
    return file_list

def MergePDF(dir_path, file_name):
    output = PdfFileWriter()
    outputPages = 0
    file_list = GetFileName(dir_path)
    for pdf_file in file_list:
        print('文件:%s' % pdf_file.split('\\')[-1], end=' ')
        # 讀取PDF文件
        input = PdfFileReader(open(pdf_file, 'rb'))
        # 獲得源PDF文件中頁(yè)面總數(shù)
        pageCount = input.getNumPages()
        outputPages += pageCount
        print('頁(yè)數(shù):%d' % pageCount)
        # 分別將page添加到輸出output中
        for iPage in range(pageCount):
            output.addPage(input.getPage(iPage))
    print('\n合并后的總頁(yè)數(shù):%d' % outputPages)
    # 寫入到目標(biāo)PDF文件
    print('PDF文件正在合并,請(qǐng)稍等......')
    with open(os.path.join(dir_path, file_name), 'wb'as outputfile:
        # 注意這里的寫法和正常的上下文文件寫入是相反的
        output.write(outputfile)
    print('PDF文件合并完成')

if __name__ == '__main__':
    # 設(shè)置存放多個(gè)pdf文件的文件夾
    dir_path = r'C:\Scientific Research\Knowladge\Ophthalmology\Chinese Ophthalmology'
    # 目標(biāo)文件的名字
    file_name = '中華眼科學(xué)(第3版)合并版.pdf'
    MergePDF(dir_path, file_name)

    本站是提供個(gè)人知識(shí)管理的網(wǎng)絡(luò)存儲(chǔ)空間,所有內(nèi)容均由用戶發(fā)布,不代表本站觀點(diǎn)。請(qǐng)注意甄別內(nèi)容中的聯(lián)系方式、誘導(dǎo)購(gòu)買等信息,謹(jǐn)防詐騙。如發(fā)現(xiàn)有害或侵權(quán)內(nèi)容,請(qǐng)點(diǎn)擊一鍵舉報(bào)。
    轉(zhuǎn)藏 分享 獻(xiàn)花(0

    0條評(píng)論

    發(fā)表

    請(qǐng)遵守用戶 評(píng)論公約

    類似文章 更多