| #!/usr/bin/python |
| |
| """Upload a local build to Google Compute Engine and run it.""" |
| |
| import argparse |
| import glob |
| import os |
| import subprocess |
| |
| |
| def upload_artifacts(args): |
| image_pat = os.path.join(args.dist_dir, '%s-img-*.zip' % args.product) |
| images = glob.glob(image_pat) |
| if len(images) == 0: |
| raise OSError('File not found: %s' + image_pat) |
| if len(images) > 1: |
| raise OSError('%s matches multiple images: %s' % ( |
| image_pat, images)) |
| subprocess.check_call( |
| 'gcloud compute ssh %s@%s -- /usr/bin/install_zip.sh . < %s' % ( |
| args.user, |
| args.instance, |
| images[0]), |
| shell=True) |
| |
| host_package = os.path.join(args.dist_dir, 'cvd-host_package.tar.gz') |
| # host_package |
| subprocess.check_call( |
| 'gcloud compute ssh %s@%s -- tar -x -z -f - < %s' % ( |
| args.user, |
| args.instance, |
| host_package), |
| shell=True) |
| |
| |
| def launch_cvd(args): |
| subprocess.check_call( |
| 'gcloud compute ssh %s@%s -- bin/launch_cvd' % ( |
| args.user, |
| args.instance), |
| shell=True) |
| |
| |
| def stop_cvd(args): |
| subprocess.call( |
| 'gcloud compute ssh %s@%s -- bin/stop_cvd' % ( |
| args.user, |
| args.instance), |
| shell=True) |
| |
| |
| def main(): |
| parser = argparse.ArgumentParser( |
| description='Fetch images from a compute engine bucket') |
| parser.add_argument( |
| '-dist_dir', |
| type=str, |
| default=os.path.join(os.environ.get('ANDROID_BUILD_TOP', '.'), |
| 'out', 'dist'), |
| help='path to the dist directory') |
| parser.add_argument( |
| '-instance', type=str, required=True, |
| help='instance to update') |
| parser.add_argument( |
| '-product', |
| type=str, |
| default=os.environ.get('TARGET_PRODUCT', 'cf_x86_phone'), |
| help='product to upload') |
| parser.add_argument( |
| '-user', type=str, default='vsoc-01', |
| help='user to update on the instance') |
| args = parser.parse_args() |
| stop_cvd(args) |
| upload_artifacts(args) |
| launch_cvd(args) |
| |
| |
| if __name__ == '__main__': |
| main() |
