Fix module import

This commit is contained in:
wqy
2022-09-17 18:42:12 +08:00
parent 47e0b48025
commit 8cf5895f1d
6 changed files with 23 additions and 23 deletions

View File

@@ -1,7 +1,7 @@
from psycopg.rows import Row
from .s0_base import *
from .change_set import ChangeSet
import utility
from .utility import *
VALVES_TYPE_PRV = 'prv'
@@ -24,7 +24,7 @@ def add_valve(name: str, id: str, node1: str, node2: str, diameter: float = 0, t
def _get_valve(name: str, id: str) -> Row | None:
return utility.query(name, f"select node1, node2, diameter, type, setting, minor_loss from valves where id = '{id}'")
return query(name, f"select node1, node2, diameter, type, setting, minor_loss from valves where id = '{id}'")
def delete_valve(name: str, id: str) -> ChangeSet:
@@ -81,7 +81,7 @@ def _set_valve(name: str, id: str, key: str, key_type: str, value: str, optional
if row == None:
return
return utility.update(name, VALVE, 'valves', 'id', id, key, key_type, row[key], value, optional)
return update(name, VALVE, 'valves', 'id', id, key, key_type, row[key], value, optional)
def set_valve_node1(name: str, id: str, node1: str) -> ChangeSet: