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

本文介绍了如何在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']