1
0
mirror of https://github.com/gryf/boxpy.git synced 2026-02-01 13:35:52 +01:00

6 Commits

Author SHA1 Message Date
60e45b9bfc Added support for Fedora 42 2025-05-20 11:27:21 +02:00
88172acd67 Refactor Run class.
When using capture=True together with subprocess.run, interactive
session will not work, as all the output will be captured. Perfectly
well seen with `boxpy ssh`. This commit will fix this issue.
2025-05-05 13:09:09 +02:00
4c3aef3a85 Fix ssh command.
There was a leftover call for changed Run object initialization.
With this commit it's correct now.
2025-05-02 12:42:17 +02:00
189fbab17d Removed license classifier in favor of SPDX entry. 2025-04-18 16:04:42 +02:00
eeebab74ad Fix some linting issues.
Also, changing workflow for image downloading fails.
2024-11-17 19:27:37 +01:00
81ab5de7c4 Bump Fedora version 2024-11-15 08:18:46 +01:00
3 changed files with 103 additions and 64 deletions

View File

@@ -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
View File

@@ -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__':

View File

@@ -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
]