X-Git-Url: https://gerrit.akraino.org/r/gitweb?p=ta%2Fcloudtaf.git;a=blobdiff_plain;f=libraries%2Fcommon%2Fstack_infos.py;fp=libraries%2Fcommon%2Fstack_infos.py;h=4ddf52f1da8e2d99e8741b9e12ce522cda3ae92c;hp=0000000000000000000000000000000000000000;hb=af5eb3ff36b92ab1d9c156ffa0391eadc73eb6ba;hpb=025a45508d009db84c34076fb4a668f712628d6d

diff --git a/libraries/common/stack_infos.py b/libraries/common/stack_infos.py
new file mode 100644
index 0000000..4ddf52f
--- /dev/null
+++ b/libraries/common/stack_infos.py
@@ -0,0 +1,77 @@
+import sys
+import os
+import json
+import paramiko
+from robot.libraries.BuiltIn import BuiltIn
+from users import *
+sys.path.append(os.path.join(os.path.dirname(__file__), '..'))
+
+
+class stack_infos:  # pylint: disable=old-style-class
+
+    INVENTORY_PATH = "/opt/cmframework/scripts/inventory.sh"
+
+    def __init__(self):
+        self._floating_ip = BuiltIn().get_variable_value("${floating_ip}")
+
+        try:
+            client = paramiko.SSHClient()
+            client.load_system_host_keys()
+            client.set_missing_host_key_policy(paramiko.AutoAddPolicy)
+            client.connect(self._floating_ip,
+                           username=cloudadmin['username'],
+                           password=cloudadmin['password'])
+            _, stdout, _ = client.exec_command(self.INVENTORY_PATH)
+            self._inventory = json.loads(stdout.read())
+        finally:
+            client.close()
+
+        self._crf_nodes = self.get_node_ip_based_on_caas_profile("caas_master")
+        if not self._crf_nodes:
+            raise Exception("crf_nodes dictionary is empty!")
+        self._storage_nodes = self.get_node_ip_based_on_caas_profile("caas_storage")
+        self._worker_nodes = self.get_node_ip_based_on_caas_profile("caas_worker")
+
+    def get_floating_ip(self):
+        return self._floating_ip
+
+    def get_crf_nodes(self):
+        return self._crf_nodes
+
+    def get_storage_nodes(self):
+        return self._storage_nodes
+
+    def get_worker_nodes(self):
+        return self._worker_nodes
+
+    def get_all_nodes(self):
+        all_nodes = self._crf_nodes.copy()
+        all_nodes.update(self._storage_nodes)
+        all_nodes.update(self._worker_nodes)
+        return all_nodes
+
+    def get_inventory(self):
+        return self._inventory
+
+    def get_node_ip_based_on_caas_profile(self, caas_profile):  # pylint: disable=invalid-name
+        node_ip = {}
+        if caas_profile in self._inventory:
+            for node in self._inventory[caas_profile]:
+                node_ip[node] = self._inventory["_meta"]["hostvars"][node]["networking"]["infra_internal"]["ip"]
+        return node_ip
+
+    def get_infra_int_if(self):
+        interface = self._inventory["_meta"]["hostvars"]["controller-1"]["networking"]["infra_internal"]["interface"]
+        return interface
+
+    def get_infra_ext_if(self):
+        iface = self._inventory["_meta"]["hostvars"]["controller-1"]["networking"]["infra_external"]["interface"]
+        return iface
+
+    def get_infra_storage_if(self):
+        iface = self._inventory["_meta"]["hostvars"]["controller-1"]["networking"]["infra_storage_cluster"]["interface"]
+        return iface
+
+    def get_virtual_env(self):
+        virtual_env = self._inventory["all"]["vars"]["virtual_env"]
+        return virtual_env