Pergunta

Eu estou tentando implementar função infer_class que, dado um método, descobre a classe à qual o método pertence.

Até agora eu tenho algo parecido com isto:

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)

Ele não funciona para @ staticmethod-S porque eu não era capaz de chegar a uma maneira de conseguir isso.

Todas as sugestões?

Aqui está infer_class em ação:

>>> 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 ...>
Foi útil?

Solução

Isso porque staticmethods realmente não são métodos. O descritor staticmethod retorna a função original como é. Não há nenhuma maneira de obter a classe através da qual a função foi acessado. Mas não há nenhuma razão real para staticmethods uso de métodos de qualquer maneira, utilize sempre classmethods.

O único uso que eu encontrei para staticmethods é objetos função de armazenamento de atributos de classe e não tê-los transformar-se em métodos.

Outras dicas

Eu tenho dificuldade trazendo-me a verdade recomendar isso, mas parece trabalho para casos simples, pelo menos:

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
Licenciado em: CC-BY-SA com atribuição
Não afiliado a StackOverflow
scroll top