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