import json
with open("predictions.json") as f:
api = json.loads(f.read())
api['TT'] = {}
for state in api:
if state == 'TT':
continue
for date in api[state]:
api['TT'][date] = api['TT'].get(date, {'delta':{}, 'total':{}})
for k in ['delta', 'total']:
api['TT'][date][k]['confirmed'] = api['TT'][date][k].get('confirmed', 0) + api[state][date][k]['confirmed']
api['TT'][date][k]['deceased'] = api['TT'][date][k].get('deceased', 0) + api[state][date][k]['deceased']
api['TT'][date][k]['recovered'] = api['TT'][date][k].get('recovered', 0) + api[state][date][k]['recovered']
api['TT'][date][k]['active'] = api['TT'][date][k].get('active', 0) + api[state][date][k]['active']
with open("predictions_tt.json", "w") as f:
f.write(json.dumps(api, sort_keys=True))