ADD N1, N2 N3 interfacing to topology generation

[oam.git] / code / network-topology-instance-generator / model / python / tapi_link.py

diff --git a/code/network-topology-instance-generator/model/python/tapi_link.py b/code/network-topology-instance-generator/model/python/tapi_link.py
index d930d55..61c6e43 100644 (file)
--- a/code/network-topology-instance-generator/model/python/tapi_link.py
+++ b/code/network-topology-instance-generator/model/python/tapi_link.py
@@ -121,7 +121,8 @@ class TapiLink(Top):
         """
 
         group = etree.Element("g")
-        group.attrib["class"] = "link"
+        class_name = self.name().split("-")[0].lower()
+        group.attrib["class"] = " ".join(["link", class_name])
         title = etree.Element("title")
         title.text = "\n TAPI Link\n id: " + \
             self.identifier() + "\n name: " + self.name()