def Run(args):
"""Run reconnect.
Args:
args: Namespace object from argparse.parse_args.
"""
cfg = config.GetAcloudConfig(args)
instances_to_reconnect = []
if args.instance_names is not None:
# user input instance name to get instance object.
instances_to_reconnect = list_instance.GetInstancesFromInstanceNames(
cfg, args.instance_names)
if not instances_to_reconnect:
instances_to_reconnect = list_instance.ChooseInstances(cfg, args.all)
reconnect_report = report.Report(command="reconnect")
for instance in instances_to_reconnect:
if instance.avd_type not in utils.AVD_PORT_DICT:
utils.PrintColorString(
"Skipping reconnect of instance %s due to "
"unknown avd type (%s)." % (instance.name, instance.avd_type),
utils.TextColors.WARNING)
continue
if not instance.islocal:
AddPublicSshRsaToInstance(cfg, constants.GCE_USER, instance.name)
ReconnectInstance(cfg.ssh_private_key_path,
instance,
reconnect_report,
cfg.extra_args_ssh_tunnel,
connect_vnc=(args.autoconnect is True))
utils.PrintDeviceSummary(reconnect_report)
def Run(args):
"""Run delete.
After delete command executed, tool will return one Report instance.
If there is no instance to delete, just reutrn empty Report.
Args:
args: Namespace object from argparse.parse_args.
Returns:
A Report instance.
"""
# Prioritize delete instances by names without query all instance info from
# GCP project.
cfg = config.GetAcloudConfig(args)
if args.instance_names:
return DeleteInstanceByNames(cfg,
args.instance_names)
if args.remote_host:
return CleanUpRemoteHost(cfg, args.remote_host, args.host_user,
args.host_ssh_private_key_path)
instances = list_instances.GetLocalInstances()
if not args.local_only and cfg.SupportRemoteInstance():
instances.extend(list_instances.GetRemoteInstances(cfg))
if args.adb_port:
instances = list_instances.FilterInstancesByAdbPort(instances,
args.adb_port)
elif not args.all:
# Provide instances list to user and let user choose what to delete if
# user didn't specify instances in args.
instances = list_instances.ChooseInstancesFromList(instances)
return DeleteInstances(cfg, instances)
def __init__(self, args):
"""Process the args into class vars.
Args:
args: Namespace object from argparse.parse_args.
"""
# Let's define the private class vars here and then process the user
# args afterwards.
self._client_adb_port = args.adb_port
self._autoconnect = None
self._instance_name_to_reuse = None
self._unlock_screen = None
self._report_internal_ip = None
self._avd_type = None
self._flavor = None
self._image_source = None
self._instance_type = None
self._local_image_dir = None
self._local_image_artifact = None
self._local_system_image_dir = None
self._local_tool_dirs = None
self._image_download_dir = None
self._num_of_instances = None
self._num_avds_per_instance = None
self._no_pull_log = None
self._remote_image = None
self._system_build_info = None
self._kernel_build_info = None
self._hw_property = None
self._remote_host = None
self._host_user = None
self._host_ssh_private_key_path = None
# Create config instance for android_build_client to query build api.
self._cfg = config.GetAcloudConfig(args)
# Reporting args.
self._serial_log_file = None
# gpu and emulator_build_id is only used for goldfish avd_type.
self._gpu = None
self._emulator_build_id = None
# Fields only used for cheeps type.
self._stable_cheeps_host_image_name = None
self._stable_cheeps_host_image_project = None
self._username = None
self._password = None
# The maximum time in seconds used to wait for the AVD to boot.
self._boot_timeout_secs = None
# The maximum time in seconds used to wait for the instance ready.
self._ins_timeout_secs = None
# The local instance id
self._local_instance_id = None
self._ProcessArgs(args)
def Run(args):
"""Run list.
Args:
args: Namespace object from argparse.parse_args.
"""
instances = GetLocalInstances()
cfg = config.GetAcloudConfig(args)
if not args.local_only and cfg.SupportRemoteInstance():
instances.extend(GetRemoteInstances(cfg))
PrintInstancesDetails(instances, args.verbose)
def Run(args):
"""Run powerwash.
After powerwash command executed, tool will return one Report instance.
Args:
args: Namespace object from argparse.parse_args.
Returns:
A Report instance.
"""
cfg = config.GetAcloudConfig(args)
if args.instance_name:
instance = list_instances.GetInstancesFromInstanceNames(
cfg, [args.instance_name])
return PowerwashFromInstance(cfg, instance[0], args.instance_id)
return PowerwashFromInstance(cfg,
list_instances.ChooseOneRemoteInstance(cfg),
args.instance_id)
def Run(args):
"""Run pull.
After pull command executed, tool will return one Report instance.
If there is no instance to pull, just return empty Report.
Args:
args: Namespace object from argparse.parse_args.
Returns:
A Report instance.
"""
cfg = config.GetAcloudConfig(args)
if args.instance_name:
instance = list_instances.GetInstancesFromInstanceNames(
cfg, [args.instance_name])
return PullFileFromInstance(cfg, instance[0], args.file_name,
args.no_prompt)
return PullFileFromInstance(cfg,
list_instances.ChooseOneRemoteInstance(cfg),
args.file_name, args.no_prompt)
def main(argv=None):
"""Main entry.
Args:
argv: A list of system arguments.
Returns:
Job status: Integer, 0 if success. None-zero if fails.
Stack trace: String of errors.
"""
args = _ParseArgs(argv)
_SetupLogging(args.log_file, args.verbose)
_VerifyArgs(args)
_LOGGER.info("Acloud version: %s", config.GetVersion())
cfg = config.GetAcloudConfig(args)
# TODO: Move this check into the functions it is actually needed.
# Check access.
# device_driver.CheckAccess(cfg)
report = None
if args.which == create_args.CMD_CREATE:
report = create.Run(args)
elif args.which == CMD_CREATE_CUTTLEFISH:
report = create_cuttlefish_action.CreateDevices(
cfg=cfg,
build_target=args.build_target,
build_id=args.build_id,
branch=args.branch,
kernel_build_id=args.kernel_build_id,
kernel_branch=args.kernel_branch,
kernel_build_target=args.kernel_build_target,
system_branch=args.system_branch,
system_build_id=args.system_build_id,
system_build_target=args.system_build_target,
gpu=args.gpu,
num=args.num,
serial_log_file=args.serial_log_file,
autoconnect=args.autoconnect,
report_internal_ip=args.report_internal_ip,
boot_timeout_secs=args.boot_timeout_secs,
ins_timeout_secs=args.ins_timeout_secs)
elif args.which == CMD_CREATE_GOLDFISH:
report = create_goldfish_action.CreateDevices(
cfg=cfg,
build_target=args.build_target,
build_id=args.build_id,
emulator_build_id=args.emulator_build_id,
branch=args.branch,
emulator_branch=args.emulator_branch,
kernel_build_id=args.kernel_build_id,
kernel_branch=args.kernel_branch,
kernel_build_target=args.kernel_build_target,
gpu=args.gpu,
num=args.num,
serial_log_file=args.serial_log_file,
autoconnect=args.autoconnect,
tags=args.tags,
report_internal_ip=args.report_internal_ip,
boot_timeout_secs=args.boot_timeout_secs)
elif args.which == delete_args.CMD_DELETE:
report = delete.Run(args)
elif args.which == list_args.CMD_LIST:
list_instances.Run(args)
elif args.which == reconnect_args.CMD_RECONNECT:
reconnect.Run(args)
elif args.which == powerwash_args.CMD_POWERWASH:
report = powerwash.Run(args)
elif args.which == pull_args.CMD_PULL:
report = pull.Run(args)
elif args.which == setup_args.CMD_SETUP:
setup.Run(args)
else:
error_msg = "Invalid command %s" % args.which
sys.stderr.write(error_msg)
return constants.EXIT_BY_WRONG_CMD, error_msg
if report and args.report_file:
report.Dump(args.report_file)
if report and report.errors:
error_msg = "\n".join(report.errors)
help_msg = _CONTACT_INFO
if report.data.get(constants.ERROR_LOG_FOLDER):
help_msg += _LOG_INFO % report.data.get(constants.ERROR_LOG_FOLDER)
sys.stderr.write("Encountered the following errors:\n%s\n\n%s.\n" %
(error_msg, help_msg))
return constants.EXIT_BY_FAIL_REPORT, error_msg
return constants.EXIT_SUCCESS, NO_ERROR_MESSAGE