import os
import shutil

def delete_missing_folders(old_folder, new_folder):
    # 获取旧文件夹和新文件夹中的子文件夹列表
    old_folders = set(os.listdir(old_folder))
    new_folders = set(os.listdir(new_folder))

    # 找出在新文件夹中但不在旧文件夹中的子文件夹
    missing_folders = new_folders - old_folders

    # 删除缺失的文件夹
    for folder in missing_folders:
        folder_path = os.path.join(new_folder, folder)
        if os.path.isdir(folder_path):
            try:
                shutil.rmtree(folder_path)  # 递归删除文件夹
                print(f"Deleted: {folder_path}")
            except Exception as e:
                print(f"Error deleting {folder_path}: {e}")

def main():
    old_folder = '/data_160TB/2021/yangkun/gha-gaussian/datasets/eg3d_data/data_new/train'
    new_folder = '/data_160TB/2021/yangkun/gha-gaussian/datasets/eg3d_data/data_new_down/train'

    delete_missing_folders(old_folder, new_folder)

if __name__ == "__main__":
    main()