1#!/usr/bin/env python 2# 3# Copyright 2019 Google Inc. 4# 5# Use of this source code is governed by a BSD-style license that can be 6# found in the LICENSE file. 7 8 9"""Create the asset and upload it.""" 10 11 12import argparse 13import common 14import os 15import subprocess 16import sys 17import utils 18 19 20def main(): 21 parser = argparse.ArgumentParser() 22 parser.add_argument('--gsutil') 23 args = parser.parse_args() 24 25 with utils.tmp_dir(): 26 cwd = os.getcwd() 27 create_script = os.path.join(common.FILE_DIR, 'create.py') 28 upload_script = os.path.join(common.FILE_DIR, 'upload.py') 29 30 try: 31 subprocess.check_call(['python', create_script, '-t', cwd]) 32 cmd = ['python', upload_script, '-t', cwd] 33 if args.gsutil: 34 cmd.extend(['--gsutil', args.gsutil]) 35 subprocess.check_call(cmd) 36 except subprocess.CalledProcessError: 37 # Trap exceptions to avoid printing two stacktraces. 38 sys.exit(1) 39 40 41if __name__ == '__main__': 42 main() 43