Fuel Config Reap + Additional Refactoring for Autodeployment
[genesis.git] / fuel / deploy / deploy.py
index 4df4f36..3305aed 100644 (file)
-import subprocess
-import sys
-import time
 import os
-from dha import DeploymentHardwareAdapter
-from dea import DeploymentEnvironmentAdapter
-
-SUPPORTED_RELEASE = 'Juno on CentOS 6.5'
-N = {'id': 0, 'status': 1, 'name': 2, 'cluster': 3, 'ip': 4, 'mac': 5,
-     'roles': 6, 'pending_roles': 7, 'online': 8}
-E = {'id': 0, 'status': 1, 'name': 2, 'mode': 3, 'release_id': 4,
-     'changes': 5, 'pending_release_id': 6}
-R = {'id': 0, 'name': 1, 'state': 2, 'operating_system': 3, 'version': 4}
-RO = {'name': 0, 'conflicts': 1}
-
-def exec_cmd(cmd):
-    process = subprocess.Popen(cmd,
-                               stdout=subprocess.PIPE,
-                               stderr=subprocess.STDOUT,
-                               shell=True)
-    return process.communicate()[0]
-
-def parse(printout):
-   parsed_list = []
-   lines = printout.splitlines()
-   for l in lines[2:]:
-        parsed = [e.strip() for e in l.split('|')]
-        parsed_list.append(parsed)
-   return parsed_list
-
-def err(error_message):
-    sys.stderr.write(error_message)
-    sys.exit(1)
-
-
-
-class Deploy(object):
-
-    def __init__(self):
-        self.supported_release = None
-
-    def get_id_list(self, list):
-        return [l[0] for l in list]
-
-    def cleanup_fuel_environments(self, env_list):
-        WAIT_LOOP = 10
-        SLEEP_TIME = 2
-        id_list = self.get_id_list(env_list)
-        for id in id_list:
-            exec_cmd('fuel env --env %s --delete' % id)
-            for i in range(WAIT_LOOP):
-                if id in self.get_id_list(parse(exec_cmd('fuel env list'))):
-                    time.sleep(SLEEP_TIME)
-                else:
-                    continue
-
-    def cleanup_fuel_nodes(self, node_list):
-        for node in node_list:
-            if node[N['status']] == 'discover':
-                exec_cmd('fuel node --node-id %s --delete-from-db'
-                         % node[N['id']])
-                exec_cmd('dockerctl shell cobbler cobbler system remove '
-                         '--name node-%s' % node[N['id']])
-
-    def check_previous_installation(self):
-        env_list = parse(exec_cmd('fuel env list'))
-        if env_list:
-            self.cleanup_fuel_environments(env_list)
-        node_list = parse(exec_cmd('fuel node list'))
-        if node_list:
-            self.cleanup_fuel_nodes(node_list)
-
-    def check_supported_release(self):
-        release_list= parse(exec_cmd('fuel release -l'))
-        for release in release_list:
-            if release[R['name']] == SUPPORTED_RELEASE:
-                self.supported_release = release
-                break
-        if not self.supported_release:
-            err("This Fuel doesn't contain the following "
-                "release: %s\n" % SUPPORTED_RELEASE)
-
-    def check_role_definitions(self):
-        role_list= parse(exec_cmd('fuel role --release %s'
-                                  % self.supported_release[R['id']]))
-        roles = [role[RO['name']] for role in role_list]
-        if 'compute' not in roles:
-            err("Role compute does not exist in release %"
-                % self.supported_release[R['name']])
-        if 'controller' not in roles:
-            err("Role controller does not exist in release %"
-                % self.supported_release[R['name']])
-
-    def check_prerequisites(self):
-        self.check_supported_release()
-        self.check_role_definitions()
-        self.check_previous_installation()
-
-    def count_discovered_nodes(self, node_list):
-        discovered_nodes = 0
-        for node in node_list:
-            if node[N['status']] == 'discover':
-                discovered_nodes += 1
-        return discovered_nodes
-
-    def wait_for_discovered_blades(self, no_of_blades):
-        WAIT_LOOP = 10
-        SLEEP_TIME = 2
-        all_discovered = False
-        node_list = parse(exec_cmd('fuel node list'))
-        for i in range(WAIT_LOOP):
-            if (self.count_discovered_nodes(node_list) < no_of_blades):
-                time.sleep(SLEEP_TIME)
-                node_list = parse(exec_cmd('fuel node list'))
-            else:
-                all_discovered = True
-                break
-        if not all_discovered:
-            err("There are %s blades defined, but not all of "
-                "them have been discovered\n" % no_of_blades)
-
-    def assign_cluster_node_ids(self, dha, dea, controllers, compute_hosts):
-        node_list= parse(exec_cmd('fuel node list'))
-        for shelf_id in dea.get_shelf_ids():
-            for blade_id in dea.get_blade_ids(shelf_id):
-                blade_mac_list = dha.get_blade_mac_addresses(
-                    shelf_id, blade_id)
-
-                found = False
-                for node in node_list:
-                    if (node[N['mac']] in blade_mac_list and
-                        node[N['status']] == 'discover'):
-                        found = True
-                        break
-                if found:
-                    if dea.is_controller(shelf_id, blade_id):
-                        controllers.append(node[N['id']])
-                    if dea.is_compute_host(shelf_id, blade_id):
-                        compute_hosts.append(node[N['id']])
-                else:
-                    err("Could not find the Node ID for blade "
-                        "with MACs %s or blade is not in "
-                        "discover status\n" % blade_mac_list)
-
-    def env_exists(self, env_name):
-        env_list = parse(exec_cmd('fuel env --list'))
-        for env in env_list:
-            if env[E['name']] == env_name and env[E['status']] == 'new':
-                return True
-        return False
-
-    def configure_environment(self, dea):
-        env_name = dea.get_environment_name()
-        exec_cmd('fuel env -c --name %s --release %s --mode ha --net neutron '
-                 '--nst vlan' % (env_name, self.supported_release[R['id']]))
-
-        if not self.env_exists(env_name):
-            err("Failed to create environment %s" % env_name)
+import shutil
+import io
+import re
+import netaddr
+import uuid
+import yaml
 
