From a30de5d8e10ed2aa9373f4a3f1d28e94f70c211c Mon Sep 17 00:00:00 2001
From: DingZQ <zqding@hotmail.com>
Date: Sun, 18 Sep 2022 09:43:34 +0800
Subject: [PATCH] Add pump and valve api

---
 main.py | 87 +++++++++++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 87 insertions(+)

diff --git a/main.py b/main.py
index 086369f..5c43590 100644
--- a/main.py
+++ b/main.py
@@ -410,6 +410,93 @@ def fastapi_set_pipe_status(name: str, pipe_id: str, status: float) -> ChangeSet
     return set_pipe_status(name, pipe_id, status)
 
 
+############################################################
+# pump 4.[PUMPS]
+############################################################
+
+@app.post("/addpump/")
+def add_pump(name: str, pump_id: str, node1: str, node2: str) -> ChangeSet:
+    return add_pump(name, pump_id, node1, node2)
+
+@app.post("/deletepump/")
+def delete_pump(name: str, pump_id: str) -> ChangeSet:
+    return delete_pump(name, pump_id)
+
+@app.get("/getpumpnode1/")
+def get_pump_node1(name: str, pump_id: str) -> str | None:
+    return get_pump_node1(name, pump_id)
+
+@app.get("/getpumpnode2/")
+def get_pump_node2(name: str, pump_id: str) -> str | None:
+    return get_pump_node2(name, pump_id)
+
+@app.post("/setpumpnode1/")
+def set_pump_node1(name: str, pump_id: str, node1: str) -> ChangeSet:
+    return set_pump_node1(name, pump_id, node1)
+
+@app.post("/setpumpnode2/")
+def set_pump_node2(name: str, pump_id: str, node2: str) -> ChangeSet:
+    return set_pump_node2(name, pump_id, node2)
+
+
+############################################################
+# valve 4.[VALVES]
+############################################################
+
+@app.post("/addvalve/")
+def add_valve(name: str, valve_id: str, node1: str, node2: str, diameter: float = 0, type: str = VALVES_TYPE_PRV, setting: float = 0, minor_loss: float = 0) -> ChangeSet:
+    return add_valve(name, valve_id, node1, node2, diameter, type, setting, minor_loss)
+
+@app.post("/deletevalve/")
+def delete_valve(name: str, valve_id: str) -> ChangeSet:
+    return delete_valve(name, valve_id)
+
+@app.get("/getvalvenode1/")
+def get_valve_node1(name: str, valve_id: str) -> str | None:
+    return get_valve_node1(name, valve_id)
+
+@app.get("/getvalvenode2/")
+def get_valve_node2(name: str, valve_id: str) -> str | None:
+    return get_valve_node2(name, valve_id)
+
+@app.get("/getvalvediameter/")
+def get_valve_diameter(name: str, valve_id: str) -> float | None:
+    return get_valve_diameter(name, valve_id)
+
+@app.get("/getvalvetype/")
+def get_valve_type(name: str, valve_id: str) -> str | None:
+    return get_valve_type(name, valve_id)
+
+@app.get("/getvalvesetting/")
+def get_valve_setting(name: str, valve_id: str) -> float | None:
+    return get_valve_setting(name, valve_id)
+
+@app.get("/getvalvesetting/")
+def get_valve_minor_loss(name: str, valve_id: str) -> float | None:
+    return get_valve_minor_loss(name, valve_id)
+
+@app.get("/getvalvesetting/")
+def set_valve_node1(name: str, valve_id: str, node1: str) -> ChangeSet:
+    return set_valve_node1(name, valve_id, node1)
+
+@app.get("/getvalvesetting/")
+def set_valve_node2(name: str, valve_id: str, node2: str) -> ChangeSet:
+    return set_valve_node2(name, valve_id, node2)
+
+@app.get("/setvalvediameter/")
+def set_valve_diameter(name: str, valve_id: str, diameter: float) -> ChangeSet:
+    return set_valve_diameter(name, valve_id, diameter)
+
+@app.get("/setvalvetype/")
+def set_valve_type(name: str, valve_id: str, type: str) -> ChangeSet:
+    return set_valve_type(name, valve_id, type)
+
+@app.get("/setvalvesetting/")
+def set_valve_setting(name: str, valve_id: str, setting: float) -> ChangeSet:
+    return set_valve_setting(name, valve_id, setting)
+
+
+
 # inp file
 @app.post("/uploadinp/", status_code=status.HTTP_200_OK)
 async def upload_inp(file: bytes = File(), name: str = None):