1 #!/usr/bin/env python 2 # 3 # Copyright 2017 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 12 import argparse 13 import common 14 import os 15 import subprocess 16 import sys 17 import utils 18 19 20 def 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 41 if __name__ == '__main__': 42 main() 43