我见过很多人从一个模块中提取所有类的例子,通常是这样的:

# 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()

这可能是非常明显的事情,但我还没有找到任何东西。有人能帮帮我吗?


当前回答

另一个适用于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()

其他回答

我经常发现自己在编写命令行实用程序时,第一个参数指的是许多不同类中的一个。例如。/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
import Foo 
dir(Foo)

import collections
dir(collections)

我不知道是否有一个“适当的”方法来做到这一点,但你的代码片段是在正确的轨道上:只需将import foo添加到foo.py,执行inspect.getmembers(foo),它应该可以正常工作。

我觉得你可以这样做。

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 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()接受一个谓词。