From b75a717f954155974fa8f08cde1760241525fc1a Mon Sep 17 00:00:00 2001
From: "WQY\\qiong" <qiongyu.wang@outlook.com>
Date: Wed, 10 May 2023 20:55:11 +0800
Subject: [PATCH] Add more dma interface

---
 tjnetwork.py | 14 ++++++++++----
 1 file changed, 10 insertions(+), 4 deletions(-)

diff --git a/tjnetwork.py b/tjnetwork.py
index 04a8cdd..043e215 100644
--- a/tjnetwork.py
+++ b/tjnetwork.py
@@ -1008,6 +1008,9 @@ def calculate_demand_to_nodes(name: str, demand: float, nodes: list[str]) -> dic
 def calculate_demand_to_region(name: str, demand: float, region: str) -> dict[str, float]:
     return api.calculate_demand_to_region(name, demand, region)
 
+def calculate_demand_to_network(name: str, demand: float) -> dict[str, float]:
+    return {}
+
 def get_region_schema(name: str) -> dict[str, dict[str, Any]]:
     return api.get_region_schema(name)
 
@@ -1029,9 +1032,15 @@ def delete_region(name: str, cs: ChangeSet) -> ChangeSet:
 # district_metering_area 33
 ############################################################
 
-def calculate_district_metering_area(name: str, nodes: list[str], part_count: int = 1, part_type: int = PARTITION_TYPE_RB) -> list[list[str]]:
+def calculate_district_metering_area_for_nodes(name: str, nodes: list[str], part_count: int = 1, part_type: int = PARTITION_TYPE_RB) -> list[list[str]]:
     return api.calculate_district_metering_area(name, nodes, part_count, part_type)
 
+def calculate_district_metering_area_for_region(name: str, region: str, part_count: int = 1, part_type: int = PARTITION_TYPE_RB) -> list[list[str]]:
+    return []
+
+def calculate_district_metering_area_for_network(name: str, part_count: int = 1, part_type: int = PARTITION_TYPE_RB) -> list[list[str]]:
+    return []
+
 def get_district_metering_area_schema(name: str) -> dict[str, dict[str, Any]]:
     return {}#api.get_district_metering_area_schema(name)
 
@@ -1048,9 +1057,6 @@ def add_district_metering_area(name: str, cs: ChangeSet) -> ChangeSet:
 def delete_district_metering_area(name: str, cs: ChangeSet) -> ChangeSet:
     return ChangeSet()#api.delete_district_metering_area(name, cs)
 
-def generate_district_metering_area(name: str, region: None | str, part_count: int = 1, part_type: int = PARTITION_TYPE_RB) -> ChangeSet:
-    return ChangeSet()
-
 
 ############################################################
 # service_area 34