+from dea import DeploymentEnvironmentAdapter
+from dha import DeploymentHardwareAdapter
+from install_fuel_master import InstallFuelMaster
+from deploy_env import CloudDeploy
+from setup_execution_environment import ExecutionEnvironment
+import common
+
+log = common.log
+exec_cmd = common.exec_cmd
+err = common.err
+check_file_exists = common.check_file_exists
+check_dir_exists = common.check_dir_exists
+create_dir_if_not_exists = common.create_dir_if_not_exists
+check_if_root = common.check_if_root
+ArgParser = common.ArgParser
+
+FUEL_VM = 'fuel'
+PATCH_DIR = 'fuel_patch'
+WORK_DIR = 'deploy'
+CWD = os.getcwd()
+
+class cd:
+    def __init__(self, new_path):
+        self.new_path = os.path.expanduser(new_path)
+
+    def __enter__(self):
+        self.saved_path = CWD
+        os.chdir(self.new_path)
+
+    def __exit__(self, etype, value, traceback):
+        os.chdir(self.saved_path)
+
+
+class AutoDeploy(object):
+
+    def __init__(self, without_fuel, storage_dir, pxe_bridge, iso_file,
+                 dea_file, dha_file):
+        self.without_fuel = without_fuel
+        self.storage_dir = storage_dir
+        self.pxe_bridge = pxe_bridge
+        self.iso_file = iso_file
+        self.dea_file = dea_file
+        self.dha_file = dha_file
+        self.dea = DeploymentEnvironmentAdapter(dea_file)
+        self.dha = DeploymentHardwareAdapter(dha_file)
+        self.fuel_conf = {}
+        self.fuel_node_id = self.dha.get_fuel_node_id()
+        self.fuel_username, self.fuel_password = self.dha.get_fuel_access()
+        self.tmp_dir = None
+
+    def modify_ip(self, ip_addr, index, val):
+        ip_str = str(netaddr.IPAddress(ip_addr))
+        decimal_list = map(int, ip_str.split('.'))
+        decimal_list[index] = val
+        return '.'.join(map(str, decimal_list))
+
+    def collect_fuel_info(self):
+        self.fuel_conf['ip'] = self.dea.get_fuel_ip()
+        self.fuel_conf['gw'] = self.dea.get_fuel_gateway()
+        self.fuel_conf['dns1'] = self.dea.get_fuel_dns()
+        self.fuel_conf['netmask'] = self.dea.get_fuel_netmask()
+        self.fuel_conf['hostname'] = self.dea.get_fuel_hostname()
+        self.fuel_conf['showmenu'] = 'yes'
+
+    def install_fuel_master(self):
+        log('Install Fuel Master')
+        new_iso = '%s/deploy-%s' \
+                  % (self.tmp_dir, os.path.basename(self.iso_file))
+        self.patch_iso(new_iso)
+        self.iso_file = new_iso
+        self.install_iso()
+
+    def install_iso(self):
+        fuel = InstallFuelMaster(self.dea_file, self.dha_file,
+                                 self.fuel_conf['ip'], self.fuel_username,
+                                 self.fuel_password, self.fuel_node_id,
+                                 self.iso_file, WORK_DIR)
+        fuel.install()
+
+    def patch_iso(self, new_iso):
+        tmp_orig_dir = '%s/origiso' % self.tmp_dir
+        tmp_new_dir = '%s/newiso' % self.tmp_dir
+        self.copy(tmp_orig_dir, tmp_new_dir)
+        self.patch(tmp_new_dir, new_iso)
+
+    def copy(self, tmp_orig_dir, tmp_new_dir):
+        log('Copying...')
+        os.makedirs(tmp_orig_dir)
+        os.makedirs(tmp_new_dir)
+        exec_cmd('fuseiso %s %s' % (self.iso_file, tmp_orig_dir))
+        with cd(tmp_orig_dir):
+            exec_cmd('find . | cpio -pd %s' % tmp_new_dir)
+        with cd(tmp_new_dir):
+            exec_cmd('fusermount -u %s' % tmp_orig_dir)
+        shutil.rmtree(tmp_orig_dir)
+        exec_cmd('chmod -R 755 %s' % tmp_new_dir)
+
+    def patch(self, tmp_new_dir, new_iso):
+        log('Patching...')
+        patch_dir = '%s/%s' % (CWD, PATCH_DIR)
+        ks_path = '%s/ks.cfg.patch' % patch_dir
+
+        with cd(tmp_new_dir):
+            exec_cmd('cat %s | patch -p0' % ks_path)
+            shutil.rmtree('.rr_moved')
+            isolinux = 'isolinux/isolinux.cfg'
+            log('isolinux.cfg before: %s'
+                % exec_cmd('grep netmask %s' % isolinux))
+            self.update_fuel_isolinux(isolinux)
+            log('isolinux.cfg after: %s'
+                % exec_cmd('grep netmask %s' % isolinux))
+
+            iso_linux_bin = 'isolinux/isolinux.bin'
+            exec_cmd('mkisofs -quiet -r -J -R -b %s '
+                     '-no-emul-boot -boot-load-size 4 '
+                     '-boot-info-table -hide-rr-moved '
+                     '-x "lost+found:" -o %s .'
+                     % (iso_linux_bin, new_iso))
+
+    def update_fuel_isolinux(self, file):
+        with io.open(file) as f:
+            data = f.read()
+        for key, val in self.fuel_conf.iteritems():
+            pattern = r'%s=[^ ]\S+' % key
+            replace = '%s=%s' % (key, val)
+            data = re.sub(pattern, replace, data)
+        with io.open(file, 'w') as f:
+            f.write(data)
+
+    def deploy_env(self):
+        dep = CloudDeploy(self.dha, self.fuel_conf['ip'], self.fuel_username,
+                          self.fuel_password, self.dea_file, WORK_DIR)
+        dep.deploy()
+
+    def setup_execution_environment(self):
+        exec_env = ExecutionEnvironment(self.storage_dir, self.pxe_bridge,
+                                        self.dha_file, self.dea)
+        exec_env.setup_environment()
+
+    def create_tmp_dir(self):
+        self.tmp_dir = '%s/fueltmp-%s' % (CWD, str(uuid.uuid1()))
+        os.makedirs(self.tmp_dir)
+
+    def deploy(self):
+        check_if_root()
+        self.collect_fuel_info()
+        if not self.without_fuel:
+            self.setup_execution_environment()
+            self.create_tmp_dir()
+            self.install_fuel_master()
+            shutil.rmtree(self.tmp_dir)
+        self.deploy_env()
+
+def check_bridge(pxe_bridge, dha_path):
+    with io.open(dha_path) as yaml_file:
+        dha_struct = yaml.load(yaml_file)
+    if dha_struct['adapter'] != 'libvirt':
+        log('Using Linux Bridge %s for booting up the Fuel Master VM'
+            % pxe_bridge)
+        r = exec_cmd('ip link show %s' % pxe_bridge)
+        if pxe_bridge in r and 'state UP' not in r:
+            err('Linux Bridge {0} is not Active, '
+                'bring it UP first: [ip link set dev {0} up]' % pxe_bridge)
+
+def parse_arguments():
+    parser = ArgParser(prog='python %s' % __file__)
+    parser.add_argument('-nf', dest='without_fuel', action='store_true',
+                        default=False,
+                        help='Do not install Fuel Master (and Node VMs when '
+                             'using libvirt)')
+    parser.add_argument('iso_file', nargs='?', action='store',
+                        default='%s/OPNFV.iso' % CWD,
+                        help='ISO File [default: OPNFV.iso]')
+    parser.add_argument('dea_file', action='store',
+                        help='Deployment Environment Adapter: dea.yaml')
+    parser.add_argument('dha_file', action='store',
+                        help='Deployment Hardware Adapter: dha.yaml')
+    parser.add_argument('storage_dir', nargs='?', action='store',
+                        default='%s/images' % CWD,
+                        help='Storage Directory [default: images]')
+    parser.add_argument('pxe_bridge', nargs='?', action='store',
+                        default='pxebr',
+                        help='Linux Bridge for booting up the Fuel Master VM '
+                             '[default: pxebr]')
+
+    args = parser.parse_args()
+
+    check_file_exists(args.dea_file)
+    check_file_exists(args.dha_file)
+
+    if not args.without_fuel:
+        log('Using OPNFV ISO file: %s' % args.iso_file)
+        check_file_exists(args.iso_file)
+        log('Using image directory: %s' % args.storage_dir)
+        create_dir_if_not_exists(args.storage_dir)
+        log('Using bridge %s to boot up Fuel Master VM on it'
+            % args.pxe_bridge)
+        check_bridge(args.pxe_bridge, args.dha_file)
+
+    return (args.without_fuel, args.storage_dir, args.pxe_bridge,
+            args.iso_file, args.dea_file, args.dha_file)
 
 
 def main():
