import os
import zipfile

def zip_subdirectories(parent_directory, target_directory):
    if not os.path.exists(target_directory):
        os.mkdir(target_directory)
    for foldername, subfolders, filenames in os.walk(parent_directory):
        # 对每个直接子目录进行操作
        for subfolder in subfolders:
            zip_path = os.path.join(target_directory, f"{subfolder}.zip")
            # 创建Zip文件
            with zipfile.ZipFile(zip_path, 'w') as zipf:
                subfolder_path = os.path.join(foldername, subfolder)
                # 遍历子目录下的所有文件和子目录
                for root, _, files in os.walk(subfolder_path):
                    for file in files:
                        # 获取文件的绝对路径
                        file_path = os.path.join(root, file)
                        # 计算文件在zip文件中的路径
                        arcname = os.path.join(subfolder, os.path.relpath(file_path, start=subfolder_path))
                        # 将文件添加到zip文件中
                        zipf.write(file_path, arcname)
        # 由于os.walk()也会递归地访问子目录，我们可以跳过后续的遍历
        break

# 调用函数
zip_subdirectories('/mnt/petrelfs/zhuchenglin/LLaVA/playground/data/eval/gqa/data', '/mnt/hwfile/zhuchenglin/playground/eval/gqa/data')