| # -*- coding: utf-8 -*- |
| # Copyright 2018 The Chromium OS Authors. All rights reserved. |
| # Use of this source code is governed by a BSD-style license that can be |
| # found in the LICENSE file. |
| |
| """Update or initialize the chroot version. |
| |
| The chroot version hooks provide manually created update files to handle |
| tasks that need to be manually run. See also `update_chroot` for updating |
| the toolchain and some other packages on the chroot. The `update_chroot` script |
| also calls this script. The chroot version and installed package versions are |
| not strongly correlated. |
| """ |
| |
| from __future__ import print_function |
| |
| import sys |
| |
| from chromite.lib import commandline |
| from chromite.lib import cros_build_lib |
| from chromite.lib import cros_sdk_lib |
| |
| |
| assert sys.version_info >= (3, 6), 'This module requires Python 3.6+' |
| |
| |
| def GetParser(): |
| """Build the ArgumentParser.""" |
| parser = commandline.ArgumentParser(description=__doc__) |
| parser.add_argument('-i', '--init-latest', |
| action='store_true', default=False, |
| help='Create the version file with the latest version ' |
| "if it doesn't exist.") |
| |
| return parser |
| |
| |
| def _ParseArgs(argv): |
| """Parse arguments.""" |
| parser = GetParser() |
| |
| opts = parser.parse_args(argv) |
| opts.Freeze() |
| |
| return opts |
| |
| |
| def main(argv): |
| """Main function.""" |
| cros_build_lib.AssertInsideChroot() |
| |
| opts = _ParseArgs(argv) |
| |
| if opts.init_latest: |
| cros_sdk_lib.InitLatestVersion() |
| else: |
| try: |
| cros_sdk_lib.RunChrootVersionHooks() |
| except cros_sdk_lib.Error as e: |
| cros_build_lib.Die(e) |