mirror of
https://github.com/gryf/boxpy.git
synced 2026-02-01 13:35:52 +01:00
Compare commits
6 Commits
| Author | SHA1 | Date | |
|---|---|---|---|
| 60e45b9bfc | |||
| 88172acd67 | |||
| 4c3aef3a85 | |||
| 189fbab17d | |||
| eeebab74ad | |||
| 81ab5de7c4 |
@@ -37,11 +37,14 @@ Tested distros
|
||||
- 20.04
|
||||
- 22.04
|
||||
- 24.04
|
||||
- 25.04
|
||||
- Fedora
|
||||
- 37
|
||||
- 38
|
||||
- 39
|
||||
- 40
|
||||
- 41
|
||||
- 42
|
||||
- Centos Stream
|
||||
- 8
|
||||
- 9
|
||||
|
||||
135
box.py
135
box.py
@@ -18,7 +18,7 @@ import requests
|
||||
import yaml
|
||||
|
||||
|
||||
__version__ = "1.10.1"
|
||||
__version__ = "1.11.2"
|
||||
|
||||
CACHE_DIR = os.environ.get('XDG_CACHE_HOME', os.path.expanduser('~/.cache'))
|
||||
CLOUD_IMAGE = "ci.iso"
|
||||
@@ -275,28 +275,29 @@ class Run:
|
||||
Helper class on subprocess.run()
|
||||
command is a list with command and its params to execute
|
||||
"""
|
||||
def __init__(self, command, capture_output=True):
|
||||
def __init__(self, command, capture=True):
|
||||
result = subprocess.run(command, encoding='utf-8',
|
||||
capture_output=capture_output)
|
||||
if result.stdout:
|
||||
LOG.debug2(result.stdout)
|
||||
if result.stderr:
|
||||
LOG.debug2(result.stderr)
|
||||
capture_output=capture)
|
||||
|
||||
self.returncode = result.returncode
|
||||
self.stdout = result.stdout.strip() if result.stdout else ''
|
||||
self.stderr = result.stderr.strip() if result.stderr else ''
|
||||
|
||||
if self.stdout:
|
||||
LOG.debug2(self.stdout)
|
||||
if self.stderr:
|
||||
LOG.debug2(self.stderr)
|
||||
|
||||
|
||||
class BoxError(Exception):
|
||||
pass
|
||||
|
||||
|
||||
class BoxNotFound(BoxError):
|
||||
class BoxNotFoundError(BoxError):
|
||||
pass
|
||||
|
||||
|
||||
class BoxVBoxFailure(BoxError):
|
||||
class BoxVBoxError(BoxError):
|
||||
pass
|
||||
|
||||
|
||||
@@ -524,7 +525,7 @@ class Config:
|
||||
def _read_filename(self, fname):
|
||||
fullpath = os.path.expanduser(os.path.expandvars(fname))
|
||||
if not os.path.exists(fullpath):
|
||||
return
|
||||
return None
|
||||
|
||||
with open(fname) as fobj:
|
||||
return fobj.read()
|
||||
@@ -633,11 +634,11 @@ class OsTypes:
|
||||
|
||||
def ubuntu(self):
|
||||
lts = ''
|
||||
major, minor = [int(x) for x in self._conf.version.split('.')]
|
||||
major, minor = (int(x) for x in self._conf.version.split('.'))
|
||||
|
||||
if major % 2 == 0 and minor == 4:
|
||||
lts = '_LTS'
|
||||
name = "Ubuntu%s%s_64" % (major, lts)
|
||||
name = f"Ubuntu{major}{lts}_64"
|
||||
|
||||
if name not in self._ostypes:
|
||||
return 'Ubuntu_64'
|
||||
@@ -648,7 +649,7 @@ class OsTypes:
|
||||
return "Fedora_64"
|
||||
|
||||
def debian(self):
|
||||
name = "Debian%s_64" % self._conf.version
|
||||
name = f"Debian{self._conf.version}_64"
|
||||
if name not in self._ostypes:
|
||||
return 'Debian_64'
|
||||
|
||||
@@ -674,7 +675,7 @@ class VBoxManage:
|
||||
def get_vm_base_path(self):
|
||||
path = self._get_vm_config()
|
||||
if not path:
|
||||
return
|
||||
return None
|
||||
|
||||
return os.path.dirname(path)
|
||||
|
||||
@@ -682,7 +683,7 @@ class VBoxManage:
|
||||
path = self._get_vm_config()
|
||||
if not path:
|
||||
LOG.warning('Configuration for "%s" not found', self.name_or_uuid)
|
||||
return
|
||||
return None
|
||||
|
||||
dom = xml.dom.minidom.parse(path)
|
||||
if len(dom.getElementsByTagName('HardDisk')) != 1:
|
||||
@@ -709,6 +710,7 @@ class VBoxManage:
|
||||
return line
|
||||
|
||||
return line.split(' ')[0].strip()
|
||||
return None
|
||||
|
||||
def get_vm_info(self):
|
||||
out = Run(['vboxmanage', 'showvminfo', self.name_or_uuid])
|
||||
@@ -809,6 +811,7 @@ class VBoxManage:
|
||||
'--delete']).returncode != 0:
|
||||
LOG.fatal('Removing VM "%s" failed', self.name_or_uuid)
|
||||
return 7
|
||||
return None
|
||||
|
||||
def create(self, conf):
|
||||
memory = convert_to_mega(conf.memory)
|
||||
@@ -831,7 +834,7 @@ class VBoxManage:
|
||||
|
||||
if not self.uuid:
|
||||
msg = f'Cannot create VM "{self.name_or_uuid}".'
|
||||
raise BoxVBoxFailure(msg)
|
||||
raise BoxVBoxError(msg)
|
||||
|
||||
port = conf.port if conf.port else self._find_unused_port()
|
||||
|
||||
@@ -854,14 +857,15 @@ class VBoxManage:
|
||||
|
||||
if Run(cmd).returncode != 0:
|
||||
LOG.fatal(f'Cannot modify VM "{self.name_or_uuid}"')
|
||||
raise BoxVBoxFailure
|
||||
raise BoxVBoxError
|
||||
|
||||
if conf.disable_nested == 'False':
|
||||
if Run(['vboxmanage', 'modifyvm', self.name_or_uuid,
|
||||
'--nested-hw-virt', 'on']).returncode != 0:
|
||||
LOG.fatal(f'Cannot set nested virtualization for VM '
|
||||
f'"{self.name_or_uuid}"')
|
||||
raise BoxVBoxFailure
|
||||
if conf.disable_nested == 'False' and Run(['vboxmanage', 'modifyvm',
|
||||
self.name_or_uuid,
|
||||
'--nested-hw-virt',
|
||||
'on']).returncode != 0:
|
||||
LOG.fatal(f'Cannot set nested virtualization for VM '
|
||||
f'"{self.name_or_uuid}"')
|
||||
raise BoxVBoxError
|
||||
|
||||
return self.uuid
|
||||
|
||||
@@ -897,7 +901,7 @@ class VBoxManage:
|
||||
if Run(['vboxmanage', 'modifymedium', 'disk', src, '--resize',
|
||||
str(size), '--move', fullpath]).returncode != 0:
|
||||
LOG.fatal('Resizing and moving image %s has failed', dst)
|
||||
raise BoxVBoxFailure
|
||||
raise BoxVBoxError
|
||||
return fullpath
|
||||
|
||||
def storageattach(self, controller_name, port, type_, image):
|
||||
@@ -921,7 +925,7 @@ class VBoxManage:
|
||||
if Run(['vboxmanage', 'startvm', self.name_or_uuid, '--type',
|
||||
type_]).returncode != 0:
|
||||
LOG.fatal('Failed to start: %s', self.name_or_uuid)
|
||||
raise BoxVBoxFailure
|
||||
raise BoxVBoxError
|
||||
|
||||
def setextradata(self, key, val):
|
||||
res = Run(['vboxmanage', 'setextradata', self.name_or_uuid, key, val])
|
||||
@@ -935,7 +939,7 @@ class VBoxManage:
|
||||
if Run(['vboxmanage', 'modifyvm', self.name_or_uuid, f'--{nic}',
|
||||
kind]).returncode != 0:
|
||||
LOG.fatal('Cannot modify VM "%s"', self.name_or_uuid)
|
||||
raise BoxVBoxFailure
|
||||
raise BoxVBoxError
|
||||
|
||||
def is_port_in_use(self, port):
|
||||
used_ports = self._get_defined_ports()
|
||||
@@ -1038,6 +1042,7 @@ class Image:
|
||||
return False
|
||||
|
||||
LOG.info('Calculating checksum for "%s"', self._img_fname)
|
||||
LOG.debug('Checksum file: "%s"', self._checksum_file)
|
||||
fname = os.path.join(self._tmp, self._checksum_file)
|
||||
expected_sum = self._get_checksum(fname)
|
||||
|
||||
@@ -1061,7 +1066,10 @@ class Image:
|
||||
|
||||
fname = os.path.join(CACHE_DIR, self._img_fname)
|
||||
LOG.header('Downloading image %s', self._img_url)
|
||||
Run(['wget', '-q', self._img_url, '-O', fname])
|
||||
result = Run(['wget', '-q', self._img_url, '-O', fname])
|
||||
if result.returncode != 0:
|
||||
LOG.fatal("Error downloading image %s", self._img_url)
|
||||
return False
|
||||
|
||||
if not self._checksum():
|
||||
# TODO: make some retry mechanism?
|
||||
@@ -1135,14 +1143,24 @@ class Fedora(Image):
|
||||
REVISION = {'37': '1.7',
|
||||
'38': '1.6',
|
||||
'39': '1.5',
|
||||
'40': '1.14'}
|
||||
'40': '1.14',
|
||||
'41': '1.4',
|
||||
'42': '1.1'}
|
||||
|
||||
def __init__(self, vbox, version, arch, fname=None):
|
||||
super().__init__(vbox, version, arch)
|
||||
revision = self.REVISION[version]
|
||||
if int(version) > 39:
|
||||
self.IMG = "Fedora-Cloud-Base-Generic.%s-%s-%s.qcow2"
|
||||
self._img_fname = self.IMG % (arch, version, revision)
|
||||
if int(version) >= 40:
|
||||
if int(version) == 40:
|
||||
# Started from Fedora 40 there is "Generic" in the image names.
|
||||
self.IMG = "Fedora-Cloud-Base-Generic.%s-%s-%s.qcow2"
|
||||
# Fedora 40 have messed up position of the items in filename.
|
||||
self._img_fname = self.IMG % (arch, version, revision)
|
||||
else:
|
||||
# But in Fedora 41 there is no dot between Generic and
|
||||
# version, but between version and arch.
|
||||
self.IMG = "Fedora-Cloud-Base-Generic-%s-%s.%s.qcow2"
|
||||
self._img_fname = self.IMG % (version, revision, arch)
|
||||
else:
|
||||
self._img_fname = self.IMG % (version, revision, arch)
|
||||
self._img_url = self.URL % (version, arch, self._img_fname)
|
||||
@@ -1170,8 +1188,7 @@ class CentosStream(Image):
|
||||
|
||||
def __init__(self, vbox, version, arch, release, fname=None):
|
||||
super().__init__(vbox, version, arch, release)
|
||||
self._checksum_file = '%s-centos-stream-%s-%s' % (self.CHKS, version,
|
||||
arch)
|
||||
self._checksum_file = f'{self.CHKS}-centos-stream-{version}-{arch}'
|
||||
self._checksum_url = self.URL % (version, arch, self.CHKS)
|
||||
# there is assumption, that we always need latest relese for specific
|
||||
# version and architecture.
|
||||
@@ -1198,6 +1215,7 @@ class CentosStream(Image):
|
||||
images.reverse()
|
||||
if images:
|
||||
return images[0]
|
||||
return None
|
||||
|
||||
def _get_checksum(self, fname):
|
||||
expected_sum = None
|
||||
@@ -1229,7 +1247,7 @@ DISTROS = {'ubuntu': {'username': 'ubuntu',
|
||||
'realname': 'fedora',
|
||||
'img_class': Fedora,
|
||||
'amd64': 'x86_64',
|
||||
'default_version': '40'},
|
||||
'default_version': '41'},
|
||||
'centos': {'username': 'centos',
|
||||
'realname': 'centos',
|
||||
'img_class': CentosStream,
|
||||
@@ -1326,9 +1344,8 @@ def vmcreate(args, conf=None):
|
||||
if not vbox.setextradata(key, getattr(conf, key)):
|
||||
return 5
|
||||
|
||||
if conf.user_data:
|
||||
if not vbox.setextradata('user_data', conf.user_data):
|
||||
return 6
|
||||
if conf.user_data and not vbox.setextradata('user_data', conf.user_data):
|
||||
return 6
|
||||
|
||||
if not vbox.setextradata('creator', 'boxpy'):
|
||||
return 13
|
||||
@@ -1434,24 +1451,21 @@ def vmcreate(args, conf=None):
|
||||
|
||||
# reread config to update fields
|
||||
conf = Config(args, vbox)
|
||||
username = DISTROS[conf.distro]["username"]
|
||||
LOG.info('You can access your VM by issuing:')
|
||||
if conf.username and conf.username != username:
|
||||
LOG.info(f'ssh -p {conf.port} -i {conf.ssh_key_path[:-4]} '
|
||||
f'{conf.username}@localhost')
|
||||
username = conf.username
|
||||
else:
|
||||
LOG.info(f'ssh -p {conf.port} -i {conf.ssh_key_path[:-4]} '
|
||||
f'{username}@localhost')
|
||||
username = DISTROS[conf.distro]["username"]
|
||||
LOG.info('You can access your VM by issuing:')
|
||||
LOG.info('ssh -p %s -i %s %s@localhost', conf.port, conf.ssh_key_path[:-4],
|
||||
username)
|
||||
LOG.info('or simply:')
|
||||
LOG.info(f'boxpy ssh {conf.name}')
|
||||
LOG.info('boxpy ssh %s', conf.name)
|
||||
return 0
|
||||
|
||||
|
||||
def vmdestroy(args):
|
||||
if isinstance(args.name, list):
|
||||
vm_names = args.name
|
||||
else:
|
||||
vm_names = [args.name]
|
||||
|
||||
vm_names = args.name if isinstance(args.name, list) else [args.name]
|
||||
|
||||
for name in vm_names:
|
||||
vbox = VBoxManage(name)
|
||||
@@ -1545,7 +1559,8 @@ def vminfo(args):
|
||||
LOG.info(line)
|
||||
|
||||
if 'user_data' in info:
|
||||
LOG.info(f'User data file path:\t{info["user_data"]}')
|
||||
LOG.info('User data file path:\t%s', info['user_data'])
|
||||
return 0
|
||||
|
||||
|
||||
def vmrebuild(args):
|
||||
@@ -1553,12 +1568,12 @@ def vmrebuild(args):
|
||||
if not vbox.get_vm_info():
|
||||
LOG.fatal(f'Cannot rebuild VM "{args.name}" - it doesn\'t exists.')
|
||||
return 20
|
||||
else:
|
||||
LOG.header('Rebuilding VM: %s', args.name)
|
||||
|
||||
LOG.header('Rebuilding VM: %s', args.name)
|
||||
|
||||
try:
|
||||
conf = Config(args, vbox)
|
||||
except BoxNotFound as ex:
|
||||
except BoxNotFoundError as ex:
|
||||
LOG.fatal(f'Error with parsing config: {ex}')
|
||||
return 8
|
||||
except yaml.YAMLError:
|
||||
@@ -1607,7 +1622,7 @@ def connect(args):
|
||||
|
||||
try:
|
||||
conf = Config(args, vbox)
|
||||
except BoxNotFound:
|
||||
except BoxNotFoundError:
|
||||
return 11
|
||||
except yaml.YAMLError:
|
||||
LOG.fatal(f'Cannot read or parse file `{args.config}` as YAML '
|
||||
@@ -1622,7 +1637,7 @@ def connect(args):
|
||||
f'@localhost:{conf.port}']
|
||||
LOG.debug('Connecting to vm `%s` using command:\n%s', args.name,
|
||||
' '.join(cmd))
|
||||
return Run(cmd, False).returncode
|
||||
return Run(cmd, capture=False).returncode
|
||||
|
||||
|
||||
def _set_vmstate(name, state, guitype=None, poweroff=False):
|
||||
@@ -1633,12 +1648,12 @@ def _set_vmstate(name, state, guitype=None, poweroff=False):
|
||||
return 20
|
||||
|
||||
if vbox.running and state == "start":
|
||||
LOG.info(f'VM "{name}" is already running.')
|
||||
return
|
||||
LOG.info('VM "%s" is already running.', name)
|
||||
return 1
|
||||
|
||||
if not vbox.running and state == "stop":
|
||||
LOG.info(f'VM "{name}" is already stopped.')
|
||||
return
|
||||
LOG.info('VM "%s" is already stopped.', name)
|
||||
return 1
|
||||
|
||||
if state == "start":
|
||||
vbox.poweron(guitype)
|
||||
@@ -1646,6 +1661,7 @@ def _set_vmstate(name, state, guitype=None, poweroff=False):
|
||||
vbox.poweroff()
|
||||
else:
|
||||
vbox.acpipowerbutton()
|
||||
return 0
|
||||
|
||||
|
||||
def vmstart(args):
|
||||
@@ -1798,7 +1814,7 @@ def main():
|
||||
LOG.set_verbose(args.verbose, args.quiet)
|
||||
|
||||
if 'func' not in args and args.version:
|
||||
LOG.info(f'boxpy {__version__}')
|
||||
LOG.info('boxpy %s', __version__)
|
||||
parser.exit()
|
||||
|
||||
if hasattr(args, 'func'):
|
||||
@@ -1806,6 +1822,7 @@ def main():
|
||||
|
||||
parser.print_help()
|
||||
parser.exit()
|
||||
return 23
|
||||
|
||||
|
||||
if __name__ == '__main__':
|
||||
|
||||
@@ -1,5 +1,5 @@
|
||||
[build-system]
|
||||
requires = ["setuptools >= 61.0"]
|
||||
requires = ["setuptools >= 77.0"]
|
||||
build-backend = "setuptools.build_meta"
|
||||
|
||||
[project]
|
||||
@@ -8,7 +8,7 @@ dynamic = ["version"]
|
||||
authors = [
|
||||
{name = "Roman Dobosz", email = "gryf73@gmail.com"}
|
||||
]
|
||||
license = {text = "GPLv3"}
|
||||
license = "GPL-3.0-or-later"
|
||||
description = "Run Linux cloud image on top of VirtualBox using commandline tool"
|
||||
readme = "README.rst"
|
||||
requires-python = ">=3.8"
|
||||
@@ -18,12 +18,12 @@ classifiers = [
|
||||
"Intended Audience :: End Users/Desktop",
|
||||
"Topic :: Terminals",
|
||||
"Topic :: Utilities",
|
||||
"License :: OSI Approved :: GNU General Public License v3 (GPLv3)",
|
||||
"Programming Language :: Python :: 3",
|
||||
"Programming Language :: Python :: 3.8",
|
||||
"Programming Language :: Python :: 3.9",
|
||||
"Programming Language :: Python :: 3.10",
|
||||
"Programming Language :: Python :: 3.11",
|
||||
"Programming Language :: Python :: 3.12",
|
||||
"Programming Language :: Python :: 3 :: Only"
|
||||
]
|
||||
dependencies = [
|
||||
@@ -43,5 +43,24 @@ py-modules = ["box"]
|
||||
[tool.setuptools.dynamic]
|
||||
version = {attr = "box.__version__"}
|
||||
|
||||
[tool.distutils.bdist_wheel]
|
||||
universal = true
|
||||
[tool.ruff]
|
||||
line-length = 79
|
||||
indent-width = 4
|
||||
|
||||
[tool.ruff.lint]
|
||||
extend-select = [
|
||||
"B", # flake8-bugbear
|
||||
"C4", # flake8-comprehensions
|
||||
"E", # pycodestyle
|
||||
"F", # pyflakes
|
||||
"FA", # flake8-future-annotations
|
||||
"G", # flake8-logging-format
|
||||
"N", # pep8-naming
|
||||
"PGH", # pygrep-hooks
|
||||
"PIE", # flake8-pie
|
||||
"RET", # flake8-return
|
||||
"SIM", # flake8-simplify
|
||||
"UP", # pyupgrade
|
||||
"W", # pycodestyle
|
||||
"YTT", # flake8-2020
|
||||
]
|
||||
|
||||
Reference in New Issue
Block a user