aboutsummaryrefslogtreecommitdiffstats
path: root/deploy/environments/execution_environment.py
diff options
context:
space:
mode:
authorDaniel Smith <daniel.smith@ericsson.com>2016-04-05 15:38:09 +0000
committerGerrit Code Review <gerrit@172.30.200.206>2016-04-05 15:38:09 +0000
commit27120b363ddd3df5474314204def6607871cad6e (patch)
treed93487ba282ddf4561460ad9ff1b69ab144b52a6 /deploy/environments/execution_environment.py
parenteaf3af01c4bdede10bd89bb55c7d572cb424aa85 (diff)
parent25ec9b6a9bfc20f8b5379af71292fd49b8cce12a (diff)
Merge "Configure the infra VMs over DHA"
Diffstat (limited to 'deploy/environments/execution_environment.py')
-rw-r--r--deploy/environments/execution_environment.py31
1 files changed, 26 insertions, 5 deletions
diff --git a/deploy/environments/execution_environment.py b/deploy/environments/execution_environment.py
index c2e7a0b23..2a4e39e08 100644
--- a/deploy/environments/execution_environment.py
+++ b/deploy/environments/execution_environment.py
@@ -50,7 +50,30 @@ class ExecutionEnvironment(object):
for file in disk_files:
delete(file)
- def define_vm(self, vm_name, temp_vm_file, disk_path, number_cpus):
+ def overwrite_xml(self, vm_xml, vm_definition_overwrite):
+ for key, value in vm_definition_overwrite.iteritems():
+ if key == 'attribute_equlas':
+ continue
+ if key == 'value':
+ vm_xml.text = str(value)
+ return
+ if key == 'attribute':
+ for attr_key, attr_value in value.iteritems():
+ vm_xml.set(attr_key, str(attr_value))
+ return
+
+ if isinstance(value, dict):
+ only_when_attribute = value.get('attribute_equlas')
+ for xml_element in vm_xml.xpath(key):
+ if only_when_attribute:
+ for attr_key, attr_value in \
+ only_when_attribute.iteritems():
+ if attr_value != xml_element.get(attr_key):
+ continue
+ self.overwrite_xml(xml_element, value)
+
+ def define_vm(self, vm_name, temp_vm_file, disk_path,
+ vm_definition_overwrite):
log('Creating VM %s with disks %s' % (vm_name, disk_path))
with open(temp_vm_file) as f:
vm_xml = etree.parse(f)
@@ -60,10 +83,8 @@ class ExecutionEnvironment(object):
uuids = vm_xml.xpath('/domain/uuid')
for uuid in uuids:
uuid.getparent().remove(uuid)
- if number_cpus:
- vcpus = vm_xml.xpath('/domain/vcpu')
- for vcpu in vcpus:
- vcpu.text = str(number_cpus)
+ self.overwrite_xml(vm_xml.xpath('/domain')[0],
+ vm_definition_overwrite)
disks = vm_xml.xpath('/domain/devices/disk')
for disk in disks:
if (disk.get('type') == 'file' and