diff --git a/opponents/copy_event_alts.py b/opponents/copy_event_alts.py index 22fd455663dc36fe414d3472eb3c867390f3987a..d37a64bf2d6bb296f6840bd4d9ed5f8a1accab55 100644 --- a/opponents/copy_event_alts.py +++ b/opponents/copy_event_alts.py @@ -35,33 +35,33 @@ def main(): delete_stages[opp.text] = list(range(n_stages)) opp_stage_counts[opp.text] = n_stages - for opp_id, costume_folder in event_costumes: - opp_base_folder = 'opponents/'+opp_id+'/' - costume_xml = ET.parse(osp.join(costume_folder, 'costume.xml')) - - # Read all folders referenced by the costume (sorted by stage) - folders = [] - for folder in costume_xml.iter('folder'): - folder_stage = int(folder.get('stage', 0)) - folders.append((folder_stage, folder.text)) - folders = sorted(folders, key=lambda t: t[0]) + #for opp_id, costume_folder in event_costumes: + # opp_base_folder = 'opponents/'+opp_id+'/' + # costume_xml = ET.parse(osp.join(costume_folder, 'costume.xml')) + # + # # Read all folders referenced by the costume (sorted by stage) + # folders = [] + # for folder in costume_xml.iter('folder'): + # folder_stage = int(folder.get('stage', 0)) + # folders.append((folder_stage, folder.text)) + # folders = sorted(folders, key=lambda t: t[0]) - # Figure out which costume stages reference the base folder. - # Remove those from delete_stages. - remove_set = set() - for stage in delete_stages[opp_id]: - best_path = folders[0][1] - for folder_stage, folder_path in folders: - if folder_stage > stage: - break + # # Figure out which costume stages reference the base folder. + # # Remove those from delete_stages. + # remove_set = set() + # for stage in delete_stages[opp_id]: + # best_path = folders[0][1] + # for folder_stage, folder_path in folders: + # if folder_stage > stage: + # break - best_path = folder_path - - if best_path == opp_base_folder: - remove_set.add(stage) + # best_path = folder_path + # + # if best_path == opp_base_folder: + # remove_set.add(stage) - for stage in remove_set: - delete_stages[opp_id].remove(stage) + # for stage in remove_set: + # delete_stages[opp_id].remove(stage) # Make the destination path for reskins. os.mkdir(osp.join(base_dir, 'opponents', 'reskins')) @@ -75,26 +75,26 @@ def main(): sh.copytree(src, dst) # Delete previously-determined stages in base folders: - for opp_id, stages in delete_stages.items(): - if opp_id == 'chiaki': - continue + #for opp_id, stages in delete_stages.items(): + # if opp_id == 'chiaki': + # continue - print("Deleting {} stages {}...".format(opp_id, str(stages))) + # print("Deleting {} stages {}...".format(opp_id, str(stages))) - n_deleted = 0 - for fname in os.listdir(osp.join(base_dir, 'opponents', opp_id)): - posename = osp.splitext(fname)[0] + # n_deleted = 0 + # for fname in os.listdir(osp.join(base_dir, 'opponents', opp_id)): + # posename = osp.splitext(fname)[0] - try: - stage = int(posename.split('-')[0]) - except ValueError: - continue + # try: + # stage = int(posename.split('-')[0]) + # except ValueError: + # continue - if stage in stages: - os.unlink(osp.join(base_dir, 'opponents', opp_id, fname)) - n_deleted += 1 + # if stage in stages: + # os.unlink(osp.join(base_dir, 'opponents', opp_id, fname)) + # n_deleted += 1 - print("Deleted {} images from {}.".format(n_deleted, opp_id)) + # print("Deleted {} images from {}.".format(n_deleted, opp_id)) if __name__ == '__main__': main()