import os
from PyPDF2 import PdfReader, PdfWriter

def GetFileName(dir_path):
    file_list = [os.path.join(dirpath, filesname) \
                 for dirpath, dirs, files in os.walk(dir_path) \
                 for filesname in files]
    return file_list

def MergePDF(dir_path, file_name):
    output = PdfWriter()
    outputPages = 0
    file_list = GetFileName(dir_path)
    for pdf_file in file_list:
        print("文件：%s" % pdf_file.split('\\')[-1], end=' ')
        # 讀取PDF文件
        input = PdfReader(open(pdf_file, "rb"))
        # 獲得源PDF文件中頁(yè)面總數(shù)
        pageCount = input.getNumPages()
        outputPages += pageCount
        print("頁(yè)數(shù)：%d" % pageCount)
        # 分別將page添加到輸出output中
        for iPage in range(pageCount):
            output.addPage(input.getPage(iPage))
    print("\n合并后的總頁(yè)數(shù):%d" % outputPages)
    # 寫入到目標(biāo)PDF文件
    print("PDF文件正在合并，請(qǐng)稍等......")
    with open(os.path.join(dir_path, file_name), "wb") as outputfile:
        # 注意這里的寫法和正常的上下文文件寫入是相反的
        output.write(outputfile)
    print("PDF文件合并完成")

if __name__ == '__main__':
    # 設(shè)置存放多個(gè)pdf文件的文件夾
    dir_path = r'D:Desktop\新建文件夾 (26)'  
    # 目標(biāo)文件的名字
    file_name = "2023.03.pdf"
    MergePDF(dir_path, file_name)