From f68ff0f31263e6883456ad6e93fceaef11d79ee2 Mon Sep 17 00:00:00 2001 From: DingZQ Date: Wed, 26 Oct 2022 23:14:15 +0800 Subject: [PATCH] Add power to AddPump --- main.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/main.py b/main.py index 38a8880..9864636 100644 --- a/main.py +++ b/main.py @@ -651,10 +651,12 @@ async def fast_get_pump_schema(network: str) -> dict[str, dict[str, Any]]: return get_pump_schema(network) @app.post("/addpump/") -async def fastapi_add_pump(network: str, pump: str, node1: str, node2: str) -> ChangeSet: +async def fastapi_add_pump(network: str, pump: str, node1: str, node2: str, power: float = 0.0) -> ChangeSet: ps = { 'id' : pump, 'node1' : node1, - 'node2' : node2 } + 'node2' : node2, + 'power' : power + } return add_pump(network, ChangeSet(ps)) @app.post("/deletepump/")