106

如何获取在 Python 中定义方法的类？

我希望以下示例打印“ __main__.FooClass”：

class FooClass:
    def foo_method(self):
        print "foo"

class BarClass(FooClass):
    pass

bar = BarClass()
print get_class_that_defined_method(bar.foo_method)
4

6 回答 6

76 
import inspect

def get_class_that_defined_method(meth):
    for cls in inspect.getmro(meth.im_class):
        if meth.__name__ in cls.__dict__: 
            return cls
    return None
于 2009-06-07T02:23:02.750 回答
9

感谢 Sr2222 指出我没有抓住重点......

这是正确的方法，就像亚历克斯的一样，但不需要导入任何东西。我不认为这是一种改进，除非继承类的层次结构很大，因为一旦找到定义类，这种方法就会停止，而不是像现在getmro这样返回整个继承。如前所述，这是一个非常不可能的情况。

def get_class_that_defined_method(method):
    method_name = method.__name__
    if method.__self__:    
        classes = [method.__self__.__class__]
    else:
        #unbound method
        classes = [method.im_class]
    while classes:
        c = classes.pop()
        if method_name in c.__dict__:
            return c
        else:
            classes = list(c.__bases__) + classes
    return None

和例子：

>>> class A(object):
...     def test(self): pass
>>> class B(A): pass
>>> class C(B): pass
>>> class D(A):
...     def test(self): print 1
>>> class E(D,C): pass

>>> get_class_that_defined_method(A().test)
<class '__main__.A'>
>>> get_class_that_defined_method(A.test)
<class '__main__.A'>
>>> get_class_that_defined_method(B.test)
<class '__main__.A'>
>>> get_class_that_defined_method(C.test)
<class '__main__.A'>
>>> get_class_that_defined_method(D.test)
<class '__main__.D'>
>>> get_class_that_defined_method(E().test)
<class '__main__.D'>
>>> get_class_that_defined_method(E.test)
<class '__main__.D'>
>>> E().test()
1

Alex 解决方案返回相同的结果。只要可以使用 Alex 方法，我就会使用它而不是这个。

于 2012-11-29T14:38:10.637 回答
8

我不知道为什么没有人提出这个问题，或者为什么当它慢得要命时，最佳答案有 50 个赞成票，但您也可以执行以下操作：

def get_class_that_defined_method(meth):
    return meth.im_class.__name__

对于 python 3，我相信这发生了变化，你需要研究.__qualname__.

于 2017-10-02T21:09:46.597 回答
7

在 Python 3 中，如果您需要实际的类对象，您可以执行以下操作：

import sys
f = Foo.my_function
vars(sys.modules[f.__module__])[f.__qualname__.split('.')[0]]  # Gets Foo object

如果函数可能属于嵌套类，则需要按如下方式进行迭代：

f = Foo.Bar.my_function
vals = vars(sys.modules[f.__module__])
for attr in f.__qualname__.split('.')[:-1]:
    vals = vals[attr]
# vals is now the class Foo.Bar
于 2019-04-19T19:43:52.240 回答
2

蟒蛇 3

以非常简单的方式解决了它：

str(bar.foo_method).split(" ", 3)[-2]

这给

'FooClass.foo_method'

在点上拆分以分别获取类和函数名

于 2020-09-17T08:07:40.940 回答
1

我开始做一些类似的事情，基本上这个想法是检查基类中的方法是否已在子类中实现。结果是我最初这样做的方式我无法检测到中间类何时实际实现该方法。

实际上，我的解决方法很简单；设置方法属性并稍后测试它的存在。这是整个事情的简化：

class A():
    def method(self):
        pass
    method._orig = None # This attribute will be gone once the method is implemented

    def run_method(self, *args, **kwargs):
        if hasattr(self.method, '_orig'):
            raise Exception('method not implemented')
        self.method(*args, **kwargs)

class B(A):
    pass

class C(B):
    def method(self):
        pass

class D(C):
    pass

B().run_method() # ==> Raises Exception: method not implemented
C().run_method() # OK
D().run_method() # OK

更新：实际上是method()从run_method()（不是精神吗？）调用并将所有未修改的参数传递给方法。

PS：这个答案不直接回答问题。恕我直言，有两个原因想知道哪个类定义了一个方法；第一个是在调试代码中指出一个类（例如在异常处理中），第二个是确定该方法是否已被重新实现（其中方法是一个打算由程序员实现的存根）。这个答案以不同的方式解决了第二种情况。

于 2014-10-15T20:44:40.847 回答