如何将同一文件夹下所有 pdf 文件合并,并按照规定将合成后的文件命名
# 使用os模块walk函数,搜索出某目录下的全部pdf文件
###获取同一个文件夹下的所有PDF文件名###
import os
import os.path
from PyPDF2 import PdfFileReader, PdfFileWriter
import time
from os import listdir
def getFileName(filepath):
file_list = []
for root, dirs, files in os.walk(filepath):
for filespath in files:
# print(os.path.join(root,filespath))
print(filespath)
file_list.append(os.path.join(root, filespath))
print(file_list)
return file_list
#合并同一个文件夹下所有PDF文件
def MergePDF(filepath, outfile):
output = PdfFileWriter()
outputPages = 0
pdf_fileName = getFileName(filepath)
for each in pdf_fileName:
print(each)
print('-=-='* 30)
# 读取源pdf文件
input = PdfFileReader(open(each, "rb"))
# 获得源pdf文件中页面总数
pageCount = input.getNumPages()
outputPages += pageCount
print(pageCount)
# 分别将page添加到输出output中
for iPage in range(0, pageCount):
output.addPage(input.getPage(iPage))
print("All Pages Number:" + str(outputPages))
# 最后写pdf文件
outputStream = open(filepath + outfile, "wb")
output.write(outputStream)
outputStream.close()
print("finished")
if __name__ == '__main__':
file = '%s\\' % dir
today = datetime.date.today()
first = today.replace(day=1)
last_month = first - datetime.timedelta(days=1)
data = str(last_month)[0:7].replace('-','')
print(str(last_month)[0:7])
out = i['门店编号'] +"-"+ i['店铺名称']+u"-draft return-%s.pdf" % data
MergePDF(file, out)
在做普华永道项目时,其中有一个需求就是填报完毕后,需要针对每个门店下载多种报表,并最后需要合并到一张 pdf 文件。(命名规则是:门店编号 +’- 店铺名称‘+’-draft return’+ 上个月的时间)这里的门店编号与店铺名称是从 Excel 表中读取的。这里需要合并的文件夹也并未写死,会针对每个门店所对应门店号下的文件夹。
注:这里的合成会按照文件名称排序进行合并的,若需求需要安照指定的顺序进行合并,可以在流程中针对文件先按照 00,01,02 这样先命好名,这样合并后的文件,内容也会首先显示的是 00.