Funciones de decoración de Python antes de llamar
Pregunta
Tengo un decorador bastante complejo escrito por otra persona. Lo que quiero hacer es llamar a una versión decorada de la función una vez basada en una decisión o llamar a la función original (no decorada) otra vez. ¿Es esto posible?
Solución
Con:
decorator(original_function)()
Sin:
original_function()
Un decorador es solo una función que toma una función como argumento y devuelve otra. La sintaxis @ es totalmente opcional. Quizás echar un vistazo a la documentación podría ayudar a aclarar las cosas.
Otros consejos
def original_function():
pass
decorated_function= decorator(original_function)
if use_decorated:
decorated_function()
else:
original_function()
Decora solo una vez y luego elige a qué versión llamar.
Aquí está la receta que se me ocurrió para el problema. También necesitaba mantener las firmas igual, así que utilicé el módulo decorador, pero podría volver a usar la plantilla para evitar eso. Básicamente, el truco consistía en agregar un atributo a la función. La función 'original' no está vinculada, por lo que debe pasar un 'self' como primer parámetro, por lo que agregué un código adicional para verificar eso también.
# 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)