python合并两个文件夹中相同文件名的PDF文件

import os
from PyPDF2 import PdfMerger

def merge_pdfs(folder1, folder2, output_folder):
    # 获取文件夹中的所有PDF文件
    files1 = set(os.listdir(folder1))
    files2 = set(os.listdir(folder2))

    # 获取两个文件夹中相同文件名的PDF文件
    common_files = files1.intersection(files2)

    # 合并PDF文件
    for file in common_files:
        if file.endswith('.pdf'):
            pdf1_path = os.path.join(folder1, file)
            pdf2_path = os.path.join(folder2, file)

            # 判断文件是否存在于两个文件夹中
            if os.path.exists(pdf1_path) and os.path.exists(pdf2_path):
                merger = PdfMerger()

                # 合并第一个文件夹中的PDF内容在前
                merger.append(pdf1_path)
                merger.append(pdf2_path)

                # 输出到指定的第三个文件夹
                output_file = os.path.join(output_folder, file)
                merger.write(output_file)
                merger.close()
                print(f'合并成功: {file}')
            else:
                print(f'文件丢失: {file}')

# 文件地址
folder1 = r'D:\'
folder2 = r'D:\'
output_folder = r'D:\'

# 确保输出文件夹存在
if not os.path.exists(output_folder):
    os.makedirs(output_folder)

merge_pdfs(folder1, folder2, output_folder)