Change SKPs to new asset management format

BUG=skia:5427
GOLD_TRYBOT_URL= https://gold.skia.org/search?issue=2169653003
DOCS_PREVIEW= https://skia.org/?cl=2169653003

Review-Url: https://codereview.chromium.org/2169653003

1 files changed, 7 insertions, 2 deletions

diff --git a/infra/bots/assets/skp/create.py b/infra/bots/assets/skp/create.py
index 68df7f07c2..6bf716c532 100755
--- a/infra/bots/assets/skp/create.py
+++ b/infra/bots/assets/skp/create.py
@@ -20,7 +20,8 @@ import utils
 SKIA_TOOLS = os.path.join(common.INFRA_BOTS_DIR, os.pardir, os.pardir, 'tools')
 
 
-def create_asset(chrome_src_path, browser_executable, target_dir):
+def create_asset(chrome_src_path, browser_executable, target_dir,
+                 upload_to_partner_bucket):
   """Create the asset."""
   browser_executable = os.path.realpath(browser_executable)
   chrome_src_path = os.path.realpath(chrome_src_path)
@@ -47,6 +48,8 @@ def create_asset(chrome_src_path, browser_executable, target_dir):
       '--output_dir', os.getcwd(),
       '--chrome_src_path', chrome_src_path,
     ]
+    if upload_to_partner_bucket:
+      webpages_playback_cmd.append('--upload_to_partner_bucket')
     try:
       subprocess.check_call(webpages_playback_cmd)
     finally:
@@ -75,8 +78,10 @@ def main():
   parser.add_argument('--target_dir', '-t', required=True)
   parser.add_argument('--chrome_src_path', '-c', required=True)
   parser.add_argument('--browser_executable', '-e', required=True)
+  parser.add_argument('--upload_to_partner_bucket', action='store_true')
   args = parser.parse_args()
-  create_asset(args.chrome_src_path, args.browser_executable, args.target_dir)
+  create_asset(args.chrome_src_path, args.browser_executable, args.target_dir,
+               args.upload_to_partner_bucket)
 
 
 if __name__ == '__main__':