我见过很多人从一个模块中提取所有类的例子，通常是这样的:
# 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()
这可能是非常明显的事情，但我还没有找到任何东西。有人能帮帮我吗?
当前回答
我经常发现自己在编写命令行实用程序时，第一个参数指的是许多不同类中的一个。例如。/something.py feature command——arguments，其中feature是一个类，command是该类上的一个方法。下面是一个基类，它可以使此操作变得简单。
假设这个基类与它的所有子类一起驻留在一个目录中。然后你可以调用ArgBaseClass(foo = bar).load_subclasses()，它将返回一个字典。例如，如果目录是这样的:
arg_base_class.py feature.py
假设Feature .py实现了类Feature(ArgBaseClass)，那么上述load_subclasses调用将返回{' Feature ': <Feature对象>}。相同的kwargs (foo = bar)将被传递到Feature类中。
#!/usr/bin/env python3
import os, pkgutil, importlib, inspect
class ArgBaseClass():
# Assign all keyword arguments as properties on self, and keep the kwargs for later.
def __init__(self, **kwargs):
self._kwargs = kwargs
for (k, v) in kwargs.items():
setattr(self, k, v)
ms = inspect.getmembers(self, predicate=inspect.ismethod)
self.methods = dict([(n, m) for (n, m) in ms if not n.startswith('_')])
# Add the names of the methods to a parser object.
def _parse_arguments(self, parser):
parser.add_argument('method', choices=list(self.methods))
return parser
# Instantiate one of each of the subclasses of this class.
def load_subclasses(self):
module_dir = os.path.dirname(__file__)
module_name = os.path.basename(os.path.normpath(module_dir))
parent_class = self.__class__
modules = {}
# Load all the modules it the package:
for (module_loader, name, ispkg) in pkgutil.iter_modules([module_dir]):
modules[name] = importlib.import_module('.' + name, module_name)
# Instantiate one of each class, passing the keyword arguments.
ret = {}
for cls in parent_class.__subclasses__():
path = cls.__module__.split('.')
ret[path[-1]] = cls(**self._kwargs)
return ret
其他回答
我经常发现自己在编写命令行实用程序时，第一个参数指的是许多不同类中的一个。例如。/something.py feature command——arguments，其中feature是一个类，command是该类上的一个方法。下面是一个基类，它可以使此操作变得简单。
假设这个基类与它的所有子类一起驻留在一个目录中。然后你可以调用ArgBaseClass(foo = bar).load_subclasses()，它将返回一个字典。例如，如果目录是这样的:
arg_base_class.py feature.py
假设Feature .py实现了类Feature(ArgBaseClass)，那么上述load_subclasses调用将返回{' Feature ': <Feature对象>}。相同的kwargs (foo = bar)将被传递到Feature类中。
#!/usr/bin/env python3
import os, pkgutil, importlib, inspect
class ArgBaseClass():
# Assign all keyword arguments as properties on self, and keep the kwargs for later.
def __init__(self, **kwargs):
self._kwargs = kwargs
for (k, v) in kwargs.items():
setattr(self, k, v)
ms = inspect.getmembers(self, predicate=inspect.ismethod)
self.methods = dict([(n, m) for (n, m) in ms if not n.startswith('_')])
# Add the names of the methods to a parser object.
def _parse_arguments(self, parser):
parser.add_argument('method', choices=list(self.methods))
return parser
# Instantiate one of each of the subclasses of this class.
def load_subclasses(self):
module_dir = os.path.dirname(__file__)
module_name = os.path.basename(os.path.normpath(module_dir))
parent_class = self.__class__
modules = {}
# Load all the modules it the package:
for (module_loader, name, ispkg) in pkgutil.iter_modules([module_dir]):
modules[name] = importlib.import_module('.' + name, module_name)
# Instantiate one of each class, passing the keyword arguments.
ret = {}
for cls in parent_class.__subclasses__():
path = cls.__module__.split('.')
ret[path[-1]] = cls(**self._kwargs)
return ret
另一个适用于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()
import Foo
dir(Foo)
import collections
dir(collections)
进入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
我不知道是否有一个“适当的”方法来做到这一点，但你的代码片段是在正确的轨道上:只需将import foo添加到foo.py，执行inspect.getmembers(foo)，它应该可以正常工作。
