bootstrap: Invoke "pip" through Python.

bootstrap/bootstrap.py

 #!/usr/bin/env python
 # Copyright 2016 The LUCI Authors. All rights reserved.
 # Use of this source code is governed under the Apache License, Version 2.0
 # that can be found in the LICENSE file.
 
 import argparse
 import contextlib
 import glob
 import logging
 import os
 import re
 import shutil
 import subprocess
 import sys
 import tempfile
 
 from util import STORAGE_URL, OBJECT_URL, LOCAL_STORAGE_PATH, LOCAL_OBJECT_URL
 from util import read_deps, merge_deps, print_deps, platform_tag
 
 LOGGER = logging.getLogger(__name__)
 
 # /path/to/recipes-py
 ROOT = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
 
 PYTHON_BAT_WIN = '@%~dp0\\..\\Scripts\\python.exe %*'
 
+IS_WINDOWS = sys.platform.startswith('win')
+
+EXE_SUFFIX = '.exe' if IS_WINDOWS else ''
+
 
 class NoWheelException(Exception):
   def __init__(self, name, version, build, source_sha):
     super(NoWheelException, self).__init__(
         'No matching wheel found for (%s==%s (build %s_%s))' %
         (name, version, build, source_sha))
 
 
 def check_pydistutils():
   if os.path.exists(os.path.expanduser('~/.pydistutils.cfg')):
@@ skipping 31 matching lines @@
 
 
 def sha_for(deps_entry):
   if 'rev' in deps_entry:
     return deps_entry['rev']
   else:
     return deps_entry['gs'].split('.')[0]
 
 
 def install(deps, cache_root):
-  bin_dir = 'Scripts' if sys.platform.startswith('win') else 'bin'
+  bin_dir = 'Scripts' if IS_WINDOWS else 'bin'
+  python = os.path.join(sys.prefix, bin_dir, 'python%s' % (EXE_SUFFIX,))
   pip = os.path.join(sys.prefix, bin_dir, 'pip')
 
   requirements = []
   # TODO(iannucci): Do this as a requirements.txt
   for name, deps_entry in deps.iteritems():
     if not deps_entry.get('implicit'):
       requirements.append('%s==%s' % (name, deps_entry['version']))
 
   if not cache_root:
     cache_root = os.path.join(ROOT, '.bootstrap_cache')
   wheel_cache_dir = os.path.join(cache_root, 'wheels')
   download_cache_dir = os.path.join(cache_root, 'download_cache')
 
   for path in (wheel_cache_dir, download_cache_dir):
     if not os.path.exists(path):
       os.makedirs(path)
 
   env = os.environ.copy()
   env['PIP_CACHE_DIR'] = download_cache_dir
 
   # Download the package into our cache directory.
   subprocess.check_call([
-      pip, 'download',
+      python, pip, 'download',
       '--find-links', 'file://' + wheel_cache_dir,
       '--dest', wheel_cache_dir,
       ] + requirements,
       env=env)
 
   # Install the downloaded packages.
   subprocess.check_call([
-      pip, 'install',
+      python, pip, 'install',
       '--find-links', 'file://' + wheel_cache_dir,
       '--no-index',
       ] + requirements,
       env=env)
 
 
 def activate_env(env, deps, quiet=False, cache_root=None):
   if hasattr(sys, 'real_prefix'):
     LOGGER.error('Already activated environment!')
     return
@@ skipping 91 matching lines @@
 
   deps = merge_deps(map(os.path.abspath, opts.deps_file))
   activate_env(os.path.abspath(opts.env_path), deps, opts.quiet,
                cache_root=os.path.abspath(opts.cache_root))
 
 
 if __name__ == '__main__':
   logging.basicConfig()
   LOGGER.setLevel(logging.DEBUG)
   sys.exit(main(sys.argv[1:]))