import os
from PyPDF2 import PdfFileReader, PdfFileWriter

# 使用os模塊的walk函數(shù)，搜索出指定目錄下的全部PDF文件
# 獲取同一目錄下的所有PDF文件的絕對路徑
def getFileName(filedir):

    file_list = [os.path.join(root, filespath) \
                 for root, dirs, files in os.walk(filedir) \
                 for filespath in files \
                 if str(filespath).endswith('pdf')
                 ]
    return file_list if file_list else []

# 合并同一目錄下的所有PDF文件
def MergePDF(filepath, outfile):

    output = PdfFileWriter()
    outputPages = 0
    pdf_fileName = getFileName(filepath)

    if pdf_fileName:
        for pdf_file in pdf_fileName:
            print("路徑：%s"%pdf_file)

            # 讀取源PDF文件
            input = PdfFileReader(open(pdf_file, "rb"))

            # 獲得源PDF文件中頁面總數(shù)
            pageCount = input.getNumPages()
            outputPages += pageCount
            print("頁數(shù)：%d"%pageCount)

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

        print("合并后的總頁數(shù):%d."%outputPages)
        # 寫入到目標PDF文件
        outputStream = open(os.path.join(filepath, outfile), "wb")
        output.write(outputStream)
        outputStream.close()
        print("PDF文件合并完成！")

    else:
        print("沒有可以合并的PDF文件！")

# 主函數(shù)
def main():
    file_dir = input('請輸入存有Pdf的文件夾').replace('/','//')# 存放PDF的原文件夾
    outfile = "pick_me.pdf" # 輸出的PDF文件的名稱
    MergePDF(file_dir, outfile)
    print('done')

main()