import sys
class add_path():
def __init__(self, path):
self.path = path
def __enter__(self):
sys.path.insert(0, self.path)
def __exit__(self, exc_type, exc_value, traceback):
try:
sys.path.remove(self.path)
except ValueError:
pass
然后要导入模块,您可以执行以下操作:
with add_path('/path/to/dir'):
mod = __import__('mymodule')
从with
语句主体退出时,sys.path
将恢复到原始状态。如果仅使用该块中的模块,则可能还希望从sys.modules
以下位置删除其引用:
del sys.modules['mymodule']