2
0
mirror of https://github.com/m-labs/artiq.git synced 2024-12-28 20:53:35 +08:00

tools/file_import: make sure sys.path is always restored

This commit is contained in:
Sebastien Bourdeauducq 2016-01-31 20:33:17 +01:00
parent 0d18942869
commit cf465dae0c

View File

@ -85,11 +85,11 @@ def file_import(filename, prefix="file_import_"):
path = os.path.dirname(os.path.realpath(filename))
sys.path.insert(0, path)
loader = importlib.machinery.SourceFileLoader(modname, filename)
module = loader.load_module()
sys.path.remove(path)
try:
loader = importlib.machinery.SourceFileLoader(modname, filename)
module = loader.load_module()
finally:
sys.path.remove(path)
return module