b186b1f67a
Using click's "required" argument checks to see if the value is passed or not. Therefore we can check before flag before running a command. So we can remove duplicate code. Story: 2010867 Task: 48556 Change-Id: Ie07d8ff921c8b883d323f8bbc062ce1130039848 Signed-off-by: Charles Short <charles.short@windriver.com>
73 lines
2.3 KiB
Python
73 lines
2.3 KiB
Python
"""
|
|
Copyright (c) 2023 Wind River Systems, Inc.
|
|
|
|
SPDX-License-Identifier: Apache-2.0
|
|
|
|
"""
|
|
|
|
import shutil
|
|
import sys
|
|
|
|
import click
|
|
|
|
from apt_ostree.bootstrap import create_ostree
|
|
from apt_ostree.cmd.options import compose_options
|
|
from apt_ostree.cmd import pass_state_context
|
|
from apt_ostree.log import complete_step
|
|
from apt_ostree.log import log_step
|
|
from apt_ostree.ostree import ostree_commit
|
|
from apt_ostree.utils import run_command
|
|
|
|
|
|
@click.command(short_help="Create treefile")
|
|
@pass_state_context
|
|
@compose_options
|
|
def create(state, repo, base, branch, edit):
|
|
if not state.repo.exists():
|
|
click.secho(f"Creating ostree repository: {state.repo}")
|
|
run_command(["ostree", "init", f"--repo={state.repo}",
|
|
"--mode=archive-z2"])
|
|
else:
|
|
click.secho(f"Found ostree repository: {state.repo}")
|
|
|
|
click.secho(f"Found ostree branch: {state.branch}")
|
|
|
|
if not state.base.exists():
|
|
click.secho("Configuration directory does not exist.", fg="red")
|
|
sys.exit(1)
|
|
click.secho(f"Found configuration directory: {state.base}")
|
|
|
|
config = state.base.joinpath("bootstrap.yaml")
|
|
if not config.exists():
|
|
click.sceho("bootstrap.yaml does not exist.", fg="red")
|
|
sys.exit(1)
|
|
else:
|
|
click.secho("Found configuration file bootstrap.yaml.")
|
|
|
|
with complete_step(f"Setting up workspace for {branch}."):
|
|
workspace = state.workspace
|
|
workdir = workspace.joinpath(f"build/{branch}")
|
|
rootfs = workdir.joinpath("rootfs")
|
|
|
|
log_step(f"Building workspace for {branch} in {workspace}")
|
|
if workdir.exists():
|
|
log_step("Found working directory from previous run...removing.")
|
|
shutil.rmtree(workdir)
|
|
workdir.mkdir(parents=True, exist_ok=True)
|
|
rootfs.mkdir(parents=True, exist_ok=True)
|
|
|
|
log_step("Running bdebstrap, please wait.")
|
|
verbosity = "-q"
|
|
if state.debug:
|
|
verbosity = "-v"
|
|
run_command(
|
|
["bdebstrap", "-c", "bootstrap.yaml", verbosity,
|
|
"--force", "--name", str(state.branch), "--target", str(rootfs),
|
|
"--output", str(workdir)], cwd=state.base)
|
|
|
|
create_ostree(rootfs)
|
|
ostree_commit(state,
|
|
rootfs,
|
|
subject="Commit by apt-ostree",
|
|
msg="Initialized by apt-ostree")
|