Add asset management scripts

infra/bots/assets/scripts/create_and_upload.py

+#!/usr/bin/env python
+#
+# Copyright 2016 Google Inc.
+#
+# Use of this source code is governed by a BSD-style license that can be
+# found in the LICENSE file.
+
+
+"""Create the asset and upload it."""

[rmistry, 2016/06/15 13:47:18]

[Optional]
Will need more maintenance work to keep this script around since you already
have create.py and upload.py
I personally would just use those two scripts but upto you.

[borenet, 2016/06/15 14:31:27]

IMO being able to create and upload the asset from a temporary directory which
gets auto-cleaned is a compelling reason to have this script.  Agreed that may
require some maintenance, but that really should just be limited to plumbing
through arguments to create.py.

+
+
+import argparse
+import os
+import subprocess
+import sys
+
+FILE_DIR = os.path.dirname(os.path.abspath(__file__))
+INFRA_BOTS_DIR = os.path.realpath(os.path.join(FILE_DIR, os.pardir, os.pardir))
+
+sys.path.insert(0, INFRA_BOTS_DIR)
+import utils
+
+
+def main():
+  parser = argparse.ArgumentParser()
+  parser.add_argument('--gsutil')
+  args = parser.parse_args()
+
+  with utils.tmp_dir():
+    cwd = os.getcwd()
+    create_script = os.path.join(FILE_DIR, 'create.py')
+    upload_script = os.path.join(FILE_DIR, 'upload.py')
+
+    try:
+      subprocess.check_call(['python', create_script, '-t', cwd])
+      subprocess.check_call(['python', upload_script, '-t', cwd,
+                             '--gsutil', args.gsutil])
+    except subprocess.CalledProcessError:
+      # Trap exceptions to avoid printing two stacktraces.
+      sys.exit(1)
+
+
+if __name__ == '__main__':
+  main()