forked from VimPlug/jedi
03b4177d3d
In the previous implementation, Jedi would's traverse_parents function traversed parent directories to the system root every time. This would inadvertently add every folder to the system root every time. Obviously, this is not the behavior desired for the import system. This pull request provides a new argument to the traverse_parents function, "root", which represents the root parent for the search. This argument defaults to None, thereby preserving the existing behavior of the function. I chose to duplicate some code for performance reasons. Since I'm trying to avoid too much path manipulation magic, we do: * a search to a valid specified root, OR * a simple upward search until hitting the system root when there is no valid root specified.
43 lines
1.2 KiB
Python
43 lines
1.2 KiB
Python
import os
|
|
from contextlib import contextmanager
|
|
|
|
|
|
def traverse_parents(path, root=None, include_current=False):
|
|
"""Iterate directories from a path to search root
|
|
|
|
:path: the path of the script/directory to check.
|
|
:root: the root of the upward search. Assumes the system root if root is
|
|
None.
|
|
:include_current: includes the current file / directory.
|
|
|
|
If the root path is not a substring of the provided path, assume the root
|
|
search path as well.
|
|
"""
|
|
if not include_current:
|
|
path = os.path.dirname(path)
|
|
|
|
previous = None
|
|
if root is None or not path.startswith(root):
|
|
while previous != path:
|
|
yield path
|
|
previous = path
|
|
path = os.path.dirname(path)
|
|
else:
|
|
while previous != root:
|
|
yield path
|
|
previous = path
|
|
path = os.path.dirname(path)
|
|
|
|
|
|
@contextmanager
|
|
def monkeypatch(obj, attribute_name, new_value):
|
|
"""
|
|
Like pytest's monkeypatch, but as a value manager.
|
|
"""
|
|
old_value = getattr(obj, attribute_name)
|
|
try:
|
|
setattr(obj, attribute_name, new_value)
|
|
yield
|
|
finally:
|
|
setattr(obj, attribute_name, old_value)
|