58 lines
1.9 KiB
Python
Executable File
58 lines
1.9 KiB
Python
Executable File
#!/usr/bin/env python3
|
|
|
|
# Vitastor OpenNebula driver
|
|
# Copyright (c) Vitaliy Filippov, 2024+
|
|
# License: Apache-2.0 http://www.apache.org/licenses/LICENSE-2.0
|
|
|
|
import base64
|
|
from sys import argv, stderr
|
|
from xml.etree import ElementTree as ET
|
|
|
|
dep_file = argv[1]
|
|
with open(dep_file, 'rb') as fd:
|
|
dep_txt = base64.b64decode(fd.read())
|
|
dep = ET.fromstring(dep_txt)
|
|
|
|
vm_file = argv[2]
|
|
with open(vm_file, 'rb') as fd:
|
|
vm_txt = base64.b64decode(fd.read())
|
|
vm = ET.fromstring(vm_txt)
|
|
|
|
ET.register_namespace('qemu', 'http://libvirt.org/schemas/domain/qemu/1.0')
|
|
ET.register_namespace('one', 'http://opennebula.org/xmlns/libvirt/1.0')
|
|
|
|
vm_id = vm.find('./ID').text
|
|
context_disk_id = vm.find('./TEMPLATE/CONTEXT/DISK_ID').text
|
|
changed = 0
|
|
txt = lambda x: '' if x is None else x.text
|
|
|
|
for disk in dep.findall('./devices/disk[@type="file"]'):
|
|
try:
|
|
disk_id = disk.find('./source').attrib['file'].split('.')[-1]
|
|
vm_disk = vm.find('./TEMPLATE/DISK[DISK_ID="{}"]'.format(disk_id))
|
|
if vm_disk is None:
|
|
continue
|
|
tm_mad = txt(vm_disk.find('./TM_MAD'))
|
|
if tm_mad != 'vitastor':
|
|
continue
|
|
src_image = txt(vm_disk.find('./SOURCE'))
|
|
clone = txt(vm_disk.find('./CLONE'))
|
|
vitastor_conf = txt(vm_disk.find('./VITASTOR_CONF'))
|
|
if clone == "YES":
|
|
src_image += "-"+vm_id+"-"+disk_id
|
|
# modify
|
|
changed = 1
|
|
disk.attrib['type'] = 'network'
|
|
disk.remove(disk.find('./source'))
|
|
src = ET.SubElement(disk, 'source')
|
|
src.attrib['protocol'] = 'vitastor'
|
|
src.attrib['name'] = src_image
|
|
if vitastor_conf:
|
|
# path to config should be added to /etc/apparmor.d/local/abstractions/libvirt-qemu
|
|
config = ET.SubElement(src, 'config')
|
|
config.text = vitastor_conf
|
|
except Exception as e:
|
|
print("Error: {}".format(e), file=stderr)
|
|
|
|
ET.ElementTree(dep).write(dep_file)
|