Iterar sobre las subclases de una clase dada en un módulo dado

Question

En Python, dado un módulo X y una clase Y, ¿cómo puedo iterar o generar una lista de todas las subclases de Y que existen en el módulo X?

Answer 1

Ésta es una forma de hacerlo:

import inspect 

def get_subclasses(mod, cls): 
    """Yield the classes in module ``mod`` that inherit from ``cls``""" 
    for name, obj in inspect.getmembers(mod): 
     if hasattr(obj, "__bases__") and cls in obj.__bases__: 
      yield obj 

Answer 2

Dado el módulo de foo.py

class foo(object): pass 
class bar(foo): pass 
class baz(foo): pass 

class grar(Exception): pass 

def find_subclasses(module, clazz): 
    for name in dir(module): 
     o = getattr(module, name) 

     try: 
      if issubclass(o, clazz): 
      yield name, o 
     except TypeError: pass 

>>> import foo 
>>> list(foo.find_subclasses(foo, foo.foo)) 
[('bar', <class 'foo.bar'>), ('baz', <class 'foo.baz'>), ('foo', <class 'foo.foo'>)] 
>>> list(foo.find_subclasses(foo, object)) 
[('bar', <class 'foo.bar'>), ('baz', <class 'foo.baz'>), ('foo', <class 'foo.foo'>), ('grar', <class 'foo.grar'>)] 
>>> list(foo.find_subclasses(foo, Exception)) 
[('grar', <class 'foo.grar'>)] 

Answer 3

¿Puedo sugerir que ninguna de las respuestas de Chris AtLee y zacherates cumplir con los requisitos? creo que esta modificación a zacerates respuesta es mejor:

def find_subclasses(module, clazz): 
    for name in dir(module): 
     o = getattr(module, name) 
     try: 
      if (o != clazz) and issubclass(o, clazz): 
       yield name, o 
     except TypeError: pass 

La razón estoy de acuerdo con las respuestas dadas es que el primero no produce las clases que son una subclase distante de la clase dada, y el segundo incluye el dado clase.

Answer 4

Aunque la sugerencia de Quamrana funciona bien, hay un par de posibles mejoras que me gustaría sugerir para hacerlo más pitónico. Confían en usar el módulo de inspección de la biblioteca estándar.

1. Usted puede evitar la llamada getattr utilizando inspect.getmembers()
2. El try/catch se pueden evitar mediante el uso de inspect.isclass()

Con ellos, se puede reducir todo a una lista única de comprensión si como:

def find_subclasses(module, clazz): 
    return [ 
     cls 
      for name, cls in inspect.getmembers(module) 
       if inspect.isclass(cls) and issubclass(cls, clazz) 
    ]