'''# -*- coding:utf-8*-'''
import sys
import importlib
importlib.reload(sys)

import os
import os.path
from PyPDF2 import PdfFileReader, PdfFileWriter 
import time
time1=time.time()

# 使用os模塊walk函數(shù)，搜索出某目錄下的全部pdf文件
######################獲取同一個文件夾下的所有PDF文件名#######################
def getFileName(filepath):
 file_list = []
 for root,dirs,files in os.walk(filepath):
  for filespath in files:
   # print(os.path.join(root,filespath))
   file_list.append(os.path.join(root,filespath))

 return file_list



##########################合并同一個文件夾下所有PDF文件########################
def MergePDF(filepath,outfile):
 output=PdfFileWriter()
 outputPages=0
 in_file_path =filepath +r'input/'
 print(in_file_path)
 pdf_fileName=getFileName(in_file_path)
 i=0
 for each in pdf_fileName:
  i=i+1
  print(i,each)
  # 讀取源pdf文件
  input = PdfFileReader(open(each, "rb"))

  # 如果pdf文件已經(jīng)加密，必須首先解密才能使用pyPdf
  if input.isEncrypted == True:
   input.decrypt("map")

  # 獲得源pdf文件中頁面總數(shù)
  pageCount = input.getNumPages()
  outputPages += pageCount
  print("PageCount: ",pageCount)

  # 分別將page添加到輸出output中
  for iPage in range(0, pageCount):
   output.addPage(input.getPage(iPage))


 print("All Pages Number:"+str(outputPages))
 # 最后寫pdf文件
 out_file_path=filepath+r'output/'
 outputStream=open(out_file_path+outfile,"wb")
 output.write(outputStream)
 outputStream.close()
 print( "finished")


if __name__ == '__main__':
 file_dir = r'D:/掃描文件/'  # 待合并PDF 所在文件夾
 out=u"某某資料合并文件.pdf"  # 合并后文件名稱
 MergePDF(file_dir,out)
 time2 = time.time()
 print( u'總共耗時：' + str(time2 - time1) + 's')