Chromium Code Reviews
chromiumcodereview-hr@appspot.gserviceaccount.com (chromiumcodereview-hr) | Please choose your nickname with Settings | Help | Chromium Project | Gerrit Changes | Sign out
(12)

Side by Side Diff: infra/bots/assets/assets.py

Issue 2326663002: Fix issues in assets scripts (Closed)
Patch Set: Fix gsutil delete Created 4 years, 3 months ago
Use n/p to move between diff chunks; N/P to move between comments. Draft comments are only viewable by you.
Jump to:
View unified diff | Download patch
« no previous file with comments | « infra/bots/assets/asset_utils.py ('k') | no next file » | no next file with comments »
Toggle Intra-line Diffs ('i') | Expand Comments ('e') | Collapse Comments ('c') | Show Comments Hide Comments ('s')
OLDNEW
1 #!/usr/bin/env python 1 #!/usr/bin/env python
2 # 2 #
3 # Copyright 2016 Google Inc. 3 # Copyright 2016 Google Inc.
4 # 4 #
5 # Use of this source code is governed by a BSD-style license that can be 5 # Use of this source code is governed by a BSD-style license that can be
6 # found in the LICENSE file. 6 # found in the LICENSE file.
7 7
8 8
9 """Tool for managing assets.""" 9 """Tool for managing assets."""
10 10
11 11
12 import argparse 12 import argparse
13 import asset_utils 13 import asset_utils
14 import os 14 import os
15 import shutil 15 import shutil
16 import subprocess 16 import subprocess
17 import sys 17 import sys
18 18
19 FILE_DIR = os.path.dirname(os.path.abspath(__file__)) 19 FILE_DIR = os.path.dirname(os.path.abspath(__file__))
20 INFRA_BOTS_DIR = os.path.realpath(os.path.join(FILE_DIR, os.pardir)) 20 INFRA_BOTS_DIR = os.path.realpath(os.path.join(FILE_DIR, os.pardir))
21 21
22 sys.path.insert(0, INFRA_BOTS_DIR) 22 sys.path.insert(0, INFRA_BOTS_DIR)
23 import utils 23 import utils
24 24
25 25
26 def add(args): 26 def add(args):
27 """Add a new asset.""" 27 """Add a new asset."""
28 asset_utils.Asset.add(args.asset_name, asset_utils.MultiStore()) 28 asset_utils.Asset.add(args.asset_name,
29 asset_utils.MultiStore(gsutil=args.gsutil))
29 30
30 31
31 def remove(args): 32 def remove(args):
32 """Remove an asset.""" 33 """Remove an asset."""
33 asset_utils.Asset(args.asset_name, asset_utils.MultiStore()).remove() 34 asset_utils.Asset(args.asset_name,
35 asset_utils.MultiStore(gsutil=args.gsutil)).remove()
34 36
35 37
36 def download(args): 38 def download(args):
37 """Download the current version of an asset.""" 39 """Download the current version of an asset."""
38 asset = asset_utils.Asset(args.asset_name, asset_utils.MultiStore()) 40 asset = asset_utils.Asset(args.asset_name,
41 asset_utils.MultiStore(gsutil=args.gsutil))
39 asset.download_current_version(args.target_dir) 42 asset.download_current_version(args.target_dir)
40 43
41 44
42 def upload(args): 45 def upload(args):
43 """Upload a new version of the asset.""" 46 """Upload a new version of the asset."""
44 asset = asset_utils.Asset(args.asset_name, asset_utils.MultiStore()) 47 asset = asset_utils.Asset(args.asset_name,
48 asset_utils.MultiStore(gsutil=args.gsutil))
45 asset.upload_new_version(args.target_dir, commit=args.commit) 49 asset.upload_new_version(args.target_dir, commit=args.commit)
46 50
47 51
48 def main(argv): 52 def main(argv):
49 parser = argparse.ArgumentParser(description='Tool for managing assets.') 53 parser = argparse.ArgumentParser(description='Tool for managing assets.')
50 subs = parser.add_subparsers(help='Commands:') 54 subs = parser.add_subparsers(help='Commands:')
51 55
52 prs_add = subs.add_parser('add', help='Add a new asset.') 56 prs_add = subs.add_parser('add', help='Add a new asset.')
53 prs_add.set_defaults(func=add) 57 prs_add.set_defaults(func=add)
54 prs_add.add_argument('asset_name', help='Name of the asset.') 58 prs_add.add_argument('asset_name', help='Name of the asset.')
(...skipping 16 matching lines...) Expand all
71 prs_upload.add_argument('--target_dir', '-t', required=True) 75 prs_upload.add_argument('--target_dir', '-t', required=True)
72 prs_upload.add_argument('--gsutil') 76 prs_upload.add_argument('--gsutil')
73 prs_upload.add_argument('--commit', action='store_true') 77 prs_upload.add_argument('--commit', action='store_true')
74 78
75 args = parser.parse_args(argv) 79 args = parser.parse_args(argv)
76 args.func(args) 80 args.func(args)
77 81
78 82
79 if __name__ == '__main__': 83 if __name__ == '__main__':
80 main(sys.argv[1:]) 84 main(sys.argv[1:])
OLDNEW
« no previous file with comments | « infra/bots/assets/asset_utils.py ('k') | no next file » | no next file with comments »

Powered by Google App Engine
This is Rietveld 408576698