From 069e9ac4c5b3855a21fc86fb0e92373176ed1482 Mon Sep 17 00:00:00 2001 From: "WQY\\qiong" Date: Fri, 24 Mar 2023 20:54:16 +0800 Subject: [PATCH] Add default coordinate --- api/s2_junctions.py | 3 +++ api/s3_reservoirs.py | 3 +++ api/s4_tanks.py | 3 +++ 3 files changed, 9 insertions(+) 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'])