Chromium Code Reviews| Index: mojo/public/tools/download_shell_binary.py |
| diff --git a/mojo/public/tools/download_shell_binary.py b/mojo/public/tools/download_shell_binary.py |
| index b46139e7ae1c7107fc4966bd2c290ee048bff719..b5b9d4ae7a2c7f39230679942d1567a5677500fa 100755 |
| --- a/mojo/public/tools/download_shell_binary.py |
| +++ b/mojo/public/tools/download_shell_binary.py |
| @@ -10,6 +10,11 @@ import sys |
| import tempfile |
| import zipfile |
| +BINARY_FOR_PLATFORM = { |
| + "linux-x64" : "mojo_shell", |
| + "android-arm" : "MojoShell.apk" |
| +} |
| + |
| if not sys.platform.startswith("linux"): |
| print "Not supported for your platform" |
| sys.exit(0) |
| @@ -17,17 +22,9 @@ if not sys.platform.startswith("linux"): |
| def download(tools_directory): |
| current_path = os.path.dirname(os.path.realpath(__file__)) |
| - find_depot_tools_path = os.path.join(current_path, tools_directory) |
| - sys.path.insert(0, find_depot_tools_path) |
| - # pylint: disable=F0401 |
| - import find_depot_tools |
| - |
| prebuilt_file_path = os.path.join(current_path, "prebuilt") |
| stamp_path = os.path.join(prebuilt_file_path, "VERSION") |
| - depot_tools_path = find_depot_tools.add_depot_tools_to_path() |
| - gsutil_exe = os.path.join(depot_tools_path, "third_party", "gsutil", "gsutil") |
| - |
| version_path = os.path.join(current_path, "../VERSION") |
| with open(version_path) as version_file: |
| version = version_file.read().strip() |
| @@ -38,13 +35,28 @@ def download(tools_directory): |
| if current_version == version: |
| return 0 # Already have the right version. |
| except IOError: |
| - pass # If the stamp file does not exist we need to download a new binary. |
| + pass # If the stamp file does not exist we need to download new binaries. |
| - platform = "linux-x64" # TODO: configurate |
| - basename = platform + ".zip" |
| + for platform in ["linux-x64", "android-arm"]: |
| + download_version_for_platform(version, platform, tools_directory) |
| + |
| + with open(stamp_path, 'w') as stamp_file: |
| + stamp_file.write(version) |
|
qsr
2015/02/06 14:37:37
Are you missing a return value after this?
blundell
2015/02/06 14:41:09
Done.
|
| +def download_version_for_platform(version, platform, tools_directory): |
| + current_path = os.path.dirname(os.path.realpath(__file__)) |
| + prebuilt_file_path = os.path.join(current_path, "prebuilt") |
| + find_depot_tools_path = os.path.join(current_path, tools_directory) |
| + sys.path.insert(0, find_depot_tools_path) |
| + # pylint: disable=F0401 |
| + import find_depot_tools |
| + |
| + basename = platform + ".zip" |
| gs_path = "gs://mojo/shell/" + version + "/" + basename |
| + depot_tools_path = find_depot_tools.add_depot_tools_to_path() |
| + gsutil_exe = os.path.join(depot_tools_path, "third_party", "gsutil", "gsutil") |
| + |
| with tempfile.NamedTemporaryFile() as temp_zip_file: |
| # We're downloading from a public bucket which does not need authentication, |
| # but the user might have busted credential files somewhere such as ~/.boto |
| @@ -70,19 +82,18 @@ def download(tools_directory): |
| print e.output |
| sys.exit(1) |
| + binary_name = BINARY_FOR_PLATFORM[platform] |
| with zipfile.ZipFile(temp_zip_file.name) as z: |
| - zi = z.getinfo("mojo_shell") |
| + zi = z.getinfo(binary_name) |
| mode = zi.external_attr >> 16 |
| z.extract(zi, prebuilt_file_path) |
| - os.chmod(os.path.join(prebuilt_file_path, "mojo_shell"), mode) |
| + os.chmod(os.path.join(prebuilt_file_path, binary_name), mode) |
| - with open(stamp_path, 'w') as stamp_file: |
| - stamp_file.write(version) |
| return 0 |
|
qsr
2015/02/06 14:37:37
And you do not seem to be using this one.
blundell
2015/02/06 14:41:09
whoops, restored the return 0 to the parent functi
|
| def main(): |
| - parser = argparse.ArgumentParser(description="Download mojo_shell binary " |
| + parser = argparse.ArgumentParser(description="Download mojo_shell binaries " |
| "from google storage") |
| parser.add_argument("--tools-directory", |
| dest="tools_directory", |