我见过很多人从一个模块中提取所有类的例子，通常是这样的:
# 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()
这可能是非常明显的事情，但我还没有找到任何东西。有人能帮帮我吗?
当前回答
我觉得你可以这样做。
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())`
如果你需要自己的类
其他回答
我觉得你可以这样做。
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())`
如果你需要自己的类
我不知道是否有一个“适当的”方法来做到这一点，但你的代码片段是在正确的轨道上:只需将import foo添加到foo.py，执行inspect.getmembers(foo)，它应该可以正常工作。
以下内容可以放在文件的顶部:
def get_classes():
import inspect, sys
return dict(inspect.getmembers(
sys.modules[__name__],
lambda member: inspect.isclass(member) and member.__module__ == __name__
))
注意，这可以放在模块的顶部，因为我们已经将逻辑包装在函数定义中。如果希望字典作为顶级对象存在，则需要将定义放在文件的底部，以确保包括所有类。
进入Python解释器。输入help ('module_name')，然后按Enter。 例如help('os')。 在这里，我把输出的一部分粘贴在下面:
class statvfs_result(__builtin__.object)
| statvfs_result: Result from statvfs or fstatvfs.
|
| This object may be accessed either as a tuple of
| (bsize, frsize, blocks, bfree, bavail, files, ffree, favail, flag, namemax),
| or via the attributes f_bsize, f_frsize, f_blocks, f_bfree, and so on.
|
| See os.statvfs for more information.
|
| Methods defined here:
|
| __add__(...)
| x.__add__(y) <==> x+y
|
| __contains__(...)
| x.__contains__(y) <==> y in x
这是一行，我用来获得所有的类，已定义在当前模块(即未导入)。根据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__]
在我的经验中，这是更有用的。
