Как мне определить класс, к которому принадлежит @staticmethod?

Question

Я пытаюсь реализовать 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 ...>

Answer 1

Это потому, что staticmethod на самом деле не являются методами.Дескриптор staticmethod возвращает исходную функцию как есть.Нет никакого способа получить класс, через который был получен доступ к функции.Но в любом случае нет реальной причины использовать staticmethod для методов, всегда используйте classmethod.

Единственное применение, которое я нашел для staticmethods, - это хранить объекты функций как атрибуты класса, а не превращать их в методы.

Answer 2

Мне трудно заставить себя на самом деле рекомендовать это, но, похоже, это работает, по крайней мере, для простых случаев:

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