图片转pdf脚本

from PIL import Image
import os


def process_images_and_create_pdf(folder_list, output_pdf):
    all_image_paths = []

    # 为每个文件夹创建一个字典，用于存储该文件夹内的文件路径
    folder_files_dict = {folder: [] for folder in folder_list}

    # 依次访问每个文件夹
    for folder in folder_list:
        for root, dirs, files in os.walk(folder):
            for file in files:
                if file.endswith('.jpg'):
                    file_path = os.path.join(root, file)
                    folder_files_dict[folder].append(file_path)

    # 对每个文件夹内的文件路径进行排序
    for folder in folder_files_dict:
        folder_files_dict[folder].sort()

    # 将所有文件夹内排序好的文件路径按照文件夹顺序添加到总的列表中
    for folder in folder_list:
        all_image_paths.extend(folder_files_dict[folder])

    # 打印序号和文件路径
    for i, path in enumerate(all_image_paths):
        print(f"{i + 1}: {path}")

    images = []

    # 打开并处理图片（裁掉底下30像素）
    for path in all_image_paths:
        img = Image.open(path)
        width, height = img.size
        cropped_img = img.crop((0, 0, width, height - 30))
        cropped_img = cropped_img.convert('RGB')
        images.append(cropped_img)

    # 保存为PDF文件
    images[0].save(output_pdf, save_all=True, append_images=images[1:])


if __name__ == "__main__":
    folder_list = ['封面', '上', '下']  # 替换为实际的文件夹名称
    output_pdf = 'output.pdf'  # 替换为你想要的输出PDF文件名
    process_images_and_create_pdf(folder_list, output_pdf)