import shutil
preb_base = 'G:/Tech_Animation/data_dump/Prebuilt_Dynamics_Sets'+'/'+os.environ['tt_projcode']

for asset in os.listdir(preb_base):
    asset_elem = 'Y:/{}/assets/type/Character/{}/work/elems/'.format(os.environ['tt_projcode'],asset)
    if os.path.isdir(asset_elem):
        if not os.path.isdir(asset_elem+'/TA_obj_sets/Prebuilt_Dynamics_Sets'):
                    os.makedirs( asset_elem+'/TA_obj_sets/Prebuilt_Dynamics_Sets')
        for set_type in ['Cloth','Hair']:
            if set_type in os.listdir(preb_base+'/'+asset):
                shutil.copytree(preb_base+'/'+asset+'/'+set_type,asset_elem+'/TA_obj_sets/Prebuilt_Dynamics_Sets'+'/'+set_type)
                
##############################################################

import shutil
import json
import ast
preb_base = 'G:/Tech_Animation/data_dump/Prebuilt_Dynamics_Sets'+'/'+os.environ['tt_projcode']

for asset in os.listdir(preb_base):
    asset_elem = 'Y:/{}/assets/type/Character/{}/work/elems/'.format(os.environ['tt_projcode'],asset)
    if os.path.isdir(asset_elem):
        if not os.path.isdir(asset_elem+'/TA_obj_sets/Prebuilt_Dynamics_Sets'):
            os.makedirs( asset_elem+'/TA_obj_sets/Prebuilt_Dynamics_Sets')
    
        sets_by_tags,obj_dict = preb_export_sets_ui_funcs.SetQuery.get_set_dicts(ass_list = [asset],parse_data = True,use_cmdsls=False)
        for sim_type in obj_dict[asset]:
            for set_ in obj_dict[asset][sim_type]:
                obj_dict[asset][sim_type][set_] = {'objects':[obj_ .split(':')[-1] for obj_ in obj_dict[asset][sim_type][set_]]}
                type_detect = [a for a in sets_by_tags[asset][sim_type] if set_ in sets_by_tags[asset][sim_type][a]][0]
                obj_dict[asset][sim_type][set_]['type'] = type_detect
        
                jsn_file_path = asset_elem+'/TA_obj_sets/Prebuilt_Dynamics_Sets'+'/'+sim_type+'.json'
                with open(jsn_file_path,'w') as set_json:
                    json.dump(obj_dict[asset][sim_type],set_json)