#!/usr/bin/python

import os
import os.path

import json
import sys

argc = len(sys.argv)

json_input_file  = sys.argv[1] if (argc>=2) and (argc<=4)  else 'jamendo-evened-out-10000-dataset-trackids.json'
input_dir  = sys.argv[2] if (argc==3) or (argc==4)  else 'download-json-evened-out'
output_file = sys.argv[3] if argc==4 else "jamendo-evened-out-10000-dataset-info.json"


evened_out_json_data=open(json_input_file).read()
evened_out_jamendo_ids=json.loads(evened_out_json_data)

#json_files = [ jf for jf in os.listdir(input_dir) if os.path.isfile(os.path.join(input_dir,jf)) ]

json_dataset = []

for jid in evened_out_jamendo_ids:
#for jf in json_files:
    json_filename = os.path.join(input_dir,jid + ".json")
    json_data=open(json_filename).read()
    data = json.loads(json_data)

    for rec in data[u'results']:
        print "Adding track id: " + rec[u"id"]
        json_dataset.append(rec)

print "Writing out JSON file ..."
with open(output_file, "w") as json_ofile:
    json.dump(json_dataset,json_ofile)
print "Done"