Index: build/android/pylib/sdk/split_select.py |
diff --git a/build/android/pylib/sdk/split_select.py b/build/android/pylib/sdk/split_select.py |
new file mode 100644 |
index 0000000000000000000000000000000000000000..e20466214239d9ca4c3819ad3d02d10eebc72e44 |
--- /dev/null |
+++ b/build/android/pylib/sdk/split_select.py |
@@ -0,0 +1,58 @@ |
+# Copyright 2015 The Chromium Authors. All rights reserved. |
+# Use of this source code is governed by a BSD-style license that can be |
+# found in the LICENSE file. |
+ |
+"""This module wraps Android's split-select tool.""" |
+ |
+import os |
+ |
+from pylib import cmd_helper |
+from pylib import constants |
+from pylib.utils import timeout_retry |
+ |
+_SPLIT_SELECT_PATH = os.path.join(constants.ANDROID_SDK_TOOLS, 'split-select') |
+ |
+def _RunSplitSelectCmd(args): |
+ """Runs a split-select command. |
+ |
+ Args: |
+ args: A list of arguments for split-select. |
+ |
+ Returns: |
+ The output of the command. |
+ """ |
+ cmd = [_SPLIT_SELECT_PATH] + args |
+ status, output = cmd_helper.GetCmdStatusAndOutput(cmd) |
+ if status != 0: |
+ raise Exception('Failed running command "%s" with output "%s".' % |
+ (' '.join(cmd), output)) |
+ return output |
+ |
+def _SplitConfig(device): |
+ """Returns a config specifying which APK splits are required by the device. |
+ |
+ Args: |
+ device: A DeviceUtils object. |
+ """ |
+ return ('%s-r%s-%s:%s' % |
+ (device.language, |
+ device.country, |
+ device.screen_density, |
+ device.product_cpu_abi)) |
+ |
+def SelectSplits(device, base_apk, split_apks): |
+ """Determines which APK splits the device requires. |
+ |
+ Args: |
+ device: A DeviceUtils object. |
+ base_apk: The path of the base APK. |
+ split_apks: A list of paths of APK splits. |
+ |
+ Returns: |
+ The list of APK splits that the device requires. |
+ """ |
+ config = _SplitConfig(device) |
+ args = ['--target', config, '--base', base_apk] |
+ for split in split_apks: |
+ args.extend(['--split', split]) |
+ return _RunSplitSelectCmd(args).splitlines() |