diff --git a/api/s2_junctions.py b/api/s2_junctions.py index 28aadd0..2d3bc80 100644 --- a/api/s2_junctions.py +++ b/api/s2_junctions.py @@ -16,6 +16,9 @@ def get_junction(name: str, id: str) -> dict[str, Any]: if j == None: return {} xy = get_node_coord(name, id) + if xy == {}: + write(name, f"insert into coordinates (node, coord) values ('{id}', '(0,0)');") + xy = get_node_coord(name, id) d = {} d['id'] = str(j['id']) d['x'] = float(xy['x']) diff --git a/api/s3_reservoirs.py b/api/s3_reservoirs.py index d3a3d7f..6fee308 100644 --- a/api/s3_reservoirs.py +++ b/api/s3_reservoirs.py @@ -17,6 +17,9 @@ def get_reservoir(name: str, id: str) -> dict[str, Any]: if r == None: return {} xy = get_node_coord(name, id) + if xy == {}: + write(name, f"insert into coordinates (node, coord) values ('{id}', '(0,0)');") + xy = get_node_coord(name, id) d = {} d['id'] = str(r['id']) d['x'] = float(xy['x']) diff --git a/api/s4_tanks.py b/api/s4_tanks.py index 45341fe..af8ae5c 100644 --- a/api/s4_tanks.py +++ b/api/s4_tanks.py @@ -27,6 +27,9 @@ def get_tank(name: str, id: str) -> dict[str, Any]: if t == None: return {} xy = get_node_coord(name, id) + if xy == {}: + write(name, f"insert into coordinates (node, coord) values ('{id}', '(0,0)');") + xy = get_node_coord(name, id) d = {} d['id'] = str(t['id']) d['x'] = float(xy['x'])