OLD | NEW |
1 #!/usr/bin/env python | |
2 # Copyright 2015 The Chromium Authors. All rights reserved. | 1 # Copyright 2015 The Chromium Authors. All rights reserved. |
3 # Use of this source code is governed by a BSD-style license that can be | 2 # Use of this source code is governed by a BSD-style license that can be |
4 # found in the LICENSE file. | 3 # found in the LICENSE file. |
5 | 4 |
6 import os | 5 import os |
7 import subprocess | 6 import subprocess |
8 import sys | 7 import sys |
9 import find_depot_tools | |
10 | 8 |
11 def download_from_public_bucket(gs_path, output_path): | 9 def download_from_public_bucket(gs_path, output_path, depot_tools_path): |
12 depot_tools_path = find_depot_tools.add_depot_tools_to_path() | |
13 gsutil_exe = os.path.join(depot_tools_path, "third_party", "gsutil", "gsutil") | 10 gsutil_exe = os.path.join(depot_tools_path, "third_party", "gsutil", "gsutil") |
14 | 11 |
15 # We're downloading from a public bucket which does not need authentication, | 12 # We're downloading from a public bucket which does not need authentication, |
16 # but the user might have busted credential files somewhere such as ~/.boto | 13 # but the user might have busted credential files somewhere such as ~/.boto |
17 # that the gsutil script will try (and fail) to use. Setting these | 14 # that the gsutil script will try (and fail) to use. Setting these |
18 # environment variables convinces gsutil not to attempt to use these, but | 15 # environment variables convinces gsutil not to attempt to use these, but |
19 # also generates a useless warning about failing to load the file. We want | 16 # also generates a useless warning about failing to load the file. We want |
20 # to discard this warning but still preserve all output in the case of an | 17 # to discard this warning but still preserve all output in the case of an |
21 # actual failure. So, we run the script and capture all output and then | 18 # actual failure. So, we run the script and capture all output and then |
22 # throw the output away if the script succeeds (return code 0). | 19 # throw the output away if the script succeeds (return code 0). |
23 env = os.environ.copy() | 20 env = os.environ.copy() |
24 env["AWS_CREDENTIAL_FILE"] = "" | 21 env["AWS_CREDENTIAL_FILE"] = "" |
25 env["BOTO_CONFIG"] = "" | 22 env["BOTO_CONFIG"] = "" |
26 try: | 23 try: |
27 subprocess.check_output( | 24 subprocess.check_output( |
28 [gsutil_exe, | 25 [gsutil_exe, |
29 "--bypass_prodaccess", | 26 "--bypass_prodaccess", |
30 "cp", | 27 "cp", |
31 gs_path, | 28 gs_path, |
32 output_path], | 29 output_path], |
33 stderr=subprocess.STDOUT, | 30 stderr=subprocess.STDOUT, |
34 env=env) | 31 env=env) |
35 except subprocess.CalledProcessError as e: | 32 except subprocess.CalledProcessError as e: |
36 print e.output | 33 print e.output |
37 sys.exit(1) | 34 sys.exit(1) |
OLD | NEW |