main.py

import os
from PyPDF2 import PdfReader, PdfWriter
from datetime import datetime

def duplicate_pages(input_pdf_path, output_pdf_path):
    reader = PdfReader(input_pdf_path)
    writer = PdfWriter()

    # 遍历每一页，并重复添加到 writer 中
    for page_num in range(len(reader.pages)):
        page = reader.pages[page_num]
        writer.add_page(page)
        writer.add_page(page)

    # 写入到新的 PDF 文件
    with open(output_pdf_path, 'wb') as output_pdf:
        writer.write(output_pdf)

def process_pdf_files(input_folder, output_folder):
    if not os.path.exists(output_folder):
        os.makedirs(output_folder)
    
    for filename in os.listdir(input_folder):
        if filename.endswith(".pdf"):
            input_pdf_path = os.path.join(input_folder, filename)
            timestamp = datetime.now().strftime('%Y%m%d%H%M%S')
            base_filename = os.path.splitext(filename)[0]
            output_pdf_path = os.path.join(output_folder, f"{base_filename}_重复_{timestamp}.pdf")
            duplicate_pages(input_pdf_path, output_pdf_path)
            print(f"Processed {filename} -> {output_pdf_path}")

# 定义输入和输出文件夹
input_folder = 'input'  # 输入文件夹路径
output_folder = 'output'  # 输出文件夹路径

process_pdf_files(input_folder, output_folder)