llvm-project/lldb/packages/Python/lldbsuite/test/lldb_pylint_helper.py
Zachary Turner c1b7cd72db Python 3 - Turn on absolute imports, and fix existing imports.
Absolute imports were introduced in Python 2.5 as a feature
(e.g. from __future__ import absolute_import), and made default
in Python 3.

When absolute imports are enabled, the import system changes in
a couple of ways:

1) The `import foo` syntax will *only* search sys.path.  If `foo`
   isn't in sys.path, it won't be found.  Period.  Without absolute
   imports, the import system will also search the same directory
   that the importing file resides in, so that you can easily
   import from the same folder.

2) From inside a package, you can use a dot syntax to refer to higher
   levels of the current package.  For example, if you are in the
   package lldbsuite.test.utility, then ..foo refers to
   lldbsuite.test.foo.  You can use this notation with the
   `from X import Y` syntax to write intra-package references.  For
   example, using the previous locationa s a starting point, writing
   `from ..support import seven` would import lldbsuite.support.seven

Since this is now the default behavior in Python 3, this means that
importing from the same directory with `import foo` *no longer works*.
As a result, the only way to have portable code is to force absolute
imports for all versions of Python.

See PEP 0328 [https://www.python.org/dev/peps/pep-0328/] for more
information about absolute and relative imports.

Differential Revision: http://reviews.llvm.org/D14342
Reviewed By: Todd Fiala

llvm-svn: 252191
2015-11-05 19:22:28 +00:00

170 lines
6.1 KiB
Python

"""
The LLVM Compiler Infrastructure
This file is distributed under the University of Illinois Open Source
License. See LICENSE.TXT for details.
Sync lldb and related source from a local machine to a remote machine.
This facilitates working on the lldb sourcecode on multiple machines
and multiple OS types, verifying changes across all.
Provides helper support for adding lldb test paths to the python path.
"""
from __future__ import print_function
from __future__ import absolute_import
# System modules
import os
import platform
import subprocess
import sys
# Third-party modules
# LLDB modules
def add_lldb_test_paths(check_dir):
# pylint: disable=line-too-long
"""Adds lldb test-related paths to the python path.
Starting with the given directory and working upward through
each parent directory up to the root, it looks for the lldb
test directory. When found, the lldb test directory and its
child test_runner/lib directory will be added to the python
system path.
Instructions for use:
This method supports a simple way of getting pylint to be able
to reliably lint lldb python test scripts (including the test
infrastructure itself). To do so, add the following to a
.pylintrc file in your home directory:
[Master]
init-hook='import os; import sys; sys.path.append(os.path.expanduser("~/path/to/lldb/test")); import lldb_pylint_helper; lldb_pylint_helper.add_lldb_test_paths(os.getcwd()); print("sys.path={}\n".format(sys.path))'
Replace ~/path/to/lldb/test with a valid path to your local lldb source
tree. Note you can have multiple lldb source trees on your system, and
this will work just fine. The path in your .pylintrc is just needed to
find the paths needed for pylint in whatever lldb source tree you're in.
pylint will use the python files in whichever tree it is run from.
Note it is critical that the init-hook line be contained on a single line.
You can remove the print line at the end once you know the pythonpath is
getting set up the way you expect.
With these changes, you will be able to run the following, for example.
cd lldb/sourcetree/1-of-many/test/lang/c/anonymous
pylint TestAnonymous.py
This will work, and include all the lldb/sourcetree/1-of-many lldb-specific
python directories to your path.
You can then run it in another lldb source tree on the same machine like
so:
cd lldb/sourcetree/2-of-many/test/functionalities/inferior-assert
pyline TestInferiorAssert.py
and this will properly lint that file, using the lldb-specific python
directories from the 2-of-many source tree.
Note at the time I'm writing this, our tests are in pretty sad shape
as far as a stock pylint setup goes. But we need to start somewhere :-)
@param check_dir specifies a directory that will be used to start
looking for the lldb test infrastructure python library paths.
"""
# Add the test-related packages themselves.
add_lldb_test_package_paths(check_dir)
# Add the lldb directory itself
add_lldb_module_directory()
def add_lldb_module_directory():
"""
Desired Approach:
Part A: find an lldb
1. Walk up the parent chain from the current directory, looking for
a directory matching *build*. If we find that, use it as the
root of a directory search for an lldb[.exe] executable.
2. If 1 fails, use the path and look for an lldb[.exe] in there.
If Part A ends up with an lldb, go to part B. Otherwise, give up
on the lldb python module path.
Part B: use the output from 'lldb[.exe] -P' to find the lldb dir.
Current approach:
If Darwin, use 'xcrun lldb -P'; others: find lldb on path.
Drawback to current approach:
If the tester is changing the SB API (adding new methods), pylint
will not know about them as it is using the wrong lldb python module.
In practice, this should be minor.
"""
try:
lldb_module_path = None
if platform.system() == 'Darwin':
# Use xcrun to find the selected lldb.
lldb_module_path = subprocess.check_output(["xcrun", "lldb", "-P"])
elif platform.system() == 'Windows':
lldb_module_path = subprocess.check_output(
["lldb.exe", "-P"], shell=True)
else:
# Use the shell to run lldb from the path.
lldb_module_path = subprocess.check_output(
["lldb", "-P"], shell=True)
# Trim the result.
if lldb_module_path is not None:
lldb_module_path = lldb_module_path.strip()
# If we have a result, add it to the path
if lldb_module_path is not None and len(lldb_module_path) > 0:
sys.path.insert(0, lldb_module_path)
# pylint: disable=broad-except
except Exception as exception:
print("failed to find python path: {}".format(exception))
def add_lldb_test_package_paths(check_dir):
"""Adds the lldb test infrastructure modules to the python path.
See add_lldb_test_paths for more details.
@param check_dir the directory of the test.
"""
check_dir = os.path.realpath(check_dir)
while check_dir and len(check_dir) > 0:
# If the current directory is test, it might be the lldb/test
# directory. If so, we've found an anchor that will allow us
# to add the relevant lldb-sourcetree-relative python lib
# dirs.
if os.path.basename(check_dir) == 'test':
# If this directory has a dotest.py file in it,
# then this is an lldb test tree. Add the
# test directories to the python path.
if os.path.exists(os.path.join(check_dir, "dotest.py")):
sys.path.insert(0, check_dir)
sys.path.insert(0, os.path.join(
check_dir, "test_runner", "lib"))
break
# Continue looking up the parent chain until we have no more
# directories to check.
new_check_dir = os.path.dirname(check_dir)
# We're done when the new check dir is not different
# than the current one.
if new_check_dir == check_dir:
break
check_dir = new_check_dir