如何在Python中临时修改sys.path？

Question

在我的Python测试脚本中，我多次使用此模式：

sys.path.insert(0, "somedir")
mod =  __import__(mymod)
sys.path.pop(0)

有没有更简洁的方法来临时修改搜索路径？

Answer 1

您可以使用一个简单的上下文管理器：

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']