python之pdf合并

有一个朋友向我提到了pdf合并的事情。在我印象中python可以处理pdf。于是故事就开始了。下面这是搬砖行为。
参考链接:Python之合并PDF文件

1. 输入文件

2. python代码

import os
from PyPDF2 import PdfFileReader, PdfFileWriter

# 参数       文件路径  输出文件名
def pdfUnit(filePath,outPdfName):
    totalFile=os.listdir(filePath)
    output = PdfFileWriter()
    outputPages=0
    for each in totalFile:
        eachFilePath=os.path.join(filePath, each)
        inputPdf = PdfFileReader(eachFilePath, "rb")
        # 获得源PDF文件中页面总数
        pageNumber = inputPdf.getNumPages()
        outputPages += pageNumber
        print("页数：%d"%pageNumber)
        # 分别将page添加到输出output中
        for index in range(pageNumber):
            output.addPage(inputPdf.getPage(index))
    outputStream = open(os.path.join(filePath, outPdfName), "wb")
    output.write(outputStream)
    outputStream.close()

filePath=r"C:\Users\xxx\Desktop\pdf"
pdfUnit(filePath,'total.pdf')

3. 输出结果

下一篇文章：python之pdf分页