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/")