pdf合并小工具python

前言

使用python写了个合并pdf的小工具，共享代码
相对比较简陋，将就着看吧，需要的可以自己再润色一下

代码

from os import rename
from glob import glob
from PyPDF2 import PdfFileReader,PdfFileWriter

fileNameLen = 8

def reName(pdfNames):
    for i in range(len(pdfNames)):
        newName = pdfNames[i].rjust(fileNameLen, '0')
        rename(pdfNames[i], newName)
        print("name: " + pdfNames[i] + "new name: " + newName)

    return pdfNames


def getFile():
    pdfNames = glob("*.pdf")
    print(pdfNames)
    return pdfNames


def mergePdfFile():
    outPut = PdfFileWriter()
    outputPages = 0

    pdfNames = getFile()
    reName(pdfNames)
    pdfNames = getFile()
    for fileName in pdfNames:
        p = fileName
        inputpdf = PdfFileReader(p)
        pageCount = inputpdf.getNumPages()
        outputPages += pageCount
        for pageNum in range(pageCount):
            outPut.addPage(inputpdf.getPage(pageNum))

    print("All page Number is: " + outputPages.__str__())

    outPutStream = open("output.pdf", "wb")
    outPut.write(outPutStream)
    outPutStream.close()


if __name__ == '__main__':
    mergePdfFile()