Formatting: Format all python code with black.
This CL is probably not what you're looking for, it's only
automated formatting. Ignore it with
`git blame --ignore-rev <revision>` for this commit.
BUG=b:233893248
TEST=CQ
Change-Id: I66591d7a738d241aed3290138c0f68065ab10a6d
Reviewed-on: https://chromium-review.googlesource.com/c/chromiumos/chromite/+/3879174
Reviewed-by: Mike Frysinger <vapier@chromium.org>
Tested-by: Alex Klein <saklein@chromium.org>
diff --git a/scripts/cros_generate_quick_provision_payload.py b/scripts/cros_generate_quick_provision_payload.py
index 0d535f7..212e8e2 100644
--- a/scripts/cros_generate_quick_provision_payload.py
+++ b/scripts/cros_generate_quick_provision_payload.py
@@ -16,59 +16,71 @@
def ParseArguments(argv):
- """Returns a namespace for the CLI arguments."""
- parser = commandline.ArgumentParser(description=__doc__)
- parser.add_argument('--image', type='path', required=True,
- help='The path to local image to build the quick '
- 'provision payloads for.')
- parser.add_argument('--output', type='path',
- help='The output directory to generate quick '
- 'provision payloads for.', default='.')
+ """Returns a namespace for the CLI arguments."""
+ parser = commandline.ArgumentParser(description=__doc__)
+ parser.add_argument(
+ "--image",
+ type="path",
+ required=True,
+ help="The path to local image to build the quick "
+ "provision payloads for.",
+ )
+ parser.add_argument(
+ "--output",
+ type="path",
+ help="The output directory to generate quick "
+ "provision payloads for.",
+ default=".",
+ )
- opts = parser.parse_args(argv)
- # Check if output is valid directory.
- if not os.path.isdir(opts.output):
- parser.error('Please pass in a valid output directory.')
+ opts = parser.parse_args(argv)
+ # Check if output is valid directory.
+ if not os.path.isdir(opts.output):
+ parser.error("Please pass in a valid output directory.")
- opts.Freeze()
+ opts.Freeze()
- return opts
+ return opts
def CreateKernelQuickProvisionPayload(image, output):
- with osutils.TempDir() as temp_dir:
- # Extract kernel.
- kern = os.path.join(temp_dir, 'kern')
- partition_lib.ExtractKernel(image, os.path.join(temp_dir, kern))
- # Compress kernel.
- cros_build_lib.CompressFile(
- kern,
- os.path.join(output, constants.QUICK_PROVISION_PAYLOAD_KERNEL))
+ with osutils.TempDir() as temp_dir:
+ # Extract kernel.
+ kern = os.path.join(temp_dir, "kern")
+ partition_lib.ExtractKernel(image, os.path.join(temp_dir, kern))
+ # Compress kernel.
+ cros_build_lib.CompressFile(
+ kern, os.path.join(output, constants.QUICK_PROVISION_PAYLOAD_KERNEL)
+ )
def CreateRootQuickProvisionPayload(image, output):
- with osutils.TempDir() as temp_dir:
- # Extract root.
- root = os.path.join(temp_dir, 'root')
- partition_lib.ExtractRoot(image, os.path.join(temp_dir, root))
- # Compress root.
- cros_build_lib.CompressFile(
- root,
- os.path.join(output, constants.QUICK_PROVISION_PAYLOAD_ROOTFS))
+ with osutils.TempDir() as temp_dir:
+ # Extract root.
+ root = os.path.join(temp_dir, "root")
+ partition_lib.ExtractRoot(image, os.path.join(temp_dir, root))
+ # Compress root.
+ cros_build_lib.CompressFile(
+ root, os.path.join(output, constants.QUICK_PROVISION_PAYLOAD_ROOTFS)
+ )
def CreateStatefulQuickProvisionPayload(image, output):
- # Create stateful quick provision payload.
- paygen_stateful_payload_lib.GenerateStatefulPayload(image, output)
- # Change output ownership of file.
+ # Create stateful quick provision payload.
+ paygen_stateful_payload_lib.GenerateStatefulPayload(image, output)
+ # Change output ownership of file.
def main(argv):
- opts = ParseArguments(argv)
+ opts = ParseArguments(argv)
- parallel.RunParallelSteps([
- # Stateful generation is usually the slowest.
- lambda: CreateStatefulQuickProvisionPayload(opts.image, opts.output),
- lambda: CreateKernelQuickProvisionPayload(opts.image, opts.output),
- lambda: CreateRootQuickProvisionPayload(opts.image, opts.output),
- ])
+ parallel.RunParallelSteps(
+ [
+ # Stateful generation is usually the slowest.
+ lambda: CreateStatefulQuickProvisionPayload(
+ opts.image, opts.output
+ ),
+ lambda: CreateKernelQuickProvisionPayload(opts.image, opts.output),
+ lambda: CreateRootQuickProvisionPayload(opts.image, opts.output),
+ ]
+ )