From 0dba1e0b6d0df2a5e595f35c2d7d3e47dcc15420 Mon Sep 17 00:00:00 2001
From: DingZQ <zqding@hotmail.com>
Date: Sun, 6 Nov 2022 12:35:11 +0800
Subject: [PATCH] Add energy methods

---
 main.py | 33 ++++++++++++++++++++++++++++++++-
 1 file changed, 32 insertions(+), 1 deletion(-)

diff --git a/main.py b/main.py
index e19898f..134b55f 100644
--- a/main.py
+++ b/main.py
@@ -972,9 +972,40 @@ async def fastapi_get_rule(network: str) -> dict[str, Any]:
 @app.post("/setruleproperties/")
 async def fastapi_set_rule(network: str, req: Request) -> ChangeSet:
     props = await req.json()
-    return set_rule(network, props)
+    return set_rule(network, ChangeSet(props))
 
 
+############################################################
+# energy 15.[ENERGY]
+############################################################
+
+@app.get("/getglobalenergyschema/")
+async def fastapi_get_global_energy_schema(network: str) -> dict[str, dict[str, Any]]:
+    return get_global_energy_schema()
+
+@app.get("/getglobalenergyproperties/")
+async def fastapi_get_global_energy_properties(network: str) -> dict[str, Any]:
+    return get_global_energy(network)
+
+@app.post("/setglobalenergyproperties/")
+async def fastapi_set_global_energy_properties(network: str, req: Request) -> ChangeSet:
+    props = await req.json()
+    return set_global_energy(network, ChangeSet(props))
+
+@app.get("/getpumpenergyschema/")
+async def fastapi_get_pump_energy_schema(network: str) -> dict[str, dict[str, Any]]:
+    return get_pump_energy_schema(network)
+
+@app.get("/getpumpenergyproperties//")
+async def fastapi_get_pump_energy_proeprties(network: str, pump: str) -> dict[str, Any]:
+    return get_pump_energy(network, pump)
+
+@app.get("/setpumpenergyproperties//")
+async def fastapi_set_pump_energy_properties(network: str, pump: str, req: Request) -> ChangeSet:
+    props = await req.json()
+    ps = { 'id' : pump } | props
+    return set_pump_energy(network, ChangeSet(ps))
+
 ############################################################
 # emitter 16.[EMITTERS]
 ############################################################