Итерация над подклассами данного класса в данном модуле

Ответ 1

Вот один из способов сделать это:

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

Ответ 2

Хотя предложение Quamrana прекрасно работает, есть несколько возможных улучшений, которые я хотел бы предложить сделать более питоновскими. Они полагаются на использование модуля проверки из стандартной библиотеки.

• Вы можете избежать вызова getattr, используя inspect.getmembers()
• Попробуйте /catch можно избежать, используя inspect.isclass()

С их помощью вы можете уменьшить все это до единого понимания списка, если хотите:

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

Ответ 3

Могу ли я предположить, что ни один из ответов Криса Атли и Захатеса не отвечает требованиям? Я думаю, что эта модификация для ответа zacerates лучше:

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

Я не согласен с этими ответами в том, что первый не создает классы, которые являются отдаленным подклассом данного класса, а второй включает данный класс.

Ответ 4

Учитывая модуль 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'>)]