1 files changed, 43 insertions, 0 deletions
diff --git a/odl-pipeline/lib/utils/node_manager.py b/odl-pipeline/lib/utils/node_manager.py
new file mode 100755
index 0000000..d11065f
--- /dev/null
+++ b/odl-pipeline/lib/utils/node_manager.py
@@ -0,0 +1,43 @@
+#
+# Copyright (c) 2015 All rights reserved
+# This program and the accompanying materials
+# are made available under the terms of the Apache License, Version 2.0
+# which accompanies this distribution, and is available at
+#
+# http://www.apache.org/licenses/LICENSE-2.0
+#
+#
+from ssh_util import SshUtil
+
+
+class NodeManager(object):
+
+    env_nodes = []
+    env_node_dict = {}
+    primary_controller = None
+
+    def __init__(self, config=None):
+        if config:
+            for (node_name, node_config) in config.iteritems():
+                self.add_node(node_name, node_config)
+
+    def add_node(self, node_name, node_config):
+        from node import Node
+        if not node_config.get('address'):
+            node_config['address'] = self.get_address_of_node(node_name)
+        node = Node(node_name, dict=node_config)
+        self.env_nodes.append(node)
+        self.env_node_dict[node_name] = node
+        return node
+
+    def get_nodes(self):
+        return self.env_nodes
+
+    def get_node(self, name):
+        return self.env_node_dict[name]
+
+    @classmethod
+    def gen_ssh_config(cls, node):
+        if node not in cls.env_nodes:
+            cls.env_nodes.append(node)
+        SshUtil.gen_ssh_config(cls.env_nodes)