"""
Author: Fanxing Shang
Date: 2023-1-10
   //////////////////// __1号脚本__  ->  'test01.py'//////////////////////

    __用途__  ->  用于将一个文件夹中所有图片的格式都变成指定格式类型(.jpg/可选)
"""

from PIL import Image
import os

def convert_to_jpg(input_folder, output_folder):
    # 遍历指定文件夹中的所有文件
    for filename in os.listdir(input_folder):
        # 判断文件是否为图片格式（文件夹中需要被转化的图片格式类型，通常会写入所有的图片格式类型）   # 需要修改的地方(1)
        if any(filename.endswith(extension) for extension in ['.png', '.jpeg', '.gif', '.bmp', 'jpg']):
            # 构造输入和输出文件的完整路径
            input_path = os.path.join(input_folder, filename)
            output_path = os.path.join(output_folder, os.path.splitext(filename)[0] + '.jpg')  #写入要被转换的类型的后缀名 .xxx
                                                                                                # 需要修改的地方(2)
            # 打开图片并保存为jpg格式
            image = Image.open(input_path)
            image.convert('RGB').save(output_path, 'JPEG')  #写入要被转换的类型的后缀名(大写，注意：除了jpg和gpeg的都要写JPEG外，其他的直接转换成大写)
                                                                    # 需要修改的地方(3)
# 指定需要转换的文件夹和输出的文件夹

if __name__ == "__main__":
    # 调用函数进行转换
    input_folder = input("请输入源文件地址")
    output_folder = input("请输入输出文件地址")
    convert_to_jpg(input_folder, output_folder)