Create ietf-network-topology json

[oam.git] / code / network-generator / model / python / o_ran_termination_point.py

diff --git a/code/network-generator/model/python/o_ran_termination_point.py b/code/network-generator/model/python/o_ran_termination_point.py
index ff9dc89..6d575e7 100644 (file)
--- a/code/network-generator/model/python/o_ran_termination_point.py
+++ b/code/network-generator/model/python/o_ran_termination_point.py
@@ -18,15 +18,50 @@
 An abstract Class for O-RAN TerminationPoint
 """
 from abc import abstractmethod
-from typing import Any
-from model.python.o_ran_object import ORanObject
+from model.python.o_ran_object import IORanObject, ORanObject
 
 
-# Define an abstract O-RAN Node class
-class ORanTerminationPoint(ORanObject):
-    def __init__(self, **kwargs):
+# Define the "IORanObject" interface
+class IORanTerminationPointData(IORanObject):
+    def __init__(self, supporter: str = None, parent=None, **kwargs):
         super().__init__(**kwargs)
+        self.supporter = supporter
+        self.parent = parent
 
-    def toTopology(self):
-        result: dict[str, Any] = {"tp-id": self.name}
+
+# Define an O-RAN Termination Point (ietf-interface, onf:logical-termination-point) class
+class ORanTerminationPoint(ORanObject, IORanTerminationPointData):
+    def __init__(self, tp: IORanTerminationPointData = None, **kwargs):
+        super().__init__(tp, **kwargs)
+        self.supporter = tp["supporter"] if tp and "supporter" in tp else None
+        self.parent = tp["parent"] if tp and "parent" in tp else None
+
+    def to_topology(self):
+        result: dict[str, dict] = {"tp-id": self.name}
+        if self.supporter:
+            network_ref: str = ""
+            match str(type(self.parent)):
+                case "<class 'model.python.o_ran_smo.ORanSmo'>":
+                    network_ref = self.parent.parent.id
+                case "<class 'model.python.o_ran_near_rt_ric.ORanNearRtRic'>":
+                    network_ref = self.parent.parent.parent.id
+                case "<class 'model.python.o_ran_cu.ORanCu'>":
+                    network_ref = self.parent.parent.parent.parent.id
+                case "<class 'model.python.o_ran_du.ORanDu'>":
+                    network_ref = self.parent.parent.parent.parent.parent.id
+                case "<class 'model.python.o_ran_cloud_du.ORanCloudDu'>":
+                    network_ref = self.parent.parent.parent.parent.parent.id
+                case "<class 'model.python.o_ran_ru.ORanRu'>":
+                    network_ref = self.parent.parent.parent.parent.parent.parent.id
+                case _:
+                    print("unknown: implement " + str(type(self.parent)))
+                    network_ref = "unknown: implement " + str(type(self.parent))
+
+            result["supporting-termination-point"] = [
+                {
+                    "network-ref": network_ref,
+                    "node-ref": self.parent.name,
+                    "tp-ref": self.supporter,
+                }
+            ]
         return result