Add curve api and test

api/__init__.py

@@ -45,4 +45,6 @@ from .s10_status import get_status_schema, get_status, set_status
 
 from .s11_patterns import get_pattern_schema, get_pattern, set_pattern
 
+from .s12_curves import get_curve_schema, get_curve, set_curve
+
 from .s24_coordinates import get_node_coord

api/command.py

@@ -7,6 +7,8 @@ from .s6_pumps import *
 from .s7_valves import *
 from .s9_demands import *
 from .s10_status import *
+from .s11_patterns import *
+from .s12_curves import *
 
 
 def execute_add_command(name: str, cs: ChangeSet) -> ChangeSet:
@@ -49,6 +51,10 @@ def execute_update_command(name: str, cs: ChangeSet) -> ChangeSet:
         return set_demand(name, cs)
     elif type == 'status':
         return set_status(name, cs)
+    elif type == PATTERN:
+        return set_pattern(name, cs)
+    elif type == CURVE:
+        return set_curve(name, cs)
 
     return ChangeSet()
 

api/s11_patterns.py

@@ -30,7 +30,7 @@ def set_pattern(name: str, cs: ChangeSet) -> ChangeSet:
     for factor in cs.operations[0]['factors']:
         f_factor = float(factor)
         redo_sql += f"\ninsert into patterns (id, factor) values ({f_id}, {f_factor});"
-        new['factors'].append(f_factor)
+        new['factors'].append(factor)
 
     undo_sql = f"delete from patterns where id = {f_id};"
     undo_sql += f"\ndelete from _pattern where id = {f_id};"

api/s12_curves.py

@@ -0,0 +1,48 @@
+from .operation import *
+from .s0_base import *
+
+
+def get_curve_schema(name: str) -> dict[str, dict[str, Any]]:
+    return { 'id'     : {'type': 'str'  , 'optional': False , 'readonly': True },
+             'coords' : {'type': 'list' , 'optional': False , 'readonly': False, 
+                           'element': { 'x'  : {'type': 'float' , 'optional': False , 'readonly': False }, 
+                                        'y'  : {'type': 'float' , 'optional': False , 'readonly': False } }}}
+
+
+def get_curve(name: str, id: str) -> dict[str, Any]:
+    cus = read_all(name, f"select * from curves where id = '{id}'")
+    cs = []
+    for r in cus:
+        cs.append({ 'x': float(r['x']), 'y': float(r['y']) })
+    return { 'id': id, 'coords': cs }
+
+
+def set_curve(name: str, cs: ChangeSet) -> ChangeSet:
+    id = cs.operations[0]['id']
+
+    old = get_curve(name, id)
+    new = { 'id': id, 'coords': [] }
+
+    f_id = f"'{id}'"
+
+    # TODO: transaction ?
+    redo_sql = f"delete from curves where id = {f_id};"
+    redo_sql += f"\ndelete from _curve where id = {f_id};"
+    redo_sql += f"\ninsert into _curve (id) values ({f_id});"
+    for xy in cs.operations[0]['coords']:
+        x, y = float(xy['x']), float(xy['y'])
+        f_x, f_y = x, y
+        redo_sql += f"\ninsert into curves (id, x, y) values ({f_id}, {f_x}, {f_y});"
+        new['coords'].append({ 'x': x, 'y': y })
+
+    undo_sql = f"delete from curves where id = {f_id};"
+    undo_sql += f"\ndelete from _curve where id = {f_id};"
+    undo_sql += f"\ninsert into _curve (id) values ({f_id});"
+    for xy in old['coords']:
+        f_x, f_y = xy['x'], xy['y']
+        undo_sql += f"\ninsert into curves (id, x, y) values ({f_id}, {f_x}, {f_y});"
+
+    redo_cs = g_update_prefix | { 'type': 'curve' } | new
+    undo_cs = g_update_prefix | { 'type': 'curve' } | old
+
+    return execute_command(name, redo_sql, undo_sql, redo_cs, undo_cs)