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

m-labs · Jan 31, 2016 · cf465da · cf465da
1 parent 0d18942
commit cf465da
Showing 1 changed file with 5 additions and 5 deletions.
diff --git a/artiq/tools.py b/artiq/tools.py
@@ -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