从python上的文件夹导入所有文件

时间:2017-07-19 18:01:08

标签: python python-import

我有一个包含一些python文件的文件夹:

modules
|--> __init__.py
|--> car.py
|--> plane.py
|--> ship.py
|--> etc...

这些文件中的每一个都包含在:

car.py

NAME = "car_name"
def mainloop(*args):
    ...
    return whatever

plane.py

NAME = "some_different_name"
def mainloop(*args):
    ...
    return whatever

依旧......

我想从我的父目录动态导入它们作为这样的字典:

from modules import modules
print(modules)
{
    "car_name": <function mainloop at ...>,    
    "some_different_name": <function mainloop at ...>,
    etc..
}

我不知道文件夹或文件名上会有多少文件。 我该怎么办?

1 个答案:

答案 0 :(得分:5)

modules.py

import os, importlib
fs = [f for f in os.listdir('.') if f.endswith('.py')]
modules = {m.NAME: m.mainloop for m in map(importlib.import_module, fs)
               if hasattr(m, 'mainloop')}