python - 导入目录中的所有类？

Question

我找到了 this ，但这并不是我想要做的。

我想导入目录中所有文件中的所有类。基本上，我想替换这个:

from A import *
from B import *
from C import *

有了一些动态的东西，这样我就不必在每次添加另一个文件时都继续编辑我的 __init__.py。

---

glob 解决方案似乎等同于

import A
import B
import C

这完全不一样。

Answer 1

您可以这样做，但请记住 isinstance(cls, type) 仅适用于新式类。

import os, sys

path = os.path.dirname(os.path.abspath(__file__))

for py in [f[:-3] for f in os.listdir(path) if f.endswith('.py') and f != '__init__.py']:
    mod = __import__('.'.join([__name__, py]), fromlist=[py])
    classes = [getattr(mod, x) for x in dir(mod) if isinstance(getattr(mod, x), type)]
    for cls in classes:
        setattr(sys.modules[__name__], cls.__name__, cls)