fuel/deploy/environments/libvirt_environment.py

   1 from lxml import etree
   2 import glob
   3
   4 import common
   5 from execution_environment import ExecutionEnvironment
   6
   7 exec_cmd = common.exec_cmd
   8 err = common.err
   9 log = common.log
  10 check_dir_exists = common.check_dir_exists
  11 check_file_exists = common.check_file_exists
  12 check_if_root = common.check_if_root
  13
  14 NET_DIR = 'libvirt/networks'
  15
  16 class LibvirtEnvironment(ExecutionEnvironment):
  17
  18     def __init__(self, storage_dir, dha_file, dea, root_dir):
  19         super(LibvirtEnvironment, self).__init__(
  20             storage_dir, dha_file, root_dir)
  21         self.dea = dea
  22         self.network_dir = '%s/%s' % (self.root_dir, NET_DIR)
  23         self.node_ids = self.dha.get_all_node_ids()
  24         self.net_names = self.collect_net_names()
  25
  26     def create_storage(self, node_id, disk_path, disk_sizes):
  27         if node_id == self.fuel_node_id:
  28            disk_size = disk_sizes['fuel']
  29         else:
  30            roles = self.dea.get_node_role(node_id)
  31            role = 'controller' if 'controller' in roles else 'compute'
  32            disk_size = disk_sizes[role]
  33         exec_cmd('fallocate -l %s %s' % (disk_size, disk_path))
  34
  35     def create_vms(self):
  36         temp_dir = exec_cmd('mktemp -d')
  37         disk_sizes = self.dha.get_disks()
  38         for node_id in self.node_ids:
  39             vm_name = self.dha.get_node_property(node_id, 'libvirtName')
  40             vm_template = '%s/%s' % (self.root_dir,
  41                                      self.dha.get_node_property(
  42                                          node_id, 'libvirtTemplate'))
  43             check_file_exists(vm_template)
  44             disk_path = '%s/%s.raw' % (self.storage_dir, vm_name)
  45             self.create_storage(node_id, disk_path, disk_sizes)
  46             temp_vm_file = '%s/%s' % (temp_dir, vm_name)
  47             exec_cmd('cp %s %s' % (vm_template, temp_vm_file))
  48             self.define_vm(vm_name, temp_vm_file, disk_path)
  49         exec_cmd('rm -fr %s' % temp_dir)
  50
  51     def create_networks(self):
  52         for net_file in glob.glob('%s/*' % self.network_dir):
  53             exec_cmd('virsh net-define %s' % net_file)
  54         for net in self.net_names:
  55             log('Creating network %s' % net)
  56             exec_cmd('virsh net-autostart %s' % net)
  57             exec_cmd('virsh net-start %s' % net)
  58
  59     def delete_networks(self):
  60         for net in self.net_names:
  61             log('Deleting network %s' % net)
  62             exec_cmd('virsh net-destroy %s' % net, False)
  63             exec_cmd('virsh net-undefine %s' % net, False)
  64
  65     def get_net_name(self, net_file):
  66         with open(net_file) as f:
  67             net_xml = etree.parse(f)
  68             name_list = net_xml.xpath('/network/name')
  69             for name in name_list:
  70                 net_name = name.text
  71         return net_name
  72
  73     def collect_net_names(self):
  74         net_list = []
  75         for net_file in glob.glob('%s/*' % self.network_dir):
  76             name = self.get_net_name(net_file)
  77             net_list.append(name)
  78         return net_list
  79
  80     def delete_vms(self):
  81         for node_id in self.node_ids:
  82             self.delete_vm(node_id)
  83
  84     def setup_environment(self):
  85         check_if_root()
  86         check_dir_exists(self.network_dir)
  87         self.cleanup_environment()
  88         self.create_vms()
  89         self.create_networks()
  90
  91     def cleanup_environment(self):
  92         self.delete_vms()
  93         self.delete_networks()