Spaces:
Sleeping
Sleeping
File size: 2,280 Bytes
520da1b | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 | # check all json file in results_path
import json, os
results_path = './results/'
target_path = './results_m/'
# remove target_path
if os.path.exists(target_path):
os.system(f'rm -r {target_path}')
if not os.path.exists(target_path):
os.makedirs(target_path)
available_list = ['image_content', 'topic', 'data_facts', 'color_style', 'icon_or_clipart']
data_fact_list = ['increasing', 'decreasing', 'highlight', 'deny', 'maximum', 'minimum', 'comparison', 'none']
color_style_list = ['monochrome', 'colorful', 'grayscale', 'dual-color']
scale_and_complexity_level_list = ['icon', 'clipart', 'background']
for i, file in enumerate(os.listdir(results_path)):
if not file.endswith('.json'):
continue
# load json
with open(f'{results_path}{file}', 'r') as f:
data = json.load(f)
if 'output_case' in data:
data = data['output_case']
if 'Topic' in data:
data['topic'] = data['Topic']
del data['Topic']
if 'key_words' in data:
data['topic'] = data['key_words']
del data['key_words']
# key error
error = False
for key in available_list:
if key not in data:
error = True
break
if error:
print(f'{file} is not complete')
continue
# data_facts error
if data['data_facts'] == 'increase':
data['data_facts'] = 'increasing'
if data['data_facts'] not in data_fact_list:
print(f'{file} data_facts is error')
continue
# color_style error
if data['color_style'] == 'gray-scale' or data['color_style'] == 'black and white':
data['color_style'] = 'grayscale'
if data['color_style'] == 'duo-color' or data['color_style'] == 'duel-color':
data['color_style'] = 'dual-color'
if data['color_style'] not in color_style_list:
print(f'{file} color_style is error')
continue
# scale_and_complexity_level error
if data['icon_or_clipart'] not in scale_and_complexity_level_list:
print(f'{file} icon_or_clipart is error')
continue
data['size'] = data['icon_or_clipart']
del data['icon_or_clipart']
# save json
target_file = f'{target_path}{file}'
with open(target_file, 'w') as f:
json.dump(data, f) |