Python decorar funções antes de chamada
Pergunta
Eu tenho um decorador bastante complexo escrito por outra pessoa. O que eu quero fazer é chamar uma versão decorada do tempo função de um baseado em um descision ou ligue para a função original (não decorados) outra vez. Isso é possível?
Solução
Com:
decorator(original_function)()
Sem:
original_function()
Um decorador é apenas uma função que leva uma função como um argumento e retorna outro. A sintaxe @ é totalmente opcional. Talvez um peneirar alguns documentação poderia ajudar a esclarecer as coisas.
Outras dicas
def original_function():
pass
decorated_function= decorator(original_function)
if use_decorated:
decorated_function()
else:
original_function()
Decore apenas uma vez, e depois escolher qual versão chamada.
Aqui está a receita que eu vim acima com para o problema. I também necessário para manter as assinaturas o mesmo, então eu usei o módulo decorador mas você poderia re-jig para evitar isso. Basicamente, o truque foi adicionar um atributo para a função. A função 'original' é desacoplado por isso que você precisa para passar em um 'eu' como o primeiro parâmetro, então eu adicionei um pouco de código extra para verificar isso também.
# http://www.phyast.pitt.edu/~micheles/python/decorator-2.0.1.zip
from decorator import decorator, update_wrapper
class mustbe : pass
def wrapper ( interface_ ) :
print "inside hhh"
def call ( func, self, *args, **kwargs ) :
print "decorated"
print "calling %s.%s with args %s, %s" % (self, func.__name__, args, kwargs)
return interface_ ( self, *args, **kwargs )
def original ( instance , *args, **kwargs ) :
if not isinstance ( instance, mustbe ) :
raise TypeError, "Only use this decorator on children of mustbe"
return interface_ ( instance, *args, **kwargs )
call = decorator ( call, interface_ )
call.original = update_wrapper ( original, call )
return call
class CCC ( mustbe ):
var = "class var"
@wrapper
def foo ( self, param ) :
"""foo"""
print self.var, param
class SSS ( CCC ) :
@wrapper ( hidden_=True )
def bar ( self, a, b, c ) :
print a, b, c
if __name__ == "__main__" :
from inspect import getargspec
print ">>> i=CCC()"
i=CCC()
print ">>> i.var = 'parrot'"
i.var = 'parrot'
print ">>> i.foo.__doc__"
print i.foo.__doc__
print ">>> getargspec(i.foo)"
print getargspec(i.foo)
print ">>> i.foo(99)"
i.foo(99)
print ">>> i.foo.original.__doc__"
print i.foo.original.__doc__
print ">>> getargspec(i.foo.original)"
print getargspec(i.foo.original)
print ">>> i.foo.original(i,42)"
i.foo.original(i,42)
print ">>> j=SSS()"
j=SSS()
print ">>> j.bar(1,2,3)"
j.bar(1,2,3)