Index: download_from_google_storage.py |
diff --git a/download_from_google_storage.py b/download_from_google_storage.py |
index 999ef722a879de0e0315bb5b6bad83b4c2de975f..f73b3bd0af547d1e808d52a30b553e1ccf5af47d 100755 |
--- a/download_from_google_storage.py |
+++ b/download_from_google_storage.py |
@@ -20,8 +20,7 @@ import subprocess2 |
GSUTIL_DEFAULT_PATH = os.path.join( |
- os.path.dirname(os.path.abspath(__file__)), |
- 'third_party', 'gsutil', 'gsutil') |
+ os.path.dirname(os.path.abspath(__file__)), 'gsutil.py') |
# Maps sys.platform to what we actually want to call them. |
PLATFORM_MAPPING = { |
'cygwin': 'win', |
@@ -55,13 +54,13 @@ def GetNormalizedPlatform(): |
class Gsutil(object): |
"""Call gsutil with some predefined settings. This is a convenience object, |
and is also immutable.""" |
- def __init__(self, path, boto_path, timeout=None, bypass_prodaccess=False): |
+ def __init__(self, path, boto_path, timeout=None, version='4.7'): |
if not os.path.exists(path): |
raise FileNotFoundError('GSUtil not found in %s' % path) |
self.path = path |
self.timeout = timeout |
self.boto_path = boto_path |
- self.bypass_prodaccess = bypass_prodaccess |
+ self.version = version |
def get_sub_env(self): |
env = os.environ.copy() |
@@ -80,16 +79,12 @@ class Gsutil(object): |
return env |
def call(self, *args): |
- cmd = [sys.executable, self.path] |
- if self.bypass_prodaccess: |
- cmd.append('--bypass_prodaccess') |
+ cmd = [sys.executable, self.path, '--force-version', self.version] |
cmd.extend(args) |
return subprocess2.call(cmd, env=self.get_sub_env(), timeout=self.timeout) |
def check_call(self, *args): |
- cmd = [sys.executable, self.path] |
- if self.bypass_prodaccess: |
- cmd.append('--bypass_prodaccess') |
+ cmd = [sys.executable, self.path, '--force-version', self.version] |
cmd.extend(args) |
((out, err), code) = subprocess2.communicate( |
cmd, |
@@ -237,7 +232,7 @@ def _downloader_worker_thread(thread_num, q, force, base_url, |
if os.path.exists(output_filename): |
out_q.put('%d> Warning: deleting %s failed.' % ( |
thread_num, output_filename)) |
- code, _, err = gsutil.check_call('cp', '-q', file_url, output_filename) |
+ code, _, err = gsutil.check_call('cp', file_url, output_filename) |
if code != 0: |
out_q.put('%d> %s' % (thread_num, err)) |
ret_codes.put((code, err)) |
@@ -398,8 +393,7 @@ def main(args): |
# Make sure gsutil exists where we expect it to. |
if os.path.exists(GSUTIL_DEFAULT_PATH): |
gsutil = Gsutil(GSUTIL_DEFAULT_PATH, |
- boto_path=options.boto, |
- bypass_prodaccess=options.no_auth) |
+ boto_path=options.boto) |
else: |
parser.error('gsutil not found in %s, bad depot_tools checkout?' % |
GSUTIL_DEFAULT_PATH) |