Как определить класс, к которому принадлежит @staticmethod? - PullRequest
Новая
       17

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

4 голосов
/ 04 июня 2009

Я пытаюсь реализовать функцию 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 ]

3 голосов
/ 08 июня 2009

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

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
3 голосов
/ 04 июня 2009

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

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

...