Python украшающие функции перед вызовом - PullRequest
Новая
       8

Python украшающие функции перед вызовом

9 голосов
/ 12 ноября 2008

У меня довольно сложный декоратор, написанный кем-то другим. То, что я хочу сделать, это вызвать декорированную версию функции один раз на основе решения или вызвать оригинальную функцию (не декорированную) в другой раз. Возможно ли это?

Ответы [ 3 ]

25 голосов
/ 12 ноября 2008

С: 

decorator(original_function)()

Без: 

original_function()

Декоратор - это просто функция, которая принимает функцию в качестве аргумента и возвращает другую. Синтаксис @ не является обязательным. Возможно, просмотр некоторой документации может помочь прояснить ситуацию.

2 голосов
/ 12 ноября 2008 
def original_function():
    pass

decorated_function= decorator(original_function)

if use_decorated:
    decorated_function()
else:
    original_function()

Украсить только один раз, а затем выбрать версию для вызова.

1 голос
/ 01 мая 2009

Вот рецепт, который я придумал для решения проблемы. Мне также нужно было сохранить сигнатуры одинаковыми, поэтому я использовал модуль декоратора, но вы можете перенастроить его, чтобы избежать этого. По сути, хитрость заключалась в том, чтобы добавить атрибут к функции. Функция «original» не связана, поэтому вам нужно передать «self» в качестве первого параметра, поэтому я добавил дополнительный код, чтобы проверить это тоже. 

# 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)
...