| Index: infra/bots/assets/skp/create.py
|
| diff --git a/infra/bots/assets/skp/create.py b/infra/bots/assets/skp/create.py
|
| index 68df7f07c27762d143bd4f50c69d010cdbde7e9d..6bf716c532b05d88716d7bc340c8f9996add7179 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__':
|
|
|