scripts/cros_generate_stateful_update_payload.py

# Copyright 2018 The ChromiumOS Authors
# Use of this source code is governed by a BSD-style license that can be
# found in the LICENSE file.

"""This module is responsible for generating a stateful update payload."""

from chromite.lib import commandline
from chromite.lib.paygen import paygen_stateful_payload_lib


def ParseArguments(argv):
    """Returns a namespace for the CLI arguments."""
    parser = commandline.ArgumentParser(description=__doc__)
    parser.add_argument(
        "-i",
        "--image_path",
        type="str_path",
        required=True,
        help="The image to generate the stateful update for.",
    )
    parser.add_argument(
        "-o",
        "--output_dir",
        type="str_path",
        required=True,
        help="The path to the directory to output the stateful" "update file.",
    )
    opts = parser.parse_args(argv)
    opts.Freeze()

    return opts


def main(argv) -> None:
    opts = ParseArguments(argv)

    paygen_stateful_payload_lib.GenerateStatefulPayload(
        opts.image_path, opts.output_dir
    )

    paygen_stateful_payload_lib.GenerateZstdStatefulPayload(
        opts.image_path, opts.output_dir
    )