mirror of
https://github.com/gryf/boxpy.git
synced 2025-12-19 13:37:58 +01:00
574 lines
21 KiB
Python
Executable File
574 lines
21 KiB
Python
Executable File
#!/usr/bin/env python
|
|
|
|
import argparse
|
|
import os
|
|
import shutil
|
|
import string
|
|
import subprocess
|
|
import sys
|
|
import tempfile
|
|
import time
|
|
import uuid
|
|
import xml
|
|
|
|
|
|
META_DATA_TPL = string.Template('''\
|
|
instance-id: $instance_id
|
|
local-hostname: $vmhostname
|
|
''')
|
|
|
|
USER_DATA_TPL = string.Template('''\
|
|
#cloud-config
|
|
users:
|
|
- default
|
|
- name: ubuntu
|
|
ssh_authorized_keys:
|
|
- $ssh_key
|
|
chpasswd: { expire: False }
|
|
gecos: ubuntu
|
|
sudo: ALL=(ALL) NOPASSWD:ALL
|
|
groups: users, admin
|
|
power_state:
|
|
mode: poweroff
|
|
timeout: 10
|
|
condition: True
|
|
''')
|
|
|
|
|
|
class BoxError(Exception):
|
|
pass
|
|
|
|
|
|
class BoxNotFound(BoxError):
|
|
pass
|
|
|
|
|
|
class BoxVBoxFailure(BoxError):
|
|
pass
|
|
|
|
|
|
class BoxConvertionError(BoxError):
|
|
pass
|
|
|
|
|
|
class BoxSysCommandError(BoxError):
|
|
pass
|
|
|
|
|
|
class VBoxManage:
|
|
"""
|
|
Class for dealing with vboxmanage commands
|
|
"""
|
|
def __init__(self, name_or_uuid=None):
|
|
self.name_or_uuid = name_or_uuid
|
|
self.vm_info = {}
|
|
|
|
def get_vm_base_path(self):
|
|
path = self._get_vm_config()
|
|
if not path:
|
|
return
|
|
|
|
return os.path.dirname(path)
|
|
|
|
def get_disk_path(self):
|
|
path = self._get_vm_config()
|
|
if not path:
|
|
return
|
|
|
|
dom = xml.dom.minidom.parse(path)
|
|
if len(dom.getElementsByTagName('HardDisk')) != 1:
|
|
# don't know what to do with multiple discs
|
|
raise BoxError('Cannot deal with multiple attached disks, perhaps '
|
|
'you need to do this manually')
|
|
disk = dom.getElementsByTagName('HardDisk')[0]
|
|
location = disk.getAttribute('location')
|
|
if location.startswith('/'):
|
|
disk_path = location
|
|
else:
|
|
disk_path = os.path.join(self.get_vm_base_path(), location)
|
|
|
|
return disk_path
|
|
|
|
def get_media_size(self, media_path):
|
|
try:
|
|
out = subprocess.check_output(['vboxmanage', 'showmediuminfo',
|
|
media_path],
|
|
encoding=sys.getdefaultencoding(),
|
|
stderr=subprocess.DEVNULL)
|
|
except subprocess.CalledProcessError:
|
|
return None
|
|
|
|
for line in out.split('\n'):
|
|
if line.startswith('Capacity:'):
|
|
line = line.split('Capacity:')[1].strip()
|
|
|
|
if line.isnumeric():
|
|
return line
|
|
else:
|
|
return line.split(' ')[0].strip()
|
|
|
|
def get_vm_info(self):
|
|
try:
|
|
out = subprocess.check_output(['vboxmanage', 'showvminfo',
|
|
self.name_or_uuid],
|
|
encoding=sys.getdefaultencoding(),
|
|
stderr=subprocess.DEVNULL)
|
|
except subprocess.CalledProcessError:
|
|
return None
|
|
|
|
self.vm_info = {}
|
|
|
|
for line in out.split('\n'):
|
|
if line.startswith('Config file:'):
|
|
self.vm_info['config_file'] = line.split('Config '
|
|
'file:')[1].strip()
|
|
continue
|
|
|
|
if line.startswith('Memory size'):
|
|
mem = line.split('Memory size')[1].strip()
|
|
if mem.isnumeric():
|
|
self.vm_info['memory'] = mem
|
|
else:
|
|
# 12288MB
|
|
self.vm_info['memory'] = mem[:-2]
|
|
continue
|
|
|
|
if line.startswith('Number of CPUs:'):
|
|
self.vm_info['cpus'] = line.split('Number of CPUs:')[1].strip()
|
|
continue
|
|
|
|
if line.startswith('UUID:'):
|
|
self.vm_info['uuid'] = line.split('UUID:')[1].strip()
|
|
|
|
return self.vm_info
|
|
|
|
def poweroff(self, silent=False):
|
|
cmd = ['vboxmanage', 'controlvm', self.name_or_uuid, 'poweroff']
|
|
if silent:
|
|
subprocess.call(cmd, stderr=subprocess.DEVNULL)
|
|
else:
|
|
subprocess.call(cmd)
|
|
|
|
def vmlist(self, only_running=False, long_list=False):
|
|
subcommand = 'runningvms' if only_running else 'vms'
|
|
long_list = '-l' if long_list else '-s'
|
|
subprocess.call(['vboxmanage', 'list', subcommand, long_list])
|
|
|
|
def get_running_vms(self):
|
|
return subprocess.getoutput('vboxmanage list runningvms')
|
|
|
|
def destroy(self):
|
|
self.poweroff(silent=True)
|
|
if subprocess.call(['vboxmanage', 'unregistervm', self.name_or_uuid,
|
|
'--delete']) != 0:
|
|
raise BoxVBoxFailure(f'Removing VM {self.name_or_uuid} failed')
|
|
|
|
def create(self, cpus, memory):
|
|
self.uuid = None
|
|
|
|
try:
|
|
out = subprocess.check_output(['vboxmanage', 'createvm', '--name',
|
|
self.name_or_uuid, '--register'],
|
|
encoding=sys.getdefaultencoding())
|
|
except subprocess.CalledProcessError:
|
|
return None
|
|
|
|
for line in out.split('\n'):
|
|
print(line)
|
|
if line.startswith('UUID:'):
|
|
self.uuid = line.split('UUID:')[1].strip()
|
|
|
|
if not self.uuid:
|
|
raise BoxVBoxFailure(f'Cannot create VM "{self.name_or_uuid}".')
|
|
|
|
if subprocess.call(['vboxmanage', 'modifyvm', self.name_or_uuid,
|
|
'--memory', str(memory),
|
|
'--cpus', str(cpus),
|
|
'--boot1', 'disk',
|
|
'--acpi', 'on',
|
|
'--audio', 'none',
|
|
'--nic1', 'nat',
|
|
'--natpf1', 'guestssh,tcp,,2222,,22']) != 0:
|
|
raise BoxVBoxFailure(f'Cannot modify VM "{self.name_or_uuid}".')
|
|
|
|
return self.uuid
|
|
|
|
def convertfromraw(self, src, dst):
|
|
if subprocess.call(["vboxmanage", "convertfromraw", src, dst]) != 0:
|
|
os.unlink(src)
|
|
raise BoxVBoxFailure('Cannot convert image to VDI.')
|
|
os.unlink(src)
|
|
|
|
def closemedium(self, mediumpath):
|
|
subprocess.call(['vboxmanage', 'closemedium', 'dvd', mediumpath])
|
|
|
|
def create_controller(self, name, type_):
|
|
if subprocess.call(['vboxmanage', 'storagectl', self.name_or_uuid,
|
|
'--name', name, '--add', type_]) != 0:
|
|
raise BoxVBoxFailure(f'Adding controller {type_} has failed.')
|
|
|
|
def move_and_resize_image(self, src, dst, size):
|
|
fullpath = os.path.join(self.get_vm_base_path(), dst)
|
|
|
|
if subprocess.call(['vboxmanage', 'modifymedium', 'disk', src,
|
|
'--resize', str(size), '--move', fullpath]) != 0:
|
|
raise BoxVBoxFailure(f'Resizing and moving image {dst} has '
|
|
f'failed')
|
|
return fullpath
|
|
|
|
def storageattach(self, controller_name, port, type_, image):
|
|
if subprocess.call(['vboxmanage', 'storageattach', self.name_or_uuid,
|
|
'--storagectl', controller_name,
|
|
'--port', str(port),
|
|
'--device', '0',
|
|
'--type', type_,
|
|
'--medium', image]) != 0:
|
|
raise BoxVBoxFailure(f'Attaching {image} to VM has failed.')
|
|
|
|
def poweron(self):
|
|
if subprocess.call(['vboxmanage', 'startvm', self.name_or_uuid,
|
|
'--type', 'headless']) != 0:
|
|
raise BoxVBoxFailure(f'Failed to start: {self.name_or_uuid}.')
|
|
|
|
def _get_vm_config(self):
|
|
if self.vm_info.get('config_file'):
|
|
return self.vm_info['config_file']
|
|
|
|
self.get_vm_info()
|
|
|
|
if self.vm_info.get('config_file'):
|
|
return self.vm_info['config_file']
|
|
|
|
|
|
class VMCreate:
|
|
"""
|
|
Create vbox VM of Ubuntu server from cloud image with the following steps:
|
|
- grab the image, unless it exists in XDG_CACHE_HOME
|
|
- convert it to raw, than to VDI, remove raw
|
|
- resize it to the right size
|
|
- create cloud ISO image with some basic bootstrap
|
|
- create and register VM definition
|
|
- tweak its params
|
|
- move disk image to the Machine directory
|
|
- attach disk and iso images to it
|
|
- run and wait for initial bootstrap, than acpishutdown
|
|
- detach iso image and remove it
|
|
"""
|
|
CLOUD_IMAGE = "ci.iso"
|
|
CACHE_DIR = os.environ.get('XDG_CACHE_HOME',
|
|
os.path.expanduser('~/.cache'))
|
|
|
|
def __init__(self, args):
|
|
self.vm_name = args.name
|
|
self.cpus = args.cpus
|
|
self.memory = args.memory
|
|
self.disk_size = args.disk_size
|
|
self.ubuntu_version = args.version
|
|
self.hostname = args.hostname
|
|
self.ssh_key_path = args.key
|
|
|
|
if not self.ssh_key_path.endswith('.pub'):
|
|
self.ssh_key_path += '.pub'
|
|
if not os.path.exists(self.ssh_key_path):
|
|
self.ssh_key_path = os.path.join(os.path.expanduser("~/.ssh"),
|
|
self.ssh_key_path)
|
|
if not os.path.exists(self.ssh_key_path):
|
|
raise BoxNotFound(f'Cannot find default ssh public key: '
|
|
f'{self.ssh_key_path}')
|
|
|
|
self._img = f"ubuntu-{self.ubuntu_version}-server-cloudimg-amd64.img"
|
|
self._temp_path = None
|
|
self._disk_img = self.vm_name + '.vdi'
|
|
self._tmp = None
|
|
self._vm_base_path = None
|
|
self._vm_uuid = None
|
|
|
|
def run(self):
|
|
try:
|
|
self._prepare_temp()
|
|
self._download_image()
|
|
self._convert_and_resize()
|
|
self._create_and_setup_vm()
|
|
self._create_cloud_image()
|
|
self._attach_images_to_vm()
|
|
self._power_on_and_wait_for_ci_finish()
|
|
finally:
|
|
self._cleanup()
|
|
|
|
def _power_on_and_wait_for_ci_finish(self):
|
|
if subprocess.call(['vboxmanage', 'startvm', self._vm_uuid, '--type',
|
|
'headless']) != 0:
|
|
raise BoxVBoxFailure(f'Failed to start: {self.vm_name}.')
|
|
|
|
# give VBox some time to actually change the state of the VM before
|
|
# query
|
|
time.sleep(3)
|
|
|
|
# than, let's try to see if boostraping process has finished
|
|
print('Waiting for cloud init to finish')
|
|
while True:
|
|
if self._vm_uuid in subprocess.getoutput('vboxmanage list '
|
|
'runningvms'):
|
|
time.sleep(3)
|
|
else:
|
|
print('Done')
|
|
break
|
|
|
|
# detatch cloud image ISO
|
|
if subprocess.call(['vboxmanage', 'storageattach', self._vm_uuid,
|
|
'--storagectl', 'IDE',
|
|
'--port', '1',
|
|
'--device', '0',
|
|
'--type', 'dvddrive',
|
|
'--medium', 'none']) != 0:
|
|
raise BoxVBoxFailure(f'Failed to detach cloud image from '
|
|
f'{self.vm_name} VM.')
|
|
|
|
# and start it again
|
|
if subprocess.call(['vboxmanage', 'startvm', self._vm_uuid, '--type',
|
|
'headless']) != 0:
|
|
raise BoxVBoxFailure(f'Failed to start: {self.vm_name}.')
|
|
|
|
print('You can access your VM by issuing:')
|
|
print(f'ssh -p 2222 -i {self.ssh_key_path[:-4]} ubuntu@localhost')
|
|
|
|
def _attach_images_to_vm(self):
|
|
vdi_path = os.path.join(self._tmp, self._disk_img)
|
|
|
|
# couple of commands for changing the disk size, creating controllers
|
|
# and attaching disk and config drive to the vm.
|
|
# NOTE: modifymedium will register the disk image in Virtual Media
|
|
# Manager, while convertfromraw not.
|
|
commands = [['vboxmanage', 'modifymedium', 'disk', vdi_path,
|
|
'--resize', str(self.disk_size), '--move',
|
|
os.path.join(self._vm_base_path, self._disk_img)],
|
|
['vboxmanage', 'storagectl', self._vm_uuid, '--name',
|
|
'IDE', '--add', 'ide'],
|
|
['vboxmanage', 'storagectl', self._vm_uuid, '--name',
|
|
'SATA', '--add', 'sata'],
|
|
['vboxmanage', 'storageattach', self._vm_uuid,
|
|
'--storagectl', 'SATA',
|
|
'--port', '0',
|
|
'--device', '0',
|
|
'--type', 'hdd',
|
|
'--medium',
|
|
os.path.join(self._vm_base_path, self._disk_img)],
|
|
['vboxmanage', 'storageattach', self._vm_uuid,
|
|
'--storagectl', 'IDE',
|
|
'--port', '1',
|
|
'--device', '0',
|
|
'--type', 'dvddrive',
|
|
'--medium',
|
|
os.path.join(self._tmp, self.CLOUD_IMAGE)]]
|
|
for cmd in commands:
|
|
if subprocess.call(cmd) != 0:
|
|
cmd = ' '.join(cmd)
|
|
raise BoxVBoxFailure(f'command: {cmd} has failed')
|
|
|
|
def _create_and_setup_vm(self):
|
|
out = subprocess.check_output(['vboxmanage', 'createvm', '--name',
|
|
self.vm_name, '--register'],
|
|
encoding=sys.getdefaultencoding())
|
|
for line in out.split('\n'):
|
|
print(line)
|
|
if line.startswith('UUID:'):
|
|
self._vm_uuid = line.split('UUID:')[1].strip()
|
|
|
|
if not self._vm_uuid:
|
|
raise BoxVBoxFailure(f'Cannot create VM "{self.vm_name}".')
|
|
|
|
if subprocess.call(['vboxmanage', 'modifyvm', self._vm_uuid,
|
|
'--memory', str(self.memory),
|
|
'--cpus', str(self.cpus),
|
|
'--boot1', 'disk',
|
|
'--acpi', 'on',
|
|
'--audio', 'none',
|
|
'--nic1', 'nat',
|
|
'--natpf1', 'guestssh,tcp,,2222,,22']) != 0:
|
|
raise BoxVBoxFailure(f'Cannot modify VM "{self._vm_uuid}".')
|
|
out = subprocess.check_output(['vboxmanage', 'showvminfo',
|
|
self._vm_uuid],
|
|
encoding=sys.getdefaultencoding())
|
|
path = None
|
|
for line in out.split('\n'):
|
|
if line.startswith('Config file:'):
|
|
path = os.path.dirname(line.split('Config file:')[1].strip())
|
|
|
|
if not path:
|
|
raise BoxVBoxFailure(f'There is something wrong doing VM '
|
|
f'"{self.vm_name}" creation and registration')
|
|
|
|
self._vm_base_path = path
|
|
|
|
def _create_cloud_image(self):
|
|
# meta-data
|
|
with open(os.path.join(self._tmp, 'meta-data'), 'w') as fobj:
|
|
fobj.write(META_DATA_TPL
|
|
.substitute({'instance_id': str(uuid.uuid4()),
|
|
'vmhostname': self.hostname}))
|
|
|
|
# user-data
|
|
with open(self.ssh_key_path) as fobj:
|
|
ssh_pub_key = fobj.read().strip()
|
|
|
|
with open(os.path.join(self._tmp, 'user-data'), 'w') as fobj:
|
|
fobj.write(USER_DATA_TPL.substitute({'ssh_key': ssh_pub_key}))
|
|
|
|
mkiso = 'mkisofs' if shutil.which('mkisofs') else 'genisoimage'
|
|
|
|
# create ISO image
|
|
if subprocess.call([mkiso, '-J', '-R', '-V', 'cidata', '-o',
|
|
os.path.join(self._tmp, self.CLOUD_IMAGE),
|
|
os.path.join(self._tmp, 'user-data'),
|
|
os.path.join(self._tmp, 'meta-data')]) != 0:
|
|
raise BoxSysCommandError('Cannot create ISO image for config '
|
|
'drive')
|
|
|
|
def _prepare_temp(self):
|
|
self._tmp = tempfile.mkdtemp()
|
|
|
|
def _checksum(self):
|
|
expected_sum = None
|
|
fname = 'SHA256SUMS'
|
|
url = "https://cloud-images.ubuntu.com/releases/"
|
|
url += f"{self.ubuntu_version}/release/{fname}"
|
|
# TODO: make the verbosity switch be dependent from verbosity of the
|
|
# script.
|
|
subprocess.call(['wget', url, '-q', '-O',
|
|
os.path.join(self._tmp, fname)])
|
|
|
|
with open(os.path.join(self._tmp, fname)) as fobj:
|
|
for line in fobj.readlines():
|
|
if self._img in line:
|
|
expected_sum = line.split(' ')[0]
|
|
break
|
|
|
|
if not expected_sum:
|
|
raise BoxError('Cannot find provided cloud image')
|
|
|
|
if os.path.exists(os.path.join(self.CACHE_DIR, self._img)):
|
|
cmd = 'sha256sum ' + os.path.join(self.CACHE_DIR, self._img)
|
|
calulated_sum = subprocess.getoutput(cmd).split(' ')[0]
|
|
return calulated_sum == expected_sum
|
|
|
|
return False
|
|
|
|
def _convert_to_raw(self):
|
|
img_path = os.path.join(self.CACHE_DIR, self._img)
|
|
raw_path = os.path.join(self._tmp, self._img + ".raw")
|
|
if subprocess.call(['qemu-img', 'convert', '-O', 'raw',
|
|
img_path, raw_path]) != 0:
|
|
raise BoxConvertionError(f'Cannot convert image {self._img} to '
|
|
'RAW.')
|
|
|
|
def _convert_and_resize(self):
|
|
self._convert_to_raw()
|
|
raw_path = os.path.join(self._tmp, self._img + ".raw")
|
|
vdi_path = os.path.join(self._tmp, self._disk_img)
|
|
if subprocess.call(["vboxmanage", "convertfromraw", raw_path,
|
|
vdi_path]) != 0:
|
|
raise BoxVBoxFailure(f'Cannot convert image {self._disk_img} '
|
|
f'to VDI.')
|
|
os.unlink(raw_path)
|
|
|
|
def _download_image(self):
|
|
if self._checksum():
|
|
print(f'Image already downloaded: {self._img}')
|
|
return
|
|
|
|
url = "https://cloud-images.ubuntu.com/releases/"
|
|
url += f"{self.ubuntu_version}/release/"
|
|
img = f"ubuntu-{self.ubuntu_version}-server-cloudimg-amd64.img"
|
|
url += img
|
|
print(f'Downloading image {self._img}')
|
|
subprocess.call(['wget', '-q', url, '-O',
|
|
os.path.join(self.CACHE_DIR, self._img)])
|
|
|
|
if not self._checksum():
|
|
# TODO: make some retry mechanism?
|
|
raise BoxSysCommandError('Checksum for downloaded image differ '
|
|
'from expected.')
|
|
else:
|
|
print(f'Downloaded image {self._img}')
|
|
|
|
def _cleanup(self):
|
|
subprocess.call(['vboxmanage', 'closemedium', 'dvd',
|
|
os.path.join(self._tmp, self.CLOUD_IMAGE)])
|
|
subprocess.call(['rm', '-fr', self._tmp])
|
|
|
|
|
|
class VMDestroy:
|
|
def __init__(self, args):
|
|
self.vm_name_or_uuid = args.name
|
|
|
|
def run(self):
|
|
subprocess.call(['vboxmanage', 'controlvm', self.vm_name_or_uuid,
|
|
'poweroff'], stderr=subprocess.DEVNULL)
|
|
if subprocess.call(['vboxmanage', 'unregistervm', self.vm_name_or_uuid,
|
|
'--delete']) != 0:
|
|
raise BoxVBoxFailure(f'Removing VM {self.vm_name_or_uuid} failed')
|
|
|
|
|
|
|
|
class VMList:
|
|
def __init__(self, args):
|
|
self.running = args.running
|
|
self.long = args.long
|
|
|
|
def run(self):
|
|
subcommand = 'runningvms' if self.running else 'vms'
|
|
long_list = '-l' if self.long else '-s'
|
|
|
|
subprocess.call(['vboxmanage', 'list', subcommand, long_list])
|
|
|
|
|
|
def main():
|
|
parser = argparse.ArgumentParser(description="Automate deployment and "
|
|
"maintenance of Ubuntu VMs using "
|
|
"VirtualBox and Ubuntu cloud images")
|
|
subparsers = parser.add_subparsers(help='supported commands')
|
|
|
|
create = subparsers.add_parser('create', help='create and configure VM, '
|
|
'create corresponding assets, config '
|
|
'drive and run')
|
|
create.set_defaults(func=VMCreate)
|
|
create.add_argument('name', help='name of the VM')
|
|
create.add_argument('-m', '--memory', default=12288, type=int,
|
|
help="amount of memory in Megabytes, default 12GB")
|
|
create.add_argument('-c', '--cpus', default=6, type=int,
|
|
help="amount of CPUs to be configured. Default 6.")
|
|
create.add_argument('-d', '--disk-size', default=32768, type=int,
|
|
help="disk size to be expanded to. By default to 32GB")
|
|
create.add_argument('-v', '--version', default="18.04",
|
|
help="Ubuntu server version. Default 18.04")
|
|
create.add_argument('-n', '--hostname', default="ubuntu",
|
|
help="VM hostname. Default ubuntu")
|
|
create.add_argument('-k', '--key',
|
|
default=os.path.expanduser("~/.ssh/id_rsa"),
|
|
help="SSH key to be add to the config drive. Default "
|
|
"~/.ssh/id_rsa")
|
|
|
|
destroy = subparsers.add_parser('destroy', help='destroy VM')
|
|
destroy.add_argument('name', help='name or UUID of the VM')
|
|
destroy.set_defaults(func=VMDestroy)
|
|
|
|
list_vms = subparsers.add_parser('list', help='list VMs')
|
|
list_vms.add_argument('-l', '--long', action='store_true',
|
|
help='show detailed information '
|
|
'about VMs')
|
|
list_vms.add_argument('-r', '--running', action='store_true',
|
|
help='show only running VMs')
|
|
list_vms.set_defaults(func=VMList)
|
|
|
|
args = parser.parse_args()
|
|
|
|
try:
|
|
return args.func(args).run()
|
|
except AttributeError:
|
|
parser.print_help()
|
|
parser.exit()
|
|
|
|
|
|
if __name__ == '__main__':
|
|
main()
|