导入并运行文件夹中的所有模块 [英] Import and run all modules in folder

问题描述

有没有办法导入并运行文件夹中的所有.py文件？

我认为pytest基本上是如何进行测试发现的。

我似乎希望从文件路径导入到列表中，然后在列表中迭代调用每个模块的.main(*args)。

它还必须与Python2和3兼容

示例

• start.py
• 事物/
  • mod1.py
    • main(*args)：
  • mod2.py
    • main(*args)：

推荐答案

可以使用importlib。假设以下是简单的目录结构：

• a.py
• b.py
• c.py

a.py和b.py包含以下简单函数：

def main(name):
    print name

在c.py中，我们可以迭代我们的目录并使用importlib.import_module导入每个文件。我们必须确保使导入的模块可以全局访问，否则它们将只在for循环的本地。

c.py：

import importlib

files = ['a', 'b']

for f in files:
    globals()[f] = importlib.import_module(f)

a.main('adam')
b.main('ben') 

运行c.py会产生以下输出：

adam
ben

这篇关于导入并运行文件夹中的所有模块的文章就介绍到这了，希望我们推荐的答案对大家有所帮助，也希望大家多多支持IT屋！