Skip to content

Commit

Permalink
build-script: T3664: Allowed all options in both config file and comm…
Browse files Browse the repository at this point in the history
…and args

Moved defaults away from argparser to `defaults.py`. This unlocks the ability to
pass values that can be defined as command line arguments via a config file.

With this change logic looks like this (in order of overrides).

Pre-build config:
`data/defaults.toml` -> `build-flavors/<flavor>.toml` ->
`--<command line argument>`

Build config:
`defaults.py` -> `data/defaults.toml` -> `build-types/<type>.toml` ->
`architectures/<architecture>.toml` -> `build-flavors/<flavor>.toml` ->
`--<command line argument>`

(cherry picked from commit 8186e82)
  • Loading branch information
zdc authored and mergify[bot] committed May 17, 2024
1 parent 3ac9f6a commit 31304ab
Show file tree
Hide file tree
Showing 2 changed files with 37 additions and 26 deletions.
47 changes: 21 additions & 26 deletions scripts/image-build/build-vyos-image
Original file line number Diff line number Diff line change
Expand Up @@ -26,8 +26,6 @@ import uuid
import glob
import json
import shutil
import getpass
import platform
import argparse
import datetime
import functools
Expand Down Expand Up @@ -130,12 +128,9 @@ except OSError as e:
def field_to_option(s):
return re.sub(r'_', '-', s)

def get_default_build_by():
return "{user}@{host}".format(user= getpass.getuser(), host=platform.node())

def get_validator(optdict, name):
try:
return optdict[name][2]
return optdict[name][1]
except KeyError:
return None

Expand Down Expand Up @@ -190,38 +185,35 @@ if __name__ == "__main__":
# Options dict format:
# '$option_name_without_leading_dashes': { ('$help_string', $default_value_generator_thunk, $value_checker_thunk) }
options = {
'architecture': ('Image target architecture (amd64 or arm64)', None, lambda x: x in ['amd64', 'arm64', None]),
'build-by': ('Builder identifier (e.g. jrandomhacker@example.net)', get_default_build_by, None),
'debian-mirror': ('Debian repository mirror', None, None),
'debian-security-mirror': ('Debian security updates mirror', None, None),
'pbuilder-debian-mirror': ('Debian repository mirror for pbuilder env bootstrap', None, None),
'vyos-mirror': ('VyOS package mirror', None, None),
'build-type': ('Build type, release or development', None, lambda x: x in ['release', 'development']),
'version': ('Version number (release builds only)', None, None),
'build-comment': ('Optional build comment', lambda: '', None)
'architecture': ('Image target architecture (amd64 or arm64)', lambda x: x in ['amd64', 'arm64', None]),
'build-by': ('Builder identifier (e.g. jrandomhacker@example.net)', None),
'debian-mirror': ('Debian repository mirror', None),
'debian-security-mirror': ('Debian security updates mirror', None),
'pbuilder-debian-mirror': ('Debian repository mirror for pbuilder env bootstrap', None),
'vyos-mirror': ('VyOS package mirror', None),
'build-type': ('Build type, release or development', lambda x: x in ['release', 'development']),
'version': ('Version number (release builds only)', None),
'build-comment': ('Optional build comment', None)
}

# Create the option parser
parser = argparse.ArgumentParser()
for k, v in options.items():
help_string, default_value_thunk = v[0], v[1]
if default_value_thunk is None:
parser.add_argument('--' + k, type=str, help=help_string)
else:
parser.add_argument('--' + k, type=str, help=help_string, default=default_value_thunk())
help_string = v[0]
parser.add_argument('--' + k, type=str, help=help_string)

# Debug options
parser.add_argument('--debug', help='Enable debug output', action='store_true')
parser.add_argument('--dry-run', help='Check build configuration and exit', action='store_true')

# Custom APT entry and APT key options can be used multiple times
parser.add_argument('--custom-apt-entry', help="Custom APT entry", action='append', default=[])
parser.add_argument('--custom-apt-key', help="Custom APT key file", action='append', default=[])
parser.add_argument('--custom-package', help="Custom package to install from repositories", action='append', default=[])
parser.add_argument('--custom-apt-entry', help="Custom APT entry", action='append')
parser.add_argument('--custom-apt-key', help="Custom APT key file", action='append')
parser.add_argument('--custom-package', help="Custom package to install from repositories", action='append')

# Options relevant for non-ISO format flavors
parser.add_argument('--reuse-iso', help='Use an existing ISO file to build additional image formats', type=str, action='store', default=None)
parser.add_argument('--disk-size', help='Disk size for non-ISO image formats', type=int, action='store', default=10)
parser.add_argument('--reuse-iso', help='Use an existing ISO file to build additional image formats', type=str, action='store')
parser.add_argument('--disk-size', help='Disk size for non-ISO image formats', type=int, action='store')

# Build flavor is a positional argument
parser.add_argument('build_flavor', help='Build flavor', nargs='?', action='store')
Expand Down Expand Up @@ -295,8 +287,11 @@ if __name__ == "__main__":
args['build_dir'] = defaults.BUILD_DIR
args['pbuilder_config'] = os.path.join(defaults.BUILD_DIR, defaults.PBUILDER_CONFIG)

## Add hardcoded defaults
build_config = merge_defaults(defaults.HARDCODED_BUILD, defaults={})

## Combine the arguments with non-configurable defaults
build_config = copy.deepcopy(build_defaults)
build_config = merge_defaults(build_defaults, defaults=build_config)

## Load correct mix-ins
with open(make_toml_path(defaults.BUILD_TYPES_DIR, pre_build_config["build_type"]), 'rb') as f:
Expand Down
16 changes: 16 additions & 0 deletions scripts/image-build/defaults.py
Original file line number Diff line number Diff line change
Expand Up @@ -17,6 +17,11 @@


import os
import getpass
import platform

def get_default_build_by():
return "{user}@{host}".format(user= getpass.getuser(), host=platform.node())

# Default boot settings
boot_settings: dict[str, str] = {
Expand All @@ -27,6 +32,17 @@
'bootmode': 'normal'
}

# Hardcoded default values
HARDCODED_BUILD = {
'custom_apt_entry': [],
'custom_apt_key': [],
'custom_package': [],
'reuse_iso': None,
'disk_size': 10,
'build_by': get_default_build_by(),
'build_comment': '',
}

# Relative to the repository directory

BUILD_DIR = 'build'
Expand Down

0 comments on commit 31304ab

Please sign in to comment.