Merge "Automatic Deployment"

1 files changed, 153 insertions, 0 deletions

diff --git a/fuel/deploy/cloud_deploy/hardware_adapters/libvirt/libvirt_adapter.py b/fuel/deploy/cloud_deploy/hardware_adapters/libvirt/libvirt_adapter.py
new file mode 100644
index 0000000..d332e59
--- /dev/null
+++ b/fuel/deploy/cloud_deploy/hardware_adapters/libvirt/libvirt_adapter.py
@@ -0,0 +1,153 @@
+from lxml import etree
+from cloud import common
+from ssh_client import SSHClient
+
+exec_cmd = common.exec_cmd
+err = common.err
+LOG = common.LOG
+
+
+class LibvirtAdapter(object):
+
+    def __init__(self, mgmt_ip, username, password):
+        self.mgmt_ip = mgmt_ip
+        self.username = username
+        self.password = password
+        self.parser = etree.XMLParser(remove_blank_text=True)
+
+    def power_off_blades(self, shelf, blade_list):
+        ssh = SSHClient(self.mgmt_ip, self.username, self.password)
+        ssh.open()
+        for blade in blade_list:
+            LOG.debug('Power off blade %s in shelf %s' % (blade, shelf))
+            vm_name = 's%s_b%s' % (shelf, blade)
+            resp = ssh.execute('virsh destroy %s' % vm_name)
+            LOG.debug('response: %s' % resp)
+        ssh.close()
+
+    def power_on_blades(self, shelf, blade_list):
+        ssh = SSHClient(self.mgmt_ip, self.username, self.password)
+        ssh.open()
+        for blade in blade_list:
+            LOG.debug('Power on blade %s in shelf %s' % (blade, shelf))
+            vm_name = 's%s_b%s' % (shelf, blade)
+            resp = ssh.execute('virsh start %s' % vm_name)
+            LOG.debug('response: %s' % resp)
+        ssh.close()
+
+    def set_boot_order_blades(self, shelf, blade_list, boot_dev_list=None):
+        if not boot_dev_list:
+            boot_dev_list = ['network', 'hd']
+        ssh = SSHClient(self.mgmt_ip, self.username, self.password)
+        ssh.open()
+        temp_dir= ssh.execute('mktemp -d').strip()
+        for blade in blade_list:
+            LOG.debug('Set boot order %s on blade %s in shelf %s'
+                  % (boot_dev_list, blade, shelf))
+            vm_name = 's%s_b%s' % (shelf, blade)
+            resp = ssh.execute('virsh dumpxml %s' % vm_name)
+            xml_dump = etree.fromstring(resp, self.parser)
+            os = xml_dump.xpath('/domain/os')
+            for o in os:
+                for bootelem in ['boot', 'bootmenu']:
+                    boot = o.xpath(bootelem)
+                    for b in boot:
+                        b.getparent().remove(b)
+                for dev in boot_dev_list:
+                    b = etree.Element('boot')
+                    b.set('dev', dev)
+                    o.append(b)
+                bmenu = etree.Element('bootmenu')
+                bmenu.set('enable', 'no')
+                o.append(bmenu)
+            tree = etree.ElementTree(xml_dump)
+            xml_file = temp_dir + '/%s.xml' % vm_name
+            with open(xml_file, 'w') as f:
+                tree.write(f, pretty_print=True, xml_declaration=True)
+            ssh.execute('virsh define %s' % xml_file)
+        ssh.execute('rm -fr %s' % temp_dir)
+        ssh.close()
+
+    def get_blades_mac_addresses(self, shelf, blade_list):
+        LOG.debug('Get the MAC addresses of blades %s in shelf %s'
+              % (blade_list, shelf))
+        macs_per_blade_dict = {}
+        ssh = SSHClient(self.mgmt_ip, self.username, self.password)
+        ssh.open()
+        for blade in blade_list:
+            vm_name = 's%s_b%s' % (shelf, blade)
+            mac_list = macs_per_blade_dict[blade] = []
+            resp = ssh.execute('virsh dumpxml %s' % vm_name)
+            xml_dump = etree.fromstring(resp)
+            interfaces = xml_dump.xpath('/domain/devices/interface')
+            for interface in interfaces:
+                macs = interface.xpath('mac')
+                for mac in macs:
+                    mac_list.append(mac.get('address'))
+        ssh.close()
+        return macs_per_blade_dict
+
+    def load_image_file(self, shelf=None, blade=None, vm=None,
+                        image_path=None):
+        if shelf and blade:
+            vm_name = 's%s_b%s' % (shelf, blade)
+        else:
+            vm_name = vm
+
+        LOG.debug('Load media file %s into %s '
+                  % (image_path, 'vm %s' % vm if vm else 'blade %s in shelf %s'
+                                                         % (shelf, blade)))
+
+        ssh = SSHClient(self.mgmt_ip, self.username, self.password)
+        ssh.open()
+        temp_dir= ssh.execute('mktemp -d').strip()
+        resp = ssh.execute('virsh dumpxml %s' % vm_name)
+        xml_dump = etree.fromstring(resp)
+
+        disks = xml_dump.xpath('/domain/devices/disk')
+        for disk in disks:
+            if disk.get('device') == 'cdrom':
+                disk.set('type', 'file')
+                sources = disk.xpath('source')
+                for source in sources:
+                    disk.remove(source)
+                source = etree.SubElement(disk, 'source')
+                source.set('file', image_path)
+        tree = etree.ElementTree(xml_dump)
+        xml_file = temp_dir + '/%s.xml' % vm_name
+        with open(xml_file, 'w') as f:
+            tree.write(f, pretty_print=True, xml_declaration=True)
+        ssh.execute('virsh define %s' % xml_file)
+        ssh.execute('rm -fr %s' % temp_dir)
+        ssh.close()
+
+    def eject_image_file(self, shelf=None, blade=None, vm=None):
+        if shelf and blade:
+            vm_name = 's%s_b%s' % (shelf, blade)
+        else:
+            vm_name = vm
+
+        LOG.debug('Eject media file from %s '
+                  % 'vm %s' % vm if vm else 'blade %s in shelf %s'
+                                            % (shelf, blade))
+
+        ssh = SSHClient(self.mgmt_ip, self.username, self.password)
+        ssh.open()
+        temp_dir= ssh.execute('mktemp -d').strip()
+        resp = ssh.execute('virsh dumpxml %s' % vm_name)
+        xml_dump = etree.fromstring(resp)
+
+        disks = xml_dump.xpath('/domain/devices/disk')
+        for disk in disks:
+            if disk.get('device') == 'cdrom':
+                disk.set('type', 'block')
+                sources = disk.xpath('source')
+                for source in sources:
+                    disk.remove(source)
+        tree = etree.ElementTree(xml_dump)
+        xml_file = temp_dir + '/%s.xml' % vm_name
+        with open(xml_file, 'w') as f:
+            tree.write(f, pretty_print=True, xml_declaration=True)
+        ssh.execute('virsh define %s' % xml_file)
+        ssh.execute('rm -fr %s' % temp_dir)
+        ssh.close()