Adapt to ndk.hosts changes.

Test: ndk/checkbuild.py
Bug: None
Change-Id: I67baeb6cf6c15f08a8e5653aca95eec5382b1a31

1 files changed, 8 insertions, 3 deletions

diff --git a/build.py b/build.py
index b5ff7fae..8893c2e3 100755
--- a/build.py
+++ b/build.py
@@ -21,8 +21,12 @@ import os
 import site
 
 site.addsitedir(os.path.join(os.path.dirname(__file__), '../../ndk/build/lib'))
+site.addsitedir(os.path.join(os.path.dirname(__file__), '../../ndk'))
 
-import build_support  # pylint: disable=import-error
+# pylint: disable=import-error,wrong-import-position
+import build_support
+from ndk.hosts import Host
+# pylint: enable=import-error,wrong-import-position
 
 
 def main(args):
@@ -31,14 +35,15 @@ def main(args):
         build_support.ndk_path(),
     ]
 
-    if args.host in ('windows', 'windows64'):
+    if args.host.is_windows:
         build_cmd.append('--mingw')
 
-    if args.host != 'windows':
+    if args.host != Host.Windows:
         build_cmd.append('--try-64')
 
     build_cmd.append('--build-dir=' + os.path.join(args.out_dir, 'yasm'))
     build_support.build(build_cmd, args, intermediate_package=True)
 
+
 if __name__ == '__main__':
     build_support.run(main)