X-Git-Url: https://gerrit.o-ran-sc.org/r/gitweb?a=blobdiff_plain;f=code%2Fnetwork-generator%2Fview%2Fnetwork_viewer.py;h=03871fb891775b69978a5804c5704737bdde01a3;hb=96980c02907cbdfde751f97796fa7fce4400f756;hp=57d84e85d5d2e5d337f6f97ec9752ce22275646b;hpb=a225f908d3a76457950cef532fadfd413c98d9f1;p=oam.git

diff --git a/code/network-generator/view/network_viewer.py b/code/network-generator/view/network_viewer.py
index 57d84e8..03871fb 100644
--- a/code/network-generator/view/network_viewer.py
+++ b/code/network-generator/view/network_viewer.py
@@ -18,6 +18,7 @@ Provides functions to convert the Network into different formats
 """
 
 import json
+from typing import Dict
 from model.python.o_ran_network import ORanNetwork
 import xml.etree.ElementTree as ET
 
@@ -99,9 +100,21 @@ class NetworkViewer:
         :type filename: string
         """
         root = self.__network.toKml()
-        style = ET.Element("style")
-        style.text = self.readStylesFromFile()
-        root.append(style)
+        with open('view/kml.styles.json') as kml_styles:
+            styles:Dict[str,Dict] = json.load(kml_styles)
+            for key, value in styles.items():
+                # add style
+                style = ET.Element("Style",{"id":key})
+                line_style = ET.SubElement(style, "LineStyle")
+                color = ET.SubElement(line_style, "color")
+                color.text = value['stroke']['color']
+                width = ET.SubElement(line_style, "width")
+                width.text = value['stroke']['width']
+                poly_style = ET.SubElement(style, "PolyStyle")
+                fill = ET.SubElement(poly_style, "color")
+                fill.text = value['fill']['color']
+                root.findall(".//Document")[0].append(style)
+
         ET.ElementTree(root).write(filename,
                                       encoding="utf-8",
                                       xml_declaration=True