+    without_fuel, storage_dir, pxe_bridge, iso_file, dea_file, dha_file = \
+        parse_arguments()
 
-    yaml_path = exec_cmd('pwd').strip() + '/dea.yaml'
-    deploy = Deploy()
-
-    dea = DeploymentEnvironmentAdapter()
-
-    if not os.path.isfile(yaml_path):
-        sys.stderr.write("ERROR: File %s not found\n" % yaml_path)
-        sys.exit(1)
-
-    dea.parse_yaml(yaml_path)
-
-    dha = DeploymentHardwareAdapter(dea.get_server_type())
-
-    deploy.check_prerequisites()
-
-    dha.power_off_blades()
-
-    dha.configure_networking()
-
-    dha.reset_to_factory_defaults()
-
-    dha.set_boot_order()
-
-    dha.power_on_blades()
-
-    dha.get_blade_mac_addresses()
-
-    deploy.wait_for_discovered_blades(dea.get_no_of_blades())
-
-    controllers = []
-    compute_hosts = []
-    deploy.assign_cluster_node_ids(dha, dea, controllers, compute_hosts)
-
-    deploy.configure_environment(dea)
-
-
+    d = AutoDeploy(without_fuel, storage_dir, pxe_bridge, iso_file,
+                   dea_file, dha_file)
+    d.deploy()
 
 if __name__ == '__main__':
     main()
\ No newline at end of file