diff options
Diffstat (limited to 'src/main/java/com/google/devtools/build/lib')
-rw-r--r-- | src/main/java/com/google/devtools/build/lib/bazel/rules/python/python_stub_template.txt | 30 |
1 files changed, 12 insertions, 18 deletions
diff --git a/src/main/java/com/google/devtools/build/lib/bazel/rules/python/python_stub_template.txt b/src/main/java/com/google/devtools/build/lib/bazel/rules/python/python_stub_template.txt index 2aacd411e6..6adf21f01d 100644 --- a/src/main/java/com/google/devtools/build/lib/bazel/rules/python/python_stub_template.txt +++ b/src/main/java/com/google/devtools/build/lib/bazel/rules/python/python_stub_template.txt @@ -71,25 +71,19 @@ def CreatePythonPathEntries(python_imports, module_space): # Find the runfiles tree def FindModuleSpace(): - # Follow symlinks, looking for my module space stub_filename = os.path.abspath(sys.argv[0]) - while True: - # Found it? - module_space = stub_filename + '.runfiles' - if os.path.isdir(module_space): - break - - runfiles_pattern = "(.*\.runfiles)/.*" - if IsWindows(): - runfiles_pattern = "(.*\.runfiles)\\.*" - matchobj = re.match(runfiles_pattern, os.path.abspath(sys.argv[0])) - if matchobj: - module_space = matchobj.group(1) - break - - raise AssertionError('Cannot find .runfiles directory for %s' % - sys.argv[0]) - return module_space + module_space = stub_filename + '.runfiles' + if os.path.isdir(module_space): + return module_space + + runfiles_pattern = "(.*\.runfiles)/.*" + if IsWindows(): + runfiles_pattern = "(.*\.runfiles)\\.*" + matchobj = re.match(runfiles_pattern, os.path.abspath(sys.argv[0])) + if matchobj: + return matchobj.group(1) + + raise AssertionError('Cannot find .runfiles directory for %s' % sys.argv[0]) # Create the runfiles tree by extracting the zip file def CreateModuleSpace(): |