Como posso inferir a classe à qual uma @staticmethod pertence?
-
09-09-2019 - |
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 ...>
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