如何导入并运行文件夹中所有.py文件？需兼容Python2/3

当然可以实现这种类似pytest自动发现模块并执行的机制，而且完全能兼容Python 2和3。我帮你梳理了完整的实现思路，还附上了可直接运行的代码示例：

1. 核心思路拆解 #

本质上和pytest的测试发现逻辑类似：

• 遍历目标文件夹，筛选出所有合法的.py模块文件
• 将目标文件夹加入Python的模块搜索路径，确保能成功导入
• 兼容Python 2/3的导入API，实现跨版本模块导入
• 遍历导入的模块，调用每个模块的main(*args)方法

2. 完整start.py代码实现 #

import os
import sys

def load_and_run_modules(folder_path, *args):
    # 将目标文件夹添加到Python模块搜索路径
    if folder_path not in sys.path:
        sys.path.insert(0, folder_path)
    
    # 遍历目标文件夹下的所有文件
    for filename in os.listdir(folder_path):
        # 只处理.py文件，排除__init__.py（避免导入包模块）
        if filename.endswith('.py') and filename != '__init__.py':
            # 从文件名提取模块名（去掉.py后缀）
            module_name = filename[:-3]
            
            # 兼容Python 2和3的模块导入方式
            try:
                # Python 3+ 使用importlib标准库
                import importlib
                module = importlib.import_module(module_name)
            except ImportError:
                # Python 2 回退到imp模块
                import imp
                module = imp.load_source(module_name, os.path.join(folder_path, filename))
            
            # 检查模块是否存在可调用的main方法，避免报错
            if hasattr(module, 'main') and callable(module.main):
                print(f"=== Running {module_name}.main() ===")
                module.main(*args)

if __name__ == '__main__':
    # 定位到当前脚本所在目录下的things文件夹
    things_folder = os.path.join(os.path.dirname(__file__), 'things')
    # 传递命令行参数（过滤掉start.py本身的路径）
    load_and_run_modules(things_folder, *sys.argv[1:])

3. 模块示例与测试 #

假设你的things/目录下的模块是这样的：

# things/mod1.py
def main(*args):
    print(f"Mod1 执行，参数：{args}")

# things/mod2.py
def main(*args):
    print(f"Mod2 执行，参数：{args}")

运行python start.py hello world，你会看到输出：

=== Running mod1.main() ===
Mod1 执行，参数：('hello', 'world')
=== Running mod2.main() ===
Mod2 执行，参数：('hello', 'world')

4. 额外注意事项 #

• 如果你的模块文件名包含不符合Python命名规则的字符（比如横杠、空格），需要额外添加文件名到模块名的转换逻辑
• 若需要排除特定文件，可以在筛选环节添加更多判断条件（比如跳过以_开头的文件）

内容的提问来源于stack exchange，提问作者Cirrith