Как передать дополнительные аргументы декоратору Python?

У меня есть декоратор, как показано ниже.

<code>def myDecorator(test_func):
    return callSomeWrapper(test_func)
def callSomeWrapper(test_func):
    return test_func
@myDecorator
def someFunc():
    print 'hello'
</code>

Я хочу улучшить этот декоратор, чтобы принять другой аргумент, как показано ниже

<code>def myDecorator(test_func,logIt):
    if logIt:
        print "Calling Function: " + test_func.__name__
    return callSomeWrapper(test_func)
@myDecorator(False)
def someFunc():
    print 'Hello'
</code>

Но этот код дает ошибку,

TypeError: myDecorator() takes exactly 2 arguments (1 given)

Почему функция не передается автоматически? Как явно передать функцию функции декоратора?

 Kit Ho16 апр. 2012 г., 16:43
balki: пожалуйста, избегайте использования логического значения в качестве аргумента, это не gd-подход и уменьшите читаемость кода
 Rob Bednark05 февр. 2016 г., 01:12
@KitHo - что такое "gd"? Это "хорошо"?
 AKX16 апр. 2012 г., 16:50
@KitHo - это логический флаг, поэтому использование логического значения является правильным подходом.

Ответы на вопрос(4)

Просто чтобы предоставить другую точку зрения: синтаксис

@expr
def func(...): #stuff

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

def func(...): #stuff
func = expr(func)

Особенно,expr может быть всем, что угодно, до тех пор, пока оно оценивается как вызываемое. Вparticular конкретный,expr может быть фабрикой декораторов: вы даете ей некоторые параметры, и она дает вам декоратор. Так что, возможно, лучший способ понять вашу ситуацию

dec = decorator_factory(*args)
@dec
def func(...):

который затем может быть сокращен до

@decorator_factory(*args)
def func(...):

Конечно, так как этоlooks лайкdecorator_factory это декоратор, люди склонны называть это, чтобы отразить это. Что может сбивать с толку, когда вы пытаетесь следовать уровням косвенности.

Просто хочу добавить несколько полезных трюков, которые позволят сделать аргументы декоратора необязательными. Это также позволит повторно использовать декоратор и уменьшить вложенность

import functools

def myDecorator(test_func=None,logIt=None):
    if not test_func:
        return functools.partial(myDecorator, logIt=logIt)
    @functools.wraps(test_func)
    def f(*args, **kwargs):
        if logIt==1:
            print 'Logging level 1 for {}'.format(test_func.__name__)
        if logIt==2:
            print 'Logging level 2 for {}'.format(test_func.__name__)
        return test_func(*args, **kwargs)
    return f

#new decorator 
myDecorator_2 = myDecorator(logIt=2)

@myDecorator(logIt=2)
def pow2(i):
    return i**2

@myDecorator
def pow3(i):
    return i**3

@myDecorator_2
def pow4(i):
    return i**4

print pow2(2)
print pow3(2)
print pow4(2)
Решение Вопроса

Поскольку вы вызываете декоратор как функцию, она должна вернуть другую функцию, которая является фактическим декоратором:

def my_decorator(param):
    def actual_decorator(func):
        print("Decorating function {}, with parameter {}".format(func.__name__, param))
        return function_wrapper(func)  # assume we defined a wrapper somewhere
    return actual_decorator

Внешняя функция получит любые аргументы, которые вы передадите явно, и должна вернуть внутреннюю функцию. Внутренней функции будет передана функция для декорирования и возврата измененной функции.

Обычно вы хотите, чтобы декоратор изменил поведение функции, поместив его в функцию-оболочку. Вот пример, который при необходимости добавляет запись в журнал при вызове функции:

def log_decorator(log_enabled):
    def actual_decorator(func):
        @functools.wraps(func)
        def wrapper(*args, **kwargs):
            if log_enabled:
                print("Calling Function: " + func.__name__)
            return func(*args, **kwargs)
        return wrapper
    return actual_decorator

functools.wraps call копирует такие вещи, как имя и строку документации, в функцию-оболочку, чтобы сделать ее более похожей на оригинальную функцию.

Пример использования:

>>> @log_decorator(True)
... def f(x):
...     return x+1
...
>>> f(4)
Calling Function: f
5
 16 апр. 2012 г., 16:51
И используяfunctools.wraps Рекомендуется - он сохраняет оригинальное имя, строку документации и т. д. обернутой функции.
 16 апр. 2012 г., 16:58
@AKX: Спасибо, я добавил это во второй пример.
 16 апр. 2012 г., 19:03
@ Балки: Да, это правильно. Что смущает то, что многие люди также будут вызывать внешнюю функцию (myDecorator здесь) декоратор. Это удобно для пользователя декоратора, но может сбивать с толку, когда вы пытаетесь написать его.
 balki16 апр. 2012 г., 18:46
Так что в основном декоратор всегда принимает только один аргумент, который является функцией. Но декоратор может быть возвращаемым значением функции, которая может принимать аргументы. Это правильно?

Просто еще один способ сделать декораторы. Я считаю, что так легче всего обернуть мою голову.

import functools

class NiceDecorator:
    def __init__(self, param_foo='a', param_bar='b'):
        self.param_foo = param_foo
        self.param_bar = param_bar

    def __call__(self, func):
        @functools.wraps(func)
        def my_logic(*args, **kwargs):
            # whatever logic your decorator is supposed to implement goes in here
            print('pre action baz')
            print(self.param_bar)
            # including the call to the decorated function (if you want to do that)
            result = func(*args, **kwargs)
            print('post action beep')
            return result

        return my_logic

# usage example from here on
@NiceDecorator(param_bar='baaar')
def example():
    print('example yay')


example()

Ваш ответ на вопрос