X-Git-Url: https://gerrit.opnfv.org/gerrit/gitweb?a=blobdiff_plain;f=apex%2Fclean.py;h=f56287e141b34170893f26489131e6bd29690c02;hb=5ff098787664eb141c3c169bdc194c188520ef89;hp=af9e8ce00a217e346ded5438d2bd196f3da729e1;hpb=c6f04a5dee991a131a49c2fde9c5990fe2edac4e;p=apex.git

diff --git a/apex/clean.py b/apex/clean.py
index af9e8ce0..f56287e1 100644
--- a/apex/clean.py
+++ b/apex/clean.py
@@ -7,16 +7,21 @@
 # http://www.apache.org/licenses/LICENSE-2.0
 ##############################################################################
 
-# Clean will eventually be migrated to this file
-
 import argparse
+import fileinput
+import libvirt
 import logging
 import os
 import pyipmi
 import pyipmi.interfaces
 import sys
 
-from .common import utils
+from apex.common import (
+    constants,
+    utils)
+from apex.network import jumphost
+from apex.common.exceptions import ApexCleanException
+from virtualbmc import manager as vbmc_lib
 
 
 def clean_nodes(inventory):
@@ -41,11 +46,83 @@ def clean_nodes(inventory):
             sys.exit(1)
 
 
+def clean_vbmcs():
+    vbmc_manager = vbmc_lib.VirtualBMCManager()
+    vbmcs = vbmc_manager.list()
+    for vbmc in vbmcs:
+        logging.info("Deleting vbmc: {}".format(vbmc['domain_name']))
+        vbmc_manager.delete(vbmc['domain_name'])
+
+
+def clean_vms():
+    logging.info('Destroying all Apex VMs')
+    conn = libvirt.open('qemu:///system')
+    if not conn:
+        raise ApexCleanException('Unable to open libvirt connection')
+    pool = conn.storagePoolLookupByName('default')
+    domains = conn.listAllDomains()
+
+    for domain in domains:
+        vm = domain.name()
+        if vm != 'undercloud' and not vm.startswith('baremetal'):
+            continue
+        logging.info("Cleaning domain: {}".format(vm))
+        if domain.isActive():
+            logging.debug('Destroying domain')
+            domain.destroy()
+        domain.undefineFlags(libvirt.VIR_DOMAIN_UNDEFINE_NVRAM)
+        # delete storage volume
+        try:
+            stgvol = pool.storageVolLookupByName("{}.qcow2".format(vm))
+        except libvirt.libvirtError:
+            logging.warning("Skipping volume cleanup as volume not found for "
+                            "vm: {}".format(vm))
+            stgvol = None
+        if stgvol:
+            logging.info('Deleting storage volume')
+            stgvol.wipe(0)
+            stgvol.delete(0)
+    pool.refresh()
+
+
+def clean_ssh_keys(key_file='/root/.ssh/authorized_keys'):
+    logging.info('Removing any stack pub keys from root authorized keys')
+    if not os.path.isfile(key_file):
+        logging.warning("Key file does not exist: ".format(key_file))
+        return
+    for line in fileinput.input(key_file, inplace=True):
+        line = line.strip('\n')
+        if 'stack@undercloud' not in line:
+            print(line)
+
+
+def clean_networks():
+    logging.debug('Cleaning all network config')
+    for network in constants.OPNFV_NETWORK_TYPES:
+        logging.info("Cleaning Jump Host Network config for network "
+                     "{}".format(network))
+        jumphost.detach_interface_from_ovs(network)
+        jumphost.remove_ovs_bridge(network)
+
+    conn = libvirt.open('qemu:///system')
+    if not conn:
+        raise ApexCleanException('Unable to open libvirt connection')
+    logging.debug('Destroying all virsh networks')
+    for network in conn.listNetworks():
+        if network in constants.OPNFV_NETWORK_TYPES:
+            virsh_net = conn.networkLookupByName(network)
+            logging.debug("Destroying virsh network: {}".format(network))
+            if virsh_net.isActive():
+                virsh_net.destroy()
+            virsh_net.undefine()
+
+
 def main():
     clean_parser = argparse.ArgumentParser()
-    clean_parser.add_argument('-f',
+    clean_parser.add_argument('-i',
                               dest='inv_file',
-                              required=True,
+                              required=False,
+                              default=None,
                               help='File which contains inventory')
     args = clean_parser.parse_args(sys.argv[1:])
     os.makedirs(os.path.dirname('./apex_clean.log'), exist_ok=True)
@@ -58,7 +135,24 @@ def main():
     console.setLevel(logging.DEBUG)
     console.setFormatter(logging.Formatter(formatter))
     logging.getLogger('').addHandler(console)
-    clean_nodes(args.inv_file)
+    if args.inv_file:
+        if not os.path.isfile(args.inv_file):
+            logging.error("Inventory file not found: {}".format(args.inv_file))
+            raise FileNotFoundError("Inventory file does not exist")
+        else:
+            logging.info("Shutting down baremetal nodes")
+            clean_nodes(args.inv_file)
+    # Delete all VMs
+    clean_vms()
+    # Delete vbmc
+    clean_vbmcs()
+    # Clean network config
+    clean_networks()
+
+    # clean pub keys from root's auth keys
+    clean_ssh_keys()
+
+    logging.info('Apex clean complete!')
 
 
 if __name__ == '__main__':