Add pattern api and test

api/__init__.py

@@ -13,7 +13,7 @@ from .operation import sync_with_server
 
 from .command import execute_batch_commands
 
-from .s0_base import JUNCTION, RESERVOIR, TANK, PIPE, PUMP, VALVE
+from .s0_base import JUNCTION, RESERVOIR, TANK, PIPE, PUMP, VALVE, PATTERN, CURVE
 from .s0_base import is_node, is_junction, is_reservoir, is_tank
 from .s0_base import is_link, is_pipe, is_pump, is_valve
 from .s0_base import is_curve
@@ -43,4 +43,6 @@ from .s9_demands import get_demand_schema, get_demand, set_demand
 from .s10_status import LINK_STATUS_OPEN, LINK_STATUS_CLOSED, LINK_STATUS_ACTIVE
 from .s10_status import get_status_schema, get_status, set_status
 
+from .s11_patterns import get_pattern_schema, get_pattern, set_pattern
+
 from .s24_coordinates import get_node_coord

api/s0_base.py

@@ -3,17 +3,20 @@ from .connection import g_conn_dict as conn
 from .operation import *
 
 
-_NODE = "_node"
-_LINK = "_link"
-_CURVE = "_curve"
-_PATTERN = "_pattern"
+_NODE = '_node'
+_LINK = '_link'
+_CURVE = '_curve'
+_PATTERN = '_pattern'
 
-JUNCTION = "junction"
-RESERVOIR = "reservoir"
-TANK = "tank"
-PIPE = "pipe"
-PUMP = "pump"
-VALVE = "valve"
+JUNCTION = 'junction'
+RESERVOIR = 'reservoir'
+TANK = 'tank'
+PIPE = 'pipe'
+PUMP = 'pump'
+VALVE = 'valve'
+
+PATTERN = 'pattern'
+CURVE = 'curve'
 
 
 def _get_from(name: str, id: str, base_type: str) -> Row | None:

api/s11_patterns.py

@@ -0,0 +1,44 @@
+from .operation import *
+from .s0_base import *
+
+
+def get_pattern_schema(name: str) -> dict[str, dict[str, Any]]:
+    return { 'id'      : {'type': 'str'        , 'optional': False , 'readonly': True },
+             'factors' : {'type': 'float_list' , 'optional': False , 'readonly': False } }
+
+
+def get_pattern(name: str, id: str) -> dict[str, Any]:
+    pas = read_all(name, f"select * from patterns where id = '{id}'")
+    ps = []
+    for r in pas:
+        ps.append(float(r['factor']))
+    return { 'id': id, 'factors': ps }
+
+
+def set_pattern(name: str, cs: ChangeSet) -> ChangeSet:
+    id = cs.operations[0]['id']
+
+    old = get_pattern(name, id)
+    new = { 'id': id, 'factors': [] }
+
+    f_id = f"'{id}'"
+
+    # TODO: transaction ?
+    redo_sql = f"delete from patterns where id = {f_id};"
+    redo_sql += f"\ndelete from _pattern where id = {f_id};"
+    redo_sql += f"\ninsert into _pattern (id) values ({f_id});"
+    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)
+
+    undo_sql = f"delete from patterns where id = {f_id};"
+    undo_sql += f"\ndelete from _pattern where id = {f_id};"
+    undo_sql += f"\ninsert into _pattern (id) values ({f_id});"
+    for f_factor in old['factors']:
+        undo_sql += f"\ninsert into patterns (id, factor) values ({f_id}, {f_factor});"
+
+    redo_cs = g_update_prefix | { 'type': 'pattern' } | new
+    undo_cs = g_update_prefix | { 'type': 'pattern' } | old
+
+    return execute_command(name, redo_sql, undo_sql, redo_cs, undo_cs)

api/s9_demands.py

@@ -29,6 +29,7 @@ def set_demand(name: str, cs: ChangeSet) -> ChangeSet:
 
     f_junction = f"'{junction}'"
 
+    # TODO: transaction ?
     redo_sql = f"delete from demands where junction = {f_junction};"
     for r in cs.operations[0]['demands']:
         demand = float(r['demand'])