Home | History | Annotate | Download | only in bloaty 
      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