Create a script to generate a Topology

- add init viewer class

Issue-ID: OAM-249
Change-Id: I5802a0b49832bffafd436dd8529066fa7d28cfdf
Signed-off-by: demx8as6 <martin.skorupski@highstreet-technologies.com>

diff --git a/code/network-topology-instance-generator/view/network_viewer.py b/code/network-topology-instance-generator/view/network_viewer.py
new file mode 100644 (file)
index 0000000..dc6037a
--- /dev/null
+++ b/code/network-topology-instance-generator/view/network_viewer.py
@@ -0,0 +1,96 @@
+# Copyright 2022 highstreet technologies GmbH
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+#!/usr/bin/python
+"""
+Provides functions to convert the Network into different formats
+"""
+
+import encodings
+import json
+from lxml import etree
+from model.python.tapi_common_context import TapiCommonContext
+
+
+class NetworkViewer:
+    """
+    This class contains all functions converting the Network into different formats
+    """
+    __network: TapiCommonContext = None
+
+    # constructor
+    def __init__(self, network: TapiCommonContext):
+        self.__network = network
+
+    # json format
+
+    def json(self) -> 'NetworkViewer':
+        """
+        Getter returns the class as json object
+        :return The class itsself, as it is json serializable
+        """
+        return self
+
+    def show_as_json(self) -> dict:
+        """
+        Method printing the class in json format.
+        """
+        print(self.__network.json())
+
+    def show(self):
+        """
+        Method printing the network
+        """
+        print(self.__network)
+
+    def save(self, filename: str):
+        """
+        Method saving the class content to a file in json format.
+        :param filename: A valid path to a file on the system.
+        :type filename: string
+        """
+        with open(filename, "w", encoding='utf-8') as json_file:
+            output = self.__network.json()
+            json.dump(output, json_file,
+                      ensure_ascii=False, indent=2)
+            for key in ["Node", "Link"]:
+                print(key + "s:", len(output
+                                      ["tapi-common:context"]
+                                      ["tapi-topology:topology-context"]
+                                      ["topology"][0][key.lower()])
+                      )
+            print("File '" + filename + "' saved!")
+
+    def svg(self, filename: str):
+        """
+        Method saving the class content to a file in xml/svg format.
+
+        :param filename: A valid path to a file on the system.
+        :type filename: string
+        """
+        root = self.__network.svg(0, 0)
+        root.addprevious(
+            etree.ProcessingInstruction("xml-stylesheet",
+                                        'href="svg.style.css" type="text/css"')
+        )
+        etree.ElementTree(root).write(filename,
+                                      encoding="utf-8",
+                                      xml_declaration=True,
+                                      doctype=(
+                                          '<!DOCTYPE svg PUBLIC "-//W3C//DTD SVG 1.1//EN"\n'
+                                          '  "http://www.w3.org/Graphics/SVG/1.1/DTD/svg11.dtd">'
+                                      ),
+                                      pretty_print=True
+                                      )
+        print("File '" + filename + "' saved!")