python - stringvar - tkinter ejemplos

Iteraciones sobre 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

Aquí hay una manera 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 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 cumple con los requisitos? Creo que esta modificación para zacetear la 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 por la que no estoy de acuerdo con las respuestas dadas es que la primera no produce clases que son una subclase distante de la clase dada, y la segunda incluye la clase dada.

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. Puede evitar la llamada inspect.getmembers() utilizando inspect.getmembers()
2. El try / catch se puede evitar usando inspect.isclass()

Con ellos, puede reducir todo a una sola lista de comprensión si lo desea:

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