Fix module in terms of python style

api/s0_base.py

@@ -0,0 +1,131 @@
+from psycopg.rows import dict_row
+from api.connection import g_conn_dict as conn
+from api.operation import *
+
+_NODE = "_node"
+_LINK = "_link"
+_CURVE = "_curve"
+_PATTERN = "_pattern"
+
+JUNCTION = "JUNCTION"
+RESERVOIR = "RESERVOIR"
+TANK = "TANK"
+PIPE = "PIPE"
+PUMP = "PUMP"
+VALVE = "VALVE"
+
+
+# add
+
+def _add_id_type(name: str, id: str, type: str, table: str) -> None:
+    with conn[name].cursor() as cur:
+        sql = f"insert into {table} (id, type) values ('{id}', '{type}')"
+        cur.execute(sql)
+        redo = sql.replace("'", '"')
+        undo = f'delete from {table} where id = "{id}"'
+        add_operation(name, redo, undo)
+
+def _add_id(name: str, id: str, table: str) -> None:
+    with conn[name].cursor() as cur:
+        sql = f"insert into {table} (id) values ('{id}')"
+        cur.execute(sql)
+        redo = sql.replace("'", '"')
+        undo = f'delete from {table} where id = "{id}"'
+        add_operation(name, redo, undo)
+
+def add_node(name: str, id: str, type: str) -> None:
+    return _add_id_type(name, id, type, _NODE)
+
+def add_link(name: str, id: str, type: str) -> None:
+    return _add_id_type(name, id, type, _LINK)
+
+def add_curve(name: str, id: str) -> None:
+    return _add_id(name, id, _CURVE)
+
+def add_pattern(name: str, id: str) -> None:
+    return _add_id(name, id, _PATTERN)
+
+
+# have
+
+def _have_impl(name: str, id: str, table: str) -> bool:
+    with conn[name].cursor() as cur:
+        cur.execute(f"select * from {table} where id = '{id}'")
+        return cur.rowcount > 0
+
+def have_node(name: str, id: str) -> bool:
+    return _have_impl(name, id, _NODE)
+
+def have_link(name: str, id: str) -> bool:
+    return _have_impl(name, id, _LINK)
+
+def have_curve(name: str, id: str) -> bool:
+    return _have_impl(name, id, _CURVE)
+
+def have_pattern(name: str, id: str) -> bool:
+    return _have_impl(name, id, _PATTERN)
+
+
+# get
+def _get_impl(name: str, id: str, table: str) -> dict[str, str]:
+    with conn[name].cursor(row_factory=dict_row) as cur:
+        cur.execute(f"select * from {table} where id = '{id}'")
+        if cur.rowcount > 0:
+            return cur.fetchone()
+        else:
+            return {}
+
+def get_node(name: str, id: str) -> dict[str, str]:
+    return _get_impl(name, id, _NODE)
+
+def get_link(name: str, id: str) -> dict[str, str]:
+    return _get_impl(name, id, _LINK)
+
+def get_curve(name: str, id: str) -> dict[str, str]:
+    return _get_impl(name, id, _CURVE)
+
+def get_pattern(name: str, id: str) -> dict[str, str]:
+    return _get_impl(name, id, _PATTERN)
+
+
+# delete
+
+def _delete_id_type(name: str, id: str, type: str, table: str) -> None:
+    with conn[name].cursor() as cur:
+        sql = f"delete from {table} where id = '{id}'"
+        cur.execute(sql)
+        redo = sql.replace("'", '"')
+        undo = f'insert into {table} (id, type) values ("{id}", "{type}")'
+        add_operation(name, redo, undo)
+
+def _delete_id(name:str, id: str, table: str) -> None:
+    with conn[name].cursor() as cur:
+        sql = f"delete from {table} where id = '{id}'"
+        cur.execute(sql)
+        redo = sql.replace("'", '"')
+        undo = f'insert into {table} (id) values ("{id}")'
+        add_operation(name, redo, undo)
+
+def delete_node(name: str, id: str) -> None:
+    row = get_node(name, id)
+    if row == {}:
+        return
+    return _delete_id_type(name, id, row['type'], _NODE)
+
+def delete_link(name: str, id: str) -> None:
+    row = get_link(name, id)
+    if row == {}:
+        return
+    return _delete_id_type(name, id, row['type'], _LINK)
+
+def delete_curve(name: str, id: str) -> None:
+    row = get_curve(name, id)
+    if row == {}:
+        return
+    return _delete_id(name, id, _CURVE)
+
+def delete_pattern(name: str, id: str) -> None:
+    row = get_pattern(name, id)
+    if row == {}:
+        return
+    return _delete_id(name, id, _PATTERN)