我见过很多人从一个模块中提取所有类的例子，通常是这样的:
# 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 Foo
dir(Foo)
import collections
dir(collections)
其他回答
我可以从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']
)
))
这是一行，我用来获得所有的类，已定义在当前模块(即未导入)。根据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__]
在我的经验中，这是更有用的。
我不知道是否有一个“适当的”方法来做到这一点，但你的代码片段是在正确的轨道上:只需将import foo添加到foo.py，执行inspect.getmembers(foo)，它应该可以正常工作。
import pyclbr
print(pyclbr.readmodule(__name__).keys())
注意，stdlib的Python类浏览器模块使用静态源分析，因此它只适用于由真正的.py文件支持的模块。
试试这个:
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()接受一个谓词。
