Update odltools

[netvirt.git] / resources / tools / odltools / odltools / mdsal / models / network_topology.py

diff --git a/resources/tools/odltools/odltools/mdsal/models/network_topology.py b/resources/tools/odltools/odltools/mdsal/models/network_topology.py
index d83ed595d19ee848a96f77abc2f826f068d5a1d4..12725d60be68452237a40d9a478588290bd11112 100644 (file)
--- a/resources/tools/odltools/odltools/mdsal/models/network_topology.py
+++ b/resources/tools/odltools/odltools/mdsal/models/network_topology.py
@@ -1,10 +1,23 @@
-from odltools.mdsal.models.model import Model
+# Copyright 2018 Red Hat, Inc. and others. All Rights Reserved.
+#
+# 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.
 
+from odltools.mdsal.models.model import Model
 
 MODULE = "network-topology"
 
 
-def network_topology(store, args, mid="ovsdb:1"):
+def network_topology(store, args, mid="topology/ovsdb:1"):
     return NetworkTopology(MODULE, store, args, mid)
 
 
@@ -14,9 +27,10 @@ class NetworkTopology(Model):
     TOPOLOGY = "topology"
     NODE = "node"
     OVSDB1 = "ovsdb:1"
+    TOPOLOGY_OVSDB1 = TOPOLOGY + "/" + OVSDB1
 
     def get_clist(self):
-        return self.data[self.TOPOLOGY]
+        return self.data.get(self.TOPOLOGY, [])
 
     def get_topology_by_tid(self, tid="ovsdb:1"):
         topologies = self.get_clist()
@@ -35,3 +49,39 @@ class NetworkTopology(Model):
         for node in nodes:
             d[node[key]] = node
         return d
+
+    def get_nodes_by_dpid(self, tid="ovsdb:1"):
+        d = {}
+        nodes = self.get_nodes_by_tid(tid)
+        for node in nodes:
+            dpid = self.get_datapathid_from_node(node)
+            if dpid:
+                d[dpid] = node
+        return d
+
+    def get_dpn_host_mapping(self):
+        d = {}
+        nodes = self.get_nodes_by_tid_and_key()
+        for nodeid in nodes:
+            node = nodes.get(nodeid)
+            dpid = self.get_datapathid_from_node(node)
+            if dpid:
+                parentid = self.get_parent_nodeid(nodeid)
+                host = self.get_host_id_from_node(nodes.get(parentid))
+                d[dpid] = host
+        return d
+
+    def get_host_id_from_node(self, node):
+        extids = node.get("ovsdb:openvswitch-external-ids", {})
+        for extid in extids:
+            if extid.get("external-id-key") == "hostname":
+                return extid.get("external-id-value")
+
+    def get_datapathid_from_node(self, node):
+        datapathid = node.get("ovsdb:datapath-id")
+        if datapathid:
+            return int(datapathid.replace(':', ''), 16)
+
+    def get_parent_nodeid(self, nodeid):
+        parentid = nodeid.split('/bridge/')[0]
+        return parentid if parentid else ''