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

问题描述

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

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

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

---

解决方案

您可以使用简单的context manager：

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