From c1b4a3685ca3d3147c11e27ce86b7d3e57d7f912 Mon Sep 17 00:00:00 2001
From: DingZQ <zqding@hotmail.com>
Date: Fri, 28 Mar 2025 20:43:16 +0800
Subject: [PATCH] Refine

---
 api/__init__.py |  2 ++
 main.py         | 20 ++++++++++++++++++++
 tjnetwork.py    | 13 +++++++++++++
 3 files changed, 35 insertions(+)

diff --git a/api/__init__.py b/api/__init__.py
index 501c49a..7eed155 100644
--- a/api/__init__.py
+++ b/api/__init__.py
@@ -165,3 +165,5 @@ from .s36_wda_cal import calculate_demand_to_nodes, calculate_demand_to_region,
 from .s38_scada_info import get_scada_info_schema, get_scada_info, get_all_scada_info
 
 from .s39_user import get_user_schema, get_user, get_all_users
+
+from .s40_schema import get_scheme_schema, get_scheme, get_all_schemes
diff --git a/main.py b/main.py
index 684916d..915a382 100644
--- a/main.py
+++ b/main.py
@@ -2057,6 +2057,26 @@ async def fastapi_get_all_users(network: str) -> list[dict[Any, Any]]:
     return get_all_users(network)
 
 
+############################################################
+# scheme 40
+############################################################    
+@app.get('/getschemeschema/')
+async def fastapi_get_scheme_schema(network: str) -> dict[str, dict[Any, Any]]:
+    return get_scheme_schema(network)   
+
+@app.get('/getscheme/')
+async def fastapi_get_scheme(network: str, schema_name: str) -> dict[Any, Any]:
+    return get_scheme(network, schema_name)
+
+@app.get('/getallschemes/')
+async def fastapi_get_all_schemes(network: str) -> list[dict[Any, Any]]:
+    return get_all_schemes(network)
+
+
+
+
+
+
 
 
 
diff --git a/tjnetwork.py b/tjnetwork.py
index 05a0d09..045db3a 100644
--- a/tjnetwork.py
+++ b/tjnetwork.py
@@ -1277,3 +1277,16 @@ def get_user(name: str, user_name: str) -> dict[str, Any]:
 
 def get_all_users(name: str) -> list[dict[str, Any]]:
     return api.get_all_users(name)
+
+############################################################
+# scheme 40
+############################################################
+def get_scheme_schema(name: str) -> dict[str, dict[str, Any]]:
+    return api.get_scheme_schema(name)
+
+def get_scheme(name: str, schema_name: str) -> dict[str, Any]:
+    return api.get_scheme(name, schema_name)
+
+def get_all_schemes(name: str) -> list[dict[str, Any]]:
+    return api.get_all_schemes(name)
+