import os
import json
from utils.load_data import load_data

def write_json(path):
    for root, _, files in os.walk(path):
        if(files):
            for f in files:
                # Get input file name
                tsp_file = os.path.join(root, str(f))
                print(tsp_file)
                nodes = load_data(tsp_file)

                # Write to JSON
                data = {}

                data['nodes'] = []
                for i in range(len(nodes)):
                    data['nodes'].append({
                        'title': str(i),
                        'id': i,
                        'x': int(nodes[i][0]),
                        'y': int(nodes[i][1])
                    })

                with open('./' + os.path.splitext(os.path.basename(tsp_file))[0] + '.json', 'w') as outfile:
                    json.dump(data, outfile)