import os
from PIL import Image
import pillow_heif

# 注册 HEIF 支持
pillow_heif.register_heif_opener()

# 获取当前目录
current_dir = os.getcwd()

for filename in os.listdir(current_dir):
    if filename.lower().endswith(('.heic', '.heif')):
        try:
            # 打开 HEIC 文件
            heif_file = pillow_heif.open_heif(filename)
            image = Image.frombytes(
                heif_file.mode, 
                heif_file.size, 
                heif_file.data,
                "raw",
                heif_file.mode,
                heif_file.stride,
            )
            
            # 保存为 PNG
            output_name = os.path.splitext(filename)[0] + '.png'
            image.save(output_name, 'PNG')
            print(f"已转换: {filename} -> {output_name}")
        except Exception as e:
            print(f"转换失败 {filename}: {str(e)}")

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