From 4f9ffc3b1145cbdfbaae471175e5eeeba6ad3309 Mon Sep 17 00:00:00 2001 From: wqy Date: Sat, 17 Sep 2022 09:45:31 +0800 Subject: [PATCH] Clean tank --- api/s4_tanks.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/api/s4_tanks.py b/api/s4_tanks.py index e475017..461fc75 100644 --- a/api/s4_tanks.py +++ b/api/s4_tanks.py @@ -7,6 +7,10 @@ from .s24_coordinates import * import utility +OVERFLOW_YES = 'YES' +OVERFLOW_NO = 'NO' + + def add_tank(name: str, id: str, x: float, y: float, elevation: float, init_level: float = 0, min_level: float = 0, max_level: float = 0, diameter: float = 0, min_vol: float = 0) -> ChangeSet: sql = f"insert into tanks (id, elevation, init_level, min_level, max_level, diameter, min_vol) values ('{id}', {elevation}, {init_level}, {min_level}, {max_level}, {diameter}, {min_vol});" undo_sql = f'delete from tanks where id = "{id}";' @@ -132,8 +136,6 @@ def set_tank_vol_curve(name: str, id: str, vol_curve: str) -> ChangeSet: return _set_tank(name, id, 'vol_curve', 'str', vol_curve, True) -OVERFLOW_YES = 'YES' -OVERFLOW_NO = 'NO' def set_tank_overflow(name: str, id: str, overflow: str) -> ChangeSet: if overflow is not OVERFLOW_YES or overflow is not OVERFLOW_NO: