diff --git a/src/universal_devkit/scripts/csv_to_json.py b/src/universal_devkit/scripts/csv_to_json.py index f7af89d..74a77fd 100644 --- a/src/universal_devkit/scripts/csv_to_json.py +++ b/src/universal_devkit/scripts/csv_to_json.py @@ -1,6 +1,7 @@ import argparse import csv import json +import uuid def main(csv_in, json_out): @@ -10,12 +11,12 @@ def main(csv_in, json_out): csv_in (str): the path to the CSV input file json_out (str): the path to save the resulting JSON file """ - data_list = list(csv.DictReader(open(csv_in))) + data_list = list(csv.DictReader(open(csv_in), skipinitialspace=True)) - # for obj in data_list: - # if token not in obj: - # # Avoid overwriting existing tokens - # obj["token"] = uuid.uuid4().hex + for obj in data_list: + if "token" not in obj: + # Avoid overwriting existing tokens + obj["token"] = uuid.uuid4().hex with open(json_out, "w") as f: json.dump(data_list, f)