Updates docs for SR1 with final revision
[genesis.git] / fuel / deploy / deploy.py
index 402d0f2..178ae76 100644 (file)
@@ -1,34 +1,46 @@
+###############################################################################
+# Copyright (c) 2015 Ericsson AB and others.
+# szilard.cserey@ericsson.com
+# 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
+###############################################################################
+
+
 import os
-import shutil
 import io
 import re
 import sys
 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
+from execution_environment import ExecutionEnvironment
 import common
 
 log = common.log
 exec_cmd = common.exec_cmd
 err = common.err
+warn = common.warn
 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
+delete = common.delete
 check_if_root = common.check_if_root
 ArgParser = common.ArgParser
 
 FUEL_VM = 'fuel'
 PATCH_DIR = 'fuel_patch'
-WORK_DIR = 'deploy'
+WORK_DIR = '~/deploy'
 CWD = os.getcwd()
 
+
 class cd:
+
     def __init__(self, new_path):
         self.new_path = os.path.expanduser(new_path)
 
@@ -42,15 +54,22 @@ class cd:
 
 class AutoDeploy(object):
 
-    def __init__(self, without_fuel, storage_dir, pxe_bridge, iso_file,
-                 dea_file, dha_file):
-        self.without_fuel = without_fuel
+    def __init__(self, no_fuel, fuel_only, no_health_check, cleanup_only,
+                 cleanup, storage_dir, pxe_bridge, iso_file, dea_file,
+                 dha_file, fuel_plugins_dir):
+        self.no_fuel = no_fuel
+        self.fuel_only = fuel_only
+        self.no_health_check = no_health_check
+        self.cleanup_only = cleanup_only
+        self.cleanup = cleanup
         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.fuel_plugins_dir = fuel_plugins_dir
+        self.dea = (DeploymentEnvironmentAdapter(dea_file)
+                    if not cleanup_only else None)
         self.dha = DeploymentHardwareAdapter(dha_file)
         self.fuel_conf = {}
         self.fuel_node_id = self.dha.get_fuel_node_id()
@@ -83,7 +102,8 @@ class AutoDeploy(object):
         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)
+                                 self.iso_file, WORK_DIR,
+                                 self.fuel_plugins_dir)
         fuel.install()
 
     def patch_iso(self, new_iso):
@@ -101,7 +121,7 @@ class AutoDeploy(object):
             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)
+        delete(tmp_orig_dir)
         exec_cmd('chmod -R 755 %s' % tmp_new_dir)
 
     def patch(self, tmp_new_dir, new_iso):
@@ -111,7 +131,7 @@ class AutoDeploy(object):
 
         with cd(tmp_new_dir):
             exec_cmd('cat %s | patch -p0' % ks_path)
-            shutil.rmtree('.rr_moved')
+            delete('.rr_moved')
             isolinux = 'isolinux/isolinux.cfg'
             log('isolinux.cfg before: %s'
                 % exec_cmd('grep netmask %s' % isolinux))
@@ -137,8 +157,9 @@ class AutoDeploy(object):
             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 = CloudDeploy(self.dea, self.dha, self.fuel_conf['ip'],
+                          self.fuel_username, self.fuel_password,
+                          self.dea_file, WORK_DIR, self.no_health_check)
         return dep.deploy()
 
     def setup_execution_environment(self):
@@ -146,19 +167,36 @@ class AutoDeploy(object):
                                         self.dha_file, self.dea)
         exec_env.setup_environment()
 
+    def cleanup_execution_environment(self):
+        exec_env = ExecutionEnvironment(self.storage_dir, self.pxe_bridge,
+                                        self.dha_file, self.dea)
+        exec_env.cleanup_environment()
+
     def create_tmp_dir(self):
-        self.tmp_dir = '%s/fueltmp-%s' % (CWD, str(uuid.uuid1()))
-        os.makedirs(self.tmp_dir)
+        self.tmp_dir = '%s/fueltmp' % CWD
+        delete(self.tmp_dir)
+        create_dir_if_not_exists(self.tmp_dir)
 
     def deploy(self):
-        check_if_root()
         self.collect_fuel_info()
-        if not self.without_fuel:
+        if not self.no_fuel:
             self.setup_execution_environment()
             self.create_tmp_dir()
             self.install_fuel_master()
-            shutil.rmtree(self.tmp_dir)
-        return self.deploy_env()
+        if not self.fuel_only:
+            return self.deploy_env()
+        return True
+
+    def run(self):
+        check_if_root()
+        if self.cleanup_only:
+            self.cleanup_execution_environment()
+        else:
+            deploy_success = self.deploy()
+            if self.cleanup:
+                self.cleanup_execution_environment()
+            return deploy_success
+        return True
 
 def check_bridge(pxe_bridge, dha_path):
     with io.open(dha_path) as yaml_file:
@@ -167,56 +205,104 @@ def check_bridge(pxe_bridge, dha_path):
         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)
+        if pxe_bridge in r and 'state DOWN' in r:
+            err('Linux Bridge {0} is not Active, bring'
+                ' it UP first: [ip link set dev {0} up]'.format(pxe_bridge))
+
+
+def check_fuel_plugins_dir(dir):
+    msg = None
+    if not dir:
+        msg = 'Fuel Plugins Directory not specified!'
+    elif not os.path.isdir(dir):
+        msg = 'Fuel Plugins Directory does not exist!'
+    elif not os.listdir(dir):
+        msg = 'Fuel Plugins Directory is empty!'
+    if msg:
+        warn('%s No external plugins will be installed!' % msg)
+
 
 def parse_arguments():
     parser = ArgParser(prog='python %s' % __file__)
-    parser.add_argument('-nf', dest='without_fuel', action='store_true',
+    parser.add_argument('-nf', dest='no_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',
+    parser.add_argument('-nh', dest='no_health_check', action='store_true',
+                        default=False,
+                        help='Don\'t run health check after deployment')
+    parser.add_argument('-fo', dest='fuel_only', action='store_true',
+                        default=False,
+                        help='Install Fuel Master only (and Node VMs when '
+                             'using libvirt)')
+    parser.add_argument('-co', dest='cleanup_only', action='store_true',
+                        default=False,
+                        help='Cleanup VMs and Virtual Networks according to '
+                             'what is defined in DHA')
+    parser.add_argument('-c', dest='cleanup', action='store_true',
+                        default=False,
+                        help='Cleanup after deploy')
+    if {'-iso', '-dea', '-dha', '-h'}.intersection(sys.argv):
+        parser.add_argument('-iso', dest='iso_file', action='store', nargs='?',
+                            default='%s/OPNFV.iso' % CWD,
+                            help='ISO File [default: OPNFV.iso]')
+        parser.add_argument('-dea', dest='dea_file', action='store', nargs='?',
+                            default='%s/dea.yaml' % CWD,
+                            help='Deployment Environment Adapter: dea.yaml')
+        parser.add_argument('-dha', dest='dha_file', action='store', nargs='?',
+                            default='%s/dha.yaml' % CWD,
+                            help='Deployment Hardware Adapter: dha.yaml')
+    else:
+        parser.add_argument('iso_file', action='store', nargs='?',
+                            default='%s/OPNFV.iso' % CWD,
+                            help='ISO File [default: OPNFV.iso]')
+        parser.add_argument('dea_file', action='store', nargs='?',
+                            default='%s/dea.yaml' % CWD,
+                            help='Deployment Environment Adapter: dea.yaml')
+        parser.add_argument('dha_file', action='store', nargs='?',
+                            default='%s/dha.yaml' % CWD,
+                            help='Deployment Hardware Adapter: dha.yaml')
+    parser.add_argument('-s', dest='storage_dir', action='store',
                         default='%s/images' % CWD,
                         help='Storage Directory [default: images]')
-    parser.add_argument('pxe_bridge', nargs='?', action='store',
+    parser.add_argument('-b', dest='pxe_bridge', action='store',
                         default='pxebr',
                         help='Linux Bridge for booting up the Fuel Master VM '
                              '[default: pxebr]')
+    parser.add_argument('-p', dest='fuel_plugins_dir', action='store',
+                        help='Fuel Plugins directory')
 
     args = parser.parse_args()
+    log(args)
 
-    check_file_exists(args.dea_file)
     check_file_exists(args.dha_file)
 
-    if not args.without_fuel:
+    if not args.cleanup_only:
+        check_file_exists(args.dea_file)
+        check_fuel_plugins_dir(args.fuel_plugins_dir)
+
+    if not args.no_fuel and not args.cleanup_only:
         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)
+    kwargs = {'no_fuel': args.no_fuel, 'fuel_only': args.fuel_only,
+              'no_health_check': args.no_health_check,
+              'cleanup_only': args.cleanup_only, 'cleanup': args.cleanup,
+              'storage_dir': args.storage_dir, 'pxe_bridge': args.pxe_bridge,
+              'iso_file': args.iso_file, 'dea_file': args.dea_file,
+              'dha_file': args.dha_file,
+              'fuel_plugins_dir': args.fuel_plugins_dir}
+    return kwargs
 
 
 def main():
-    without_fuel, storage_dir, pxe_bridge, iso_file, dea_file, dha_file = \
-        parse_arguments()
+    kwargs = parse_arguments()
 
-    d = AutoDeploy(without_fuel, storage_dir, pxe_bridge, iso_file,
-                   dea_file, dha_file)
-    sys.exit(d.deploy())
+    d = AutoDeploy(**kwargs)
+    sys.exit(d.run())
 
 if __name__ == '__main__':
-    main()
\ No newline at end of file
+    main()