Create Apex Adapter
[releng.git] / modules / opnfv / installer_adapters / apex / ApexAdapter.py
index 17a27b1..29637d7 100644 (file)
 # http://www.apache.org/licenses/LICENSE-2.0
 ##############################################################################
 
+import os
+import re
+
+import opnfv.utils.SSHUtils as ssh_utils
+import opnfv.utils.OPNFVLogger as logger
+
 
 class ApexAdapter:
 
-    def __init__(self, installer_ip):
+    def __init__(self, installer_ip, pkey_file, user="stack"):
         self.installer_ip = installer_ip
+        self.installer_user = user
+        self.pkey_file = pkey_file
+        self.installer_connection = ssh_utils.get_ssh_client(
+            self.installer_ip,
+            self.installer_user,
+            pkey_file=self.pkey_file)
+        self.logger = logger.Logger("ApexHandler").getLogger()
 
-    def get_deployment_info(self):
-        pass
+    def runcmd_apex_installer(self, cmd):
+        _, stdout, stderr = (self.installer_connection.exec_command(cmd))
+        error = stderr.readlines()
+        if len(error) > 0:
+            self.logger.error("error %s" % ''.join(error))
+            return error
+        output = ''.join(stdout.readlines())
+        return output
 
     def get_nodes(self):
-        pass
+        nodes = []
+        output = self.runcmd_apex_installer(
+            "source /home/stack/stackrc;nova list")
+        lines = output.rsplit('\n')
+        if len(lines) < 4:
+            self.logger.info("No nodes found in the deployment.")
+            return None
+
+        for line in lines:
+            if 'controller' in line:
+                roles = "controller"
+            elif 'compute' in line:
+                roles = "compute"
+            else:
+                continue
+            if 'Daylight' in line:
+                roles = + ", OpenDaylight"
+            fields = line.split('|')
+            dict = {"id": re.sub('[!| ]', '', fields[1]),
+                    "roles": roles,
+                    "name": re.sub('[!| ]', '', fields[2]),
+                    "status": re.sub('[!| ]', '', fields[3]),
+                    "ip": re.sub('[!| ctlplane=]', '', fields[6])}
+            nodes.append(dict)
+
+        return nodes
+
+    def get_deployment_info(self):
+        str = "Deployment details:\n"
+        str += "\tINSTALLER:   Apex\n"
+        str += ("\tSCENARIO:    %s\n" %
+                os.getenv('DEPLOY_SCENARIO', 'Unknown'))
+        sdn = "None"
+
+        nodes = self.get_nodes()
+        if nodes is None:
+            self.logger.info("No nodes found in the deployment.")
+            return
+        num_nodes = len(nodes)
+        num_controllers = 0
+        num_computes = 0
+        for node in nodes:
+            if 'controller' in node['roles']:
+                num_controllers += 1
+            if 'compute' in node['roles']:
+                num_computes += 1
+            if 'Daylight' in node['name']:
+                sdn = 'OpenDaylight'
+
+        ha = str(num_controllers >= 3)
+
+        str += "\tHA:          %s\n" % ha
+        str += "\tNUM.NODES:   %s\n" % num_nodes
+        str += "\tCONTROLLERS: %s\n" % num_controllers
+        str += "\tCOMPUTES:    %s\n" % num_computes
+        str += "\tSDN CONTR.:  %s\n\n" % sdn
+
+        str += "\tNODES:\n"
+        for node in nodes:
+            str += ("\t  ID:     %s\n" % node['id'])
+            str += ("\t  Name:   %s\n" % node['name'])
+            str += ("\t  Roles:  %s\n" % node['roles'])
+            str += ("\t  Status: %s\n" % node['status'])
+            str += ("\t  IP:     %s\n\n" % node['ip'])
+
+        return str
+
+    def get_controller_ips(self, options=None):
+        nodes = self.get_nodes()
+        controllers = []
+        for node in nodes:
+            if "controller" in node["roles"]:
+                controllers.append(node['ip'])
+        return controllers
+
+    def get_compute_ips(self, options=None):
+        nodes = self.get_nodes()
+        computes = []
+        for node in nodes:
+            if "compute" in node["roles"]:
+                computes.append(node['ip'])
+        return computes
 
-    def get_controller_ips(self):
-        pass
+    def get_file_from_installer(self, remote_path, local_path, options=None):
+        self.logger.debug("Fetching %s from Undercloud %s" %
+                          (remote_path, self.installer_ip))
+        get_file_result = ssh_utils.get_file(self.installer_connection,
+                                             remote_path,
+                                             local_path)
+        if get_file_result is None:
+            self.logger.error("SFTP failed to retrieve the file.")
+            return 1
+        self.logger.info("%s successfully copied from Undercloud to %s" %
+                         (remote_path, local_path))
 
-    def get_compute_ips(self):
-        pass
+    def get_file_from_controller(self,
+                                 remote_path,
+                                 local_path,
+                                 ip=None,
+                                 options=None):
+        if ip is None:
+            controllers = self.get_controller_ips()
+            ip = controllers[0]
 
-    def get_file_from_installer(self, origin, target, options=None):
-        pass
+        connection = ssh_utils.get_ssh_client(ip,
+                                              'heat-admin',
+                                              pkey_file=self.pkey_file)
 
-    def get_file_from_controller(self, origin, target, ip=None, options=None):
-        pass
+        get_file_result = ssh_utils.get_file(connection,
+                                             remote_path,
+                                             local_path)
+        if get_file_result is None:
+            self.logger.error("SFTP failed to retrieve the file.")
+            return 1
+        self.logger.info("%s successfully copied from %s to %s" %
+                         (remote_path, ip, local_path))