# -*- coding:utf-8 -*-
# @Author len
# @Create 2023/10/28 19:08

import os
from PIL import Image
import glob

# 设置包含图片文件的目录和存储转换后的 JPEG 文件的目录
image_directory = r'E:\Waste\嵌入式\data\遴选赛\第二轮TFTA'
jpeg_directory = r'E:\Waste\嵌入式\data\遴选赛\第二轮TFTA'

# 如果 jpeg 存储目录不存在，则创建它
if not os.path.exists(jpeg_directory):
    os.makedirs(jpeg_directory)

# 在目录中查找所有 PNG 和 BMP 文件
image_files = glob.glob(os.path.join(image_directory, '*.png')) + glob.glob(os.path.join(image_directory, '*.bmp'))

# 遍历所有找到的图片文件并将它们转换为 JPEG
for image_file in image_files:
    # 加载图片
    img = Image.open(image_file)

    # 将图片转换为 RGB 模式，这是因为 JPEG 不支持透明通道
    img = img.convert('RGB')

    # 定义输出文件的路径
    base_filename = os.path.basename(image_file)
    jpeg_filename = os.path.splitext(base_filename)[0] + '.jpg'
    jpeg_file_path = os.path.join(jpeg_directory, jpeg_filename)

    # 将图片保存为 JPEG 格式
    img.save(jpeg_file_path)
    print(f"已将 {image_file} 转换为 {jpeg_file_path}")

print("批量转换完成。")