Update tank api and test

This commit is contained in:
wqy
2022-09-25 00:25:04 +08:00
parent 277f5a3501
commit 79fa7d3fae
6 changed files with 190 additions and 89 deletions

View File

@@ -20,7 +20,7 @@ def get_reservoir_schema(name: str) -> dict[str, dict[str, Any]]:
def _query_reservoir(name: str, id: str) -> Row | None:
return read(name, f"select id, head, pattern from reservoirs where id = '{id}'")
return read(name, f"select * from reservoirs where id = '{id}'")
def add_reservoir(name: str, id: str, x: float, y: float, head: float) -> ChangeSet:
@@ -57,6 +57,9 @@ def get_reservoir(name: str, id: str) -> dict[str, Any] | None:
def set_reservoir(name: str, id: str, properties: dict[str, Any]) -> ChangeSet:
if not is_reservoir(name, id):
return ChangeSet()
if 'pattern' in properties:
if not is_pattern(properties['pattern']):
return ChangeSet()
old = Serialize(get_reservoir(name, id), schema).to_storage()