利用PyPDF2模块合并同一文件夹下的所有PDF文件 

利用PyPDF2模块合并同一文件夹下的所有PDF文件
 1 # -*- coding:utf-8*-
 2 # 利用PyPDF2模块合并同一文件夹下的所有PDF文件
 3 # 只需修改存放PDF文件的文件夹变量：file_dir（第56行） 和 输出文件名变量: outfile （第57行）
 4 
 5 import os
 6 from PyPDF2 import PdfFileReader, PdfFileWriter
 7 import time
 8 
 9 # 使用os模块的walk函数，搜索出指定目录下的全部PDF文件
10 # 获取同一目录下的所有PDF文件的绝对路径
11 def getFileName(filedir):
12 
13     file_list = [os.path.join(root, filespath) \
14                  for root, dirs, files in os.walk(filedir) \
15                  for filespath in files \
16                  if str(filespath).endswith('pdf')
17                  ]
18     return file_list if file_list else []
19 
20 # 合并同一目录下的所有PDF文件
21 def MergePDF(filepath, outfile):
22 
23     output = PdfFileWriter()
24     outputPages = 0
25     pdf_fileName = getFileName(filepath)
26 
27     if pdf_fileName:
28         for pdf_file in pdf_fileName:
29             print("路径：%s"%pdf_file)
30 
31             # 读取源PDF文件
32             input = PdfFileReader(open(pdf_file, "rb"))
33 
34             # 获得源PDF文件中页面总数
35             pageCount = input.getNumPages()
36             outputPages += pageCount
37             print("页数：%d"%pageCount)
38 
39             # 分别将page添加到输出output中
40             for iPage in range(pageCount):
41                 output.addPage(input.getPage(iPage))
42 
43         print("合并后的总页数:%d."%outputPages)
44         # 写入到目标PDF文件
45         outputStream = open(os.path.join(filepath, outfile), "wb")
46         output.write(outputStream)
47         outputStream.close()
48         print("PDF文件合并完成！")
49 
50     else:
51         print("没有可以合并的PDF文件！")
52 
53 # 主函数
54 def main():
55     time1 = time.time()
56     file_dir = r'D:\data3' # 存放PDF的原文件夹
57     outfile = "Cheat_Sheets.pdf" # 输出的PDF文件的名称
58     MergePDF(file_dir, outfile)
59     time2 = time.time()
60     print('总共耗时：%s s.' %(time2 - time1))
61 
62 main()

 

posted @ 2020-11-04 10:37  ken-yu  阅读(400)  评论(0)    收藏  举报