Source code for benchbuild.cli.bootstrap

import os

from plumbum import cli

from benchbuild.utils import bootstrap
from benchbuild.cli.main import BenchBuild
from benchbuild import settings

CFG = settings.CFG


@BenchBuild.subcommand("bootstrap")
class BenchBuildBootstrap(cli.Application):
    """Bootstrap benchbuild external dependencies, if possible."""

    store_config = cli.Flag(
        ["-s", "--save-config"],
        help="Save benchbuild's configuration.",
        default=False)

    def main(self, *args):
        del args  # Unused

        print("Checking benchbuild binary dependencies...")
        bootstrap.provide_package("cmake")
        bootstrap.provide_package("fusermount")
        bootstrap.provide_package("unionfs")
        bootstrap.provide_package(
            'uchroot', installer=bootstrap.install_uchroot)
        bootstrap.provide_packages(CFG['bootstrap']['packages'].value)

        if self.store_config:
            config_path = ".benchbuild.yml"
            CFG.store(config_path)
            print("Storing config in {0}".format(os.path.abspath(config_path)))
            exit(0)