diff --git a/parse_data.py b/parse_data.py index 11efd70..4741b9c 100644 --- a/parse_data.py +++ b/parse_data.py @@ -4,6 +4,33 @@ import json node_names = set() edges = [] +libraries = {} +locations = {} +with open("data/locaties.csv") as fp: + reader = csv.DictReader(fp, delimiter=";") + for item in reader: + locatie = item['Locatie'].split(',') + try: + lat, lon = locatie + except ValueError as e: + lat, lon = None, None + library = { + 'name': item['Library Name'], + 'code': item['Library Code'], + 'adres': item['Adres'], + 'lat': lat, + 'lon': lon, + } + location = { + 'location': item['Location Name'], + 'code': item['Location Code'], + 'library': library + } + + libraries[library['name']] = library + locations[location['code']] = location + + with open("data/requests.csv") as fp: reader = csv.DictReader(fp, delimiter=";") for item in reader: @@ -17,7 +44,7 @@ nodes = [{'name': n} for n in node_names] print(f"{len(nodes)} nodes, {len(edges)} edges") data = { - 'nodes': nodes, + 'nodes': list(libraries.values()), #nodes, 'edges': edges }