| # -*- coding: utf-8 -*- |
| # Copyright 2020 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. |
| |
| """Utility script to update the generated config files.""" |
| |
| from __future__ import print_function |
| |
| import os |
| import sys |
| |
| from chromite.config import chromeos_config |
| from chromite.lib import commandline |
| from chromite.lib import constants |
| from chromite.lib import cros_build_lib |
| from chromite.lib import cros_logging as logging |
| from chromite.lib import osutils |
| |
| assert sys.version_info >= (3, 6), 'This module requires Python 3.6+' |
| |
| |
| def main(argv): |
| # Parse arguments to respect log levels. |
| commandline.ArgumentParser().parse_args(argv) |
| |
| # Regenerate `config_dump.json`. |
| logging.info('Regenerating config_dump.json') |
| site_config = chromeos_config.GetConfig() |
| site_config.SaveConfigToFile(constants.CHROMEOS_CONFIG_FILE) |
| |
| # Regenerate `waterfall_layout_dump.txt`. |
| logging.info('Regenerating waterfall_layout_dump.txt') |
| cmd = os.path.join(constants.CHROMITE_BIN_DIR, 'cros_show_waterfall_layout') |
| result = cros_build_lib.run([cmd], |
| capture_output=True, |
| encoding='utf-8', |
| debug_level=logging.DEBUG) |
| osutils.WriteFile(constants.WATERFALL_CONFIG_FILE, result.stdout) |
| |
| # Regenerate `luci-scheduler.cfg`. |
| logging.info('Regenerating luci-scheduler.cfg') |
| cmd = os.path.join(constants.CHROMITE_SCRIPTS_DIR, 'gen_luci_scheduler') |
| luci_result = cros_build_lib.run([cmd], |
| capture_output=True, |
| encoding='utf-8', |
| debug_level=logging.DEBUG) |
| osutils.WriteFile(constants.LUCI_SCHEDULER_CONFIG_FILE, luci_result.stdout) |