用importlib替换imp是否保持旧行为

Question

我继承了一些需要返工的代码，因为它使用了不推荐使用的imp模块，而不是importlib

为了测试功能，我创建了一个简单的测试脚本：

# test.py
def main():
print("main()")

if __name__ == "__main__":
print("__main__")
main()

当我用旧代码运行它时(下面是一个最小的例子(

# original imp based minimal example
import imp
import os
import site
script = r"./test.py"
script_dir = os.path.dirname(script)
script_name = os.path.splitext(os.path.basename(script))[0]
site.addsitedir(script_dir)
fp, pathname, description = imp.find_module(script_name, [script_dir])
imp.load_source('__main__', pathname, fp)

将输出以下内容：

__main__
main()

现在，我在使用importlib模块逐字逐句地模仿这个过程时遇到了一些困难，因为我没有用以下代码获得如上所述的输出：

# my try so far using importlib
import importlib
import os
import site
script = os.path.abspath(r"./test.py")
script_dir = os.path.dirname(script)
script_name = os.path.splitext(os.path.basename(script))[0]
site.addsitedir(script_dir)
spec = importlib.util.spec_from_file_location(script_name, script)
module = importlib.util.module_from_spec(spec)
sys.modules[script_name] = module
spec.loader.exec_module(module)

相反，它什么也不输出-(关于如何使用importlib模仿旧的(imp(行为，有什么建议吗？

Answer 1

SourceFileLoader似乎可以满足您的需求：

from importlib.machinery import SourceFileLoader
script_path = os.path.abspath(r"./test.py")
foo = SourceFileLoader("my_script", script_path).load_module()
foo.do_stuff()

Answer 2

最终我根据https://github.com/epfl-scitas/spack/blob/a60ae07083a5744607064221a0cd48204a54394e/lib/spack/llnl/util/lang.py#L598-L625:

if sys.version_info[0] == 3: 
if sys.version_info[1] >= 5:
import importlib.util
spec = importlib.util.spec_from_file_location(module_name, module_path)
module = importlib.util.module_from_spec(spec)
spec.loader.exec_module(module)
elif sys.version_info[1] < 5:
import importlib.machinery
loader = importlib.machinery.SourceFileLoader(module_name, module_path)
loader.load_module()
elif sys.version_info[0] == 2:
import imp
imp.load_source(module_name, module_path)