switch_arc_localbuild.py

#!/usr/bin/env python3
# -*- 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.
"""Switcher for Android container localbuild for ChromeOS bisecting."""

from __future__ import print_function
import argparse
import logging

from bisect_kit import android_util
from bisect_kit import arc_util
from bisect_kit import cli
from bisect_kit import codechange
from bisect_kit import common
from bisect_kit import configure
from bisect_kit import errors
from bisect_kit import locking
from bisect_kit import repo_util

logger = logging.getLogger(__name__)


def create_argument_parser():
  parents = [common.common_argument_parser, common.session_optional_parser]
  parser = argparse.ArgumentParser(parents=parents)
  cli.patching_argparser_exit(parser)
  parser.add_argument(
      '--dut',
      type=cli.argtype_notempty,
      metavar='DUT',
      default=configure.get('DUT'),
      help='DUT address')
  parser.add_argument(
      'rev',
      nargs='?',
      type=cli.argtype_notempty,
      metavar='INTRA_REV',
      default=configure.get('INTRA_REV', ''),
      help='Android build id, or bisecting intra version number')
  parser.add_argument(
      '--android_root',
      type=cli.argtype_dir_path,
      metavar='ANDROID_ROOT',
      default=configure.get('ANDROID_ROOT', ''),
      help='Android tree root')
  parser.add_argument(
      '--android_mirror',
      type=cli.argtype_dir_path,
      default=configure.get('ANDROID_MIRROR', ''),
      help='Android repo mirror path')
  parser.add_argument(
      '--flavor',
      metavar='ANDROID_FLAVOR',
      default=configure.get('ANDROID_FLAVOR'),
      help='example: cheets_x86-user')
  parser.add_argument(
      '--nobuild',
      action='store_true',
      help='Sync source code only; do not build; imply --nodeploy')
  parser.add_argument(
      '--nodeploy', action='store_true', help='Do not deploy after build')
  return parser


def switch(opts):
  config = dict(
      android_root=opts.android_root,
      flavor=opts.flavor,
      android_mirror=opts.android_mirror)
  spec_manager = android_util.AndroidSpecManager(config)
  cache = repo_util.RepoMirror(opts.android_mirror)
  code_manager = codechange.CodeManager(opts.android_root, spec_manager, cache)

  logger.info('switch source code')
  code_manager.switch(opts.rev)

  if opts.nobuild:
    return

  logger.info('build')
  # TODO(kcwu): use lower job count for non distributed compilation.
  with locking.lock_file(locking.LOCK_FILE_FOR_BUILD):
    android_util.lunch(opts.android_root, opts.flavor, 'm', '-j1000')

  if not opts.nodeploy:
    logger.info('deploy')
    arc_util.push_localbuild_to_device(opts.android_root, opts.dut, opts.flavor)


@cli.fatal_error_handler
def main(args=None):
  common.init()
  parser = create_argument_parser()
  opts = parser.parse_args(args)
  common.config_logging(opts)

  if not opts.dut:
    if not opts.nodeploy:
      raise errors.ArgumentError('--dut',
                                 'DUT can be omitted only if --nodeploy')
    if not opts.flavor:
      raise errors.ArgumentError('--flavor',
                                 'flavor must be specified if no --dut')

  if not opts.flavor:
    opts.flavor = arc_util.query_flavor(opts.dut)
    assert opts.flavor
  logger.info('use flavor=%s', opts.flavor)

  switch(opts)


if __name__ == '__main__':
  main()