mirror of
https://github.com/davidhalter/jedi.git
synced 2026-01-17 01:15:02 +08:00
27 lines
612 B
Python
27 lines
612 B
Python
import os
|
|
from contextlib import contextmanager
|
|
|
|
|
|
def traverse_parents(path, include_current=False):
|
|
if not include_current:
|
|
path = os.path.dirname(path)
|
|
|
|
previous = None
|
|
while previous != path:
|
|
yield path
|
|
previous = path
|
|
path = os.path.dirname(path)
|
|
|
|
|
|
@contextmanager
|
|
def monkeypatch(obj, attribute_name, new_value):
|
|
"""
|
|
Like pytest's monkeypatch, but as a context manager.
|
|
"""
|
|
old_value = getattr(obj, attribute_name)
|
|
try:
|
|
setattr(obj, attribute_name, new_value)
|
|
yield
|
|
finally:
|
|
setattr(obj, attribute_name, old_value)
|