从相对目录迭代导入Python脚本

Question

我有一个名为test.py的脚本，代码如下（我已经大大简化了事情）：

from foo import Bar

bar = Bar()
result = bar.do_something()

但我不仅仅有一个名为foo的脚本。我有许多名为foo的脚本，按以下目录结构组织：

└── project
    ├── code
    │   ├── test.py
    └── scripts
        ├── script_1
            └── foo.py
        ├── script_2
            └── foo.py
        ├── script_3
            └── foo.py
        ├── script_4
            └── foo.py
        ├── script_5
            └── foo.py

每个foo.py都有一个稍微不同的实现。我想对test.py做的是通过导入每个脚本并对其运行一些测试来测试所有脚本。下面是一些代码（*'s表示伪代码）

*Get all script directories*
*For each directory in script directories:*
    *import foo.py from this directory*

    bar = Bar()
    result = bar.do_something()

    *Save the result for this directory*

我该怎么做？特别是，如何迭代导入脚本，例如，*导入foo。是否从此目录中复制*？

Answer 1

我不得不做几件不同的事情，但结果代码片段如下所示：

import os, sys, importlib

# directory of your script folder
scripts_dir = os.path.dirname(os.path.dirname(os.path.realpath(__file__))+"/../scripts/")


for root, dirs, files in os.walk(scripts_dir):
    # traverse the folder and find ones that has foo.py
    if 'foo.py' in files:
        sys.path.append(root)
        out = importlib.import_module('foo', package=root)

        # call the function in foo.py.  
        #In this case, I assumed there is a function called test 
        met = getattr(out, 'test')
        print(met())

        # clean up the path and imported modules
        sys.path.remove(root)
        del sys.modules['foo']

Answer 2

我建议您更改脚本，为它们创建一个包，如图所示。然后，您可以通过以下方式单独访问每个脚本：

import scripts.script_1.foo

或

from scripts.script_1 import foo

迭代导入：

要遍历文件夹并导入它们，可以使用python的“importlib”库。您需要使用此库中的“导入模块”功能。尽管如此，您仍然需要包括_uinit。在每一个目录中。此处说明了使用此函数导入模块的示例。