Iterate over subclasses of a given class in a given module

Question

In Python, given a module X and a class Y, how can I iterate or generate a list of all subclasses of Y that exist in module X?

Answer 1

Here's one way to do it:

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