occheung
/
artiq
forked from M-Labs/artiq
1
0
Fork 0
Code Pull Requests Activity

tools/file_import: simplify, remove deprecated load_module() call 

Signed-off-by: Etienne Wodey <wodey@iqo.uni-hannover.de>
This commit is contained in:
Etienne Wodey 2021-02-09 16:13:36 +01:00 committed by Sébastien Bourdeauducq
parent e54dd08821
commit af411de639
1 changed files with 9 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

21
artiq/tools.py
View File

@ -1,7 +1,8 @@
import asyncio
import importlib.machinery
import importlib.util
import logging
import os
import pathlib
import string
import sys
@ -69,20 +70,16 @@ def short_format(v):
def file_import(filename, prefix="file_import_"):
modname = filename
i = modname.rfind("/")
if i > 0:
modname = modname[i+1:]
i = modname.find(".")
if i > 0:
modname = modname[:i]
modname = prefix + modname
filename = pathlib.Path(filename)
modname = prefix + filename.stem
path = os.path.dirname(os.path.realpath(filename))
path = str(filename.resolve().parent)
sys.path.insert(0, path)
try:
loader = importlib.machinery.SourceFileLoader(modname, filename)
module = loader.load_module()
spec = importlib.util.spec_from_file_location(modname, filename)
module = importlib.util.module_from_spec(spec)
spec.loader.exec_module(module)
finally:
sys.path.remove(path)