Я видел много примеров, когда люди извлекают все классы из модуля, обычно это выглядит примерно так:
# 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()
Возможно, это что-то очень очевидное, но я ничего не смог найти. Кто-нибудь может мне помочь?
Попробуйте сделать это:
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()
принимает предикат.
Я был в состоянии получить все, что мне нужно от реж
построен в плюс функцией 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']
)
))
import pyclbr
print(pyclbr.readmodule(__name__).keys())
Обратите внимание, что в stdlib'ы на Python браузерный модуль класса использует статический анализ исходного кода, так он работает только для модулей, которые подкреплены Реаль.файл py`.
Если вы хотите, чтобы все классы, которые относятся к текущему модулю, вы могли бы использовать это :
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)
Если вы используете Надя'ы ответьте и Вы были импортировать другие классы на ваш модуль, что занятия будут слишком импортируется.
Так что'ы почему-членов.модуль == имядобавляется в предикат, используемый на
is_class_member`. Этот оператор проверяет, что класс действительно принадлежит модуль.
Предикат-это функция (отзывные), которая возвращает логическое значение.
Еще одно решение, которое работает в 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()
Это линии, которые я использую, чтобы получить все классы, которые были определены в текущем модуле (т. е. не импортируются). Это's немного долго по Пеп-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__]
Это было более полезно, по моему опыту.
Я думаю, что вы можете сделать что-то подобное.
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())`
если вам нужны собственные классы