PDF合成

-- coding: utf-8 --

“”“Spyder EditorThis is a temporary script file.”""import osfrom PyPDF2 import PdfFileMergerfrom PyPDF2 import PdfFileReader#获取pdf页数的函数def getPdfPages(filePath): reader = PdfFileReader(filePath) #解密pdf文件 if reader.isEncrypted: reader.decrypt(’’) pageNum = reader.getNumPages() return pageNum#存储所有pdf文件路径的函数def loadAllFilesPath(rootPath,filePaths): #分别代表根目录、文件夹、文件 for root, dirs, files in os.walk(rootPath): #遍历文件 for file in files: #识别pdf文件 if file.endswith(’.pdf’) or file.endswith(’.PDF’): #获取文件绝对路径 filePath = os.path.join(root, file) #文件路径添加进列表 filePaths.append(filePath)#输入根目录的文件夹rootPath = input(‘请输入文件夹：’)#存储所有pdf文件路径的列表pdfPaths = []#存储所有pdf文件的页码和索引pageCount = 0pageIndex = []#找到所有pdf文件路径，并存储所有pdf文件路径到列表中loadAllFilesPath(rootPath,pdfPaths)#创建pdf合成器对象fileMerger = PdfFileMerger()#合并pdf文件for pdfPath in pdfPaths: fileMerger.append(pdfPath) page = getPdfPages(pdfPath) pageIndex.append(‘页码索引：’ + str(pageCount+1) + ‘~’ + str(pageCount+page)) pageCount += page#创建保存新文件的文件夹newDirPath = os.path.join(rootPath,‘合并完成的文件’)if not os.path.exists(newDirPath): os.mkdir(newDirPath)#保存合并完成的pdf文件fileMerger.write(os.path.join(newDirPath,‘合并完成的pdf文件.pdf’))#保存合并pdf文件的顺序with open(os.path.join(newDirPath,‘合并pdf文件的顺序.txt’),‘w’,encoding=‘utf-8’) as f: for pdfPath,pdfInd in zip(pdfPaths,pageIndex): f.write(pdfPath + ‘\t\t’ + pdfInd + ‘\n’)#pdf文件合并成功！print(‘pdf文件合并成功，请查看文件夹“合并完成的文件”’)