Как я могу использовать именованные аргументы в декораторе? - PullRequest
Новая
       18

Как я могу использовать именованные аргументы в декораторе?

4 голосов
/ 09 марта 2009

Если у меня есть следующая функция: 


def intercept(func):
  # do something here

@intercept(arg1=20)
def whatever(arg1,arg2):
  # do something here

Я бы хотел, чтобы перехват запускался только тогда, когда arg1 равен 20. Я хотел бы иметь возможность передавать именованные параметры в функцию. Как я мог сделать это?

Вот небольшой пример кода: 



def intercept(func):
    def intercepting_func(*args,**kargs):
        print "whatever"
        return func(*args,**kargs)
    return intercepting_func

@intercept(a="g")
def test(a,b):
    print "test with %s %s" %(a,b)

test("g","d")

Это вызывает следующее исключение TypeError: intercept () получил неожиданный аргумент ключевого слова 'a'

Ответы [ 3 ]

15 голосов
/ 10 марта 2009

Помните, что 

@foo
def bar():
    pass

эквивалентно: 

def bar():
    pass
bar = foo(bar)

так что если вы делаете: 

@foo(x=3)
def bar():
    pass

это эквивалентно: 

def bar():
    pass
bar = foo(x=3)(bar)

поэтому ваш декоратор должен выглядеть примерно так: 

def foo(x=1):
    def wrap(f):
        def f_foo(*args, **kw):
            # do something to f
            return f(*args, **kw)
        return f_foo
    return wrap

Другими словами, def wrap(f) действительно является декоратором, а foo(x=3) является вызовом функции, которая возвращает декоратор.

11 голосов
/ 09 марта 2009 
from functools import wraps

def intercept(target,**trigger):
    def decorator(func):
        names = getattr(func,'_names',None)
        if names is None:
            code = func.func_code
            names = code.co_varnames[:code.co_argcount]
        @wraps(func)
        def decorated(*args,**kwargs):
            all_args = kwargs.copy()
            for n,v in zip(names,args):
                all_args[n] = v
            for k,v in trigger.iteritems():
                if k in all_args and all_args[k] != v:
                    break
            else:
                return target(all_args)
            return func(*args,**kwargs)
        decorated._names = names
        return decorated
    return decorator

Пример: 

def interceptor1(kwargs):
    print 'Intercepted by #1!'

def interceptor2(kwargs):
    print 'Intercepted by #2!'

def interceptor3(kwargs):
    print 'Intercepted by #3!'

@intercept(interceptor1,arg1=20,arg2=5) # if arg1 == 20 and arg2 == 5
@intercept(interceptor2,arg1=20)        # elif arg1 == 20
@intercept(interceptor3,arg2=5)         # elif arg2 == 5
def foo(arg1,arg2):
    return arg1+arg2

>>> foo(3,4)
7
>>> foo(20,4)
Intercepted by #2!
>>> foo(3,5)
Intercepted by #3!
>>> foo(20,5)
Intercepted by #1!
>>>

functools.wraps делает то, что делает "простой декоратор" в вики; Обновления __doc__, __name__ и другие атрибуты декоратора.

3 голосов
/ 09 марта 2009

Вы можете сделать это, используя * args и ** kwargs в декораторе: 

def intercept(func, *dargs, **dkwargs):
    def intercepting_func(*args, **kwargs):
        if (<some condition on dargs, dkwargs, args and kwargs>):
            print 'I intercepted you.'
        return func(*args, **kwargs)
    return intercepting_func

Вам решать, как передать аргументы для управления поведением декоратора.

Чтобы сделать это максимально прозрачным для конечного пользователя, вы можете использовать «простой декоратор» на Python wiki или «декоратор-декоратор» Микеле Симионато *

...