Importar todas las clases en el directorio?

Question

Encontré this, pero eso no es todo lo que quiero hacer.

Quiero importar todas las clases en todos los archivos de un directorio. Básicamente, quiero reemplazar esto:

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

Con algo dinámico, por lo que no tengo a mantener la edición de mi __init__.py cada vez que agregar otro archivo.

---

La solución glob parece ser el equivalente de

import A 
import B 
import C 

que no es el mismo en todos.

Answer 1

Puede hacer algo como esto, aunque tenga en cuenta que isinstance(cls, type) solo funciona con clases de estilo nuevo.

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)