如何推断出@staticmethod所属的类?

时间:2009-06-04 08:30:24

标签: python decorator static-methods inspect

我正在尝试实现infer_class函数,在给定方法的情况下,该函数会计算出该方法所属的类。

到目前为止,我有这样的事情:

import inspect

def infer_class(f):
    if inspect.ismethod(f):
        return f.im_self if f.im_class == type else f.im_class
    # elif ... what about staticmethod-s?
    else:
        raise TypeError("Can't infer the class of %r" % f)

它对@ staticmethod-s不起作用,因为我无法想出办法来实现这一点。

有什么建议吗?

以下是infer_class的实际操作:

>>> class Wolf(object):
...     @classmethod
...     def huff(cls, a, b, c):
...         pass
...     def snarl(self):
...         pass
...     @staticmethod
...     def puff(k,l, m):
...         pass
... 
>>> print infer_class(Wolf.huff)
<class '__main__.Wolf'>
>>> print infer_class(Wolf().huff)
<class '__main__.Wolf'>
>>> print infer_class(Wolf.snarl)
<class '__main__.Wolf'>
>>> print infer_class(Wolf().snarl)
<class '__main__.Wolf'>
>>> print infer_class(Wolf.puff)
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
  File "<stdin>", line 6, in infer_class
TypeError: Can't infer the class of <function puff at ...>

2 个答案:

答案 0 :(得分:3)

那是因为staticmethods真的不是方法。 staticmethod描述符按原样返回原始函数。没有办法获得访问该函数的类。但是无论如何都没有真正的理由使用static方法,总是使用classmethods。

我发现静态方法的唯一用途是将函数对象存储为类属性,而不是将它们转换为方法。

答案 1 :(得分:3)

我无法让自己实际推荐这个,但它似乎适用于简单的案例,至少:

import inspect

def crack_staticmethod(sm):
    """
    Returns (class, attribute name) for `sm` if `sm` is a
    @staticmethod.
    """
    mod = inspect.getmodule(sm)
    for classname in dir(mod):
        cls = getattr(mod, classname, None)
        if cls is not None:
            try:
                ca = inspect.classify_class_attrs(cls)
                for attribute in ca:
                    o = attribute.object
                    if isinstance(o, staticmethod) and getattr(cls, sm.__name__) == sm:
                        return (cls, sm.__name__)
            except AttributeError:
                pass