from  PyPDF2 import PdfFileReader,PdfFileWriter
import os

# 合并PDF
def Merge_pdf(src_dir,pdf_list,out):
    if src_dir and os.path.isdir(src_dir):
        if pdf_list:
            merge = PdfFileWriter()
            for pdf in pdf_list:
                pdf_path = os.path.join(src_dir,pdf)
                print(pdf_path)
                if os.path.isfile(pdf_path):
                    pdf_reader = PdfFileReader(pdf_path)
                    for i in range(0,pdf_reader.getNumPages()):
                        merge.addPage(pdf_reader.getPage(i))
        else:
            merge = PdfFileWriter()
            for pdf in os.listdir(src_dir):
                if pdf.endswith('.pdf'):
                    pdf_path = os.path.join(src_dir,pdf)
                    print(pdf_path)
                    if os.path.isfile(pdf_path):
                        pdf_reader = PdfFileReader(pdf_path)
                        for i in range(0,pdf_reader.getNumPages()):
                            merge.addPage(pdf_reader.getPage(i))
                    
                        
            with open(os.path.join(src_dir,out),'wb') as f:
                merge.write(f)
                    
                    
src_dir = r'D:\工作\02.学习\书籍\pandas'
# pdf_list = ['Categorical data — pandas 2.0.1 documentation.pdf','Chart visualization — pandas 2.0.1 documentation.pdf']
out = 'Merge.pdf'
pdf_list = []
Merge_pdf(src_dir, pdf_list, out)