• Home
  • Line#
  • Scopes#
  • Navigate#
  • Raw
  • Download
1#!/usr/bin/env python
2
3import os
4import re
5import tempfile
6import shutil
7import sys
8import subprocess
9import zipfile
10
11PYTHON_BINARY = '%interpreter%'
12MAIN_FILE = '%main%'
13PYTHON_PATH = 'PYTHONPATH'
14ZIP_RUNFILES_DIRECTORY_NAME = 'runfiles'
15
16def SearchPathEnv(name):
17  search_path = os.getenv('PATH', os.defpath).split(os.pathsep)
18  for directory in search_path:
19    if directory == '': continue
20    path = os.path.join(directory, name)
21    if os.path.islink(path):
22      path = os.path.realpath(path)
23    # Check if path is actual executable file.
24    if os.path.isfile(path) and os.access(path, os.X_OK):
25      return path
26  return None
27
28def FindPythonBinary():
29  if PYTHON_BINARY.startswith('/'):
30    # Case 1: Python interpreter is directly provided with absolute path.
31    return PYTHON_BINARY
32  else:
33    # Case 2: Find Python interpreter through environment variable: PATH.
34    return SearchPathEnv(PYTHON_BINARY)
35
36# Create the runfiles tree by extracting the zip file
37def ExtractRunfiles():
38  temp_dir = tempfile.mkdtemp("", "Soong.python_")
39  zf = zipfile.ZipFile(os.path.dirname(__file__))
40  zf.extractall(temp_dir)
41  return os.path.join(temp_dir, ZIP_RUNFILES_DIRECTORY_NAME)
42
43def Main():
44  args = sys.argv[1:]
45
46  new_env = {}
47
48  try:
49    runfiles_path = ExtractRunfiles()
50
51    # Add runfiles path to PYTHONPATH.
52    python_path_entries = [runfiles_path]
53
54    # Add top dirs within runfiles path to PYTHONPATH.
55    top_entries = [os.path.join(runfiles_path, i) for i in os.listdir(runfiles_path)]
56    top_pkg_dirs = [i for i in top_entries if os.path.isdir(i)]
57    python_path_entries += top_pkg_dirs
58
59    old_python_path = os.environ.get(PYTHON_PATH)
60    separator = ':'
61    new_python_path = separator.join(python_path_entries)
62
63    # Copy old PYTHONPATH.
64    if old_python_path:
65      new_python_path += separator + old_python_path
66    new_env[PYTHON_PATH] = new_python_path
67
68    # Now look for main python source file.
69    main_filepath = os.path.join(runfiles_path, MAIN_FILE)
70    assert os.path.exists(main_filepath), \
71           'Cannot exec() %r: file not found.' % main_filepath
72    assert os.access(main_filepath, os.R_OK), \
73           'Cannot exec() %r: file not readable.' % main_filepath
74
75    python_program = FindPythonBinary()
76    if python_program is None:
77      raise AssertionError('Could not find python binary: ' + PYTHON_BINARY)
78    args = [python_program, main_filepath] + args
79
80    os.environ.update(new_env)
81
82    sys.stdout.flush()
83    retCode = subprocess.call(args)
84    exit(retCode)
85  except:
86    raise
87  finally:
88    shutil.rmtree(os.path.dirname(runfiles_path), True)
89
90if __name__ == '__main__':
91  Main()
92