Как я могу получить список всех классов в текущем модуле в Python?

Я видел много примеров людей, извлекающих все классы из модуля, обычно что-то вроде:

# foo.py
class Foo:
    pass

# test.py
import inspect
import foo

for name, obj in inspect.getmembers(foo):
    if inspect.isclass(obj):
        print obj

потрясающе.

но я не могу выяснить, как получить все классы настоящее.

# foo.py
import inspect

class Foo:
    pass

def print_classes():
    for name, obj in inspect.getmembers(???): # what do I do here?
        if inspect.isclass(obj):
            print obj

# test.py
import foo

foo.print_classes()

это, вероятно, что-то действительно очевидное, но я ничего не смог найти. Кто-нибудь может мне помочь?

9 237
python reflection inspect

9 ответов:

попробуйте это:

import sys
current_module = sys.modules[__name__]

в вашем контексте:

import sys, inspect
def print_classes():
    for name, obj in inspect.getmembers(sys.modules[__name__]):
        if inspect.isclass(obj):
            print(obj)

а еще лучше:

clsmembers = inspect.getmembers(sys.modules[__name__], inspect.isclass)

, потому что inspect.getmembers() занимает сказуемое.

301

а как же

g = globals().copy()
for name, obj in g.iteritems():

?

17

Я не знаю, есть ли "правильный" способ сделать это, но ваш фрагмент находится на правильном пути: просто добавьте import foo to foo.py-делай inspect.getmembers(foo), и он должен работать нормально.

10

я смог получить все, что мне нужно от dir встроенный плюс getattr.

# Works on pretty much everything, but be mindful that 
# you get lists of strings back

print dir(myproject)
print dir(myproject.mymodule)
print dir(myproject.mymodule.myfile)
print dir(myproject.mymodule.myfile.myclass)

# But, the string names can be resolved with getattr, (as seen below)

хотя, это выходит похожим на комок волос:

def list_supported_platforms():
    """
        List supported platforms (to match sys.platform)

        @Retirms:
            list str: platform names
    """
    return list(itertools.chain(
        *list(
            # Get the class's constant
            getattr(
                # Get the module's first class, which we wrote
                getattr(
                    # Get the module
                    getattr(platforms, item),
                    dir(
                        getattr(platforms, item)
                    )[0]
                ),
                'SYS_PLATFORMS'
            )
            # For each include in platforms/__init__.py 
            for item in dir(platforms)
            # Ignore magic, ourselves (index.py) and a base class.
            if not item.startswith('__') and item not in ['index', 'base']
        )
    ))
7
import pyclbr
print(pyclbr.readmodule(__name__).keys())

обратите внимание, что модуль браузера класса Python stdlib использует статический анализ источника, поэтому он работает только для модулей, которые поддерживаются реальным .

6

Если вы хотите иметь все классы, которые принадлежат к текущему модулю, вы можете использовать это:

import sys, inspect
def print_classes():
    is_class_member = lambda member: inspect.isclass(member) and member.__module__ == __name__
    clsmembers = inspect.getmembers(sys.modules[__name__], is_class_member)

Если вы используете ответ Нади, и вы импортировали другие классы в свой модуль, эти классы тоже будут импортированы.

вот почему member.__module__ == __name__ добавляется к предикату, используемому на is_class_member. Этот оператор проверяет, что класс действительно принадлежит модуль.

предикат-это функция (вызываемая), которая возвращает логическое значение значение.

4

другое решение, которое работает в Python 2 и 3:

#foo.py
import sys

class Foo(object):
    pass

def print_classes():
    current_module = sys.modules[__name__]
    for key in dir(current_module):
        if isinstance( getattr(current_module, key), type ):
            print(key)

# test.py
import foo
foo.print_classes()
2

Это линии, которые я использую, чтобы получить все классы, которые были определены в текущем модуле (т. е. не импортируются). Это немного долго в соответствии с PEP-8, но вы можете изменить его, как вы считаете нужным.

import sys
import inspect

classes = [name for name, obj in inspect.getmembers(sys.modules[__name__], inspect.isclass) 
          if obj.__module__ is __name__]

это дает вам список имен классов. Если вы хотите, чтобы сами объекты класса просто сохраняли obj.

classes = [obj for name, obj in inspect.getmembers(sys.modules[__name__], inspect.isclass)
          if obj.__module__ is __name__]

Это было более полезно в моем опыте.

1

Я думаю, что вы можете сделать что-то подобное.

class custom(object):
    __custom__ = True
class Alpha(custom):
    something = 3
def GetClasses():
    return [x for x in globals() if hasattr(globals()[str(x)], '__custom__')]
print(GetClasses())`

Если вам нужны собственные классы

0