Вложенные декораторы функций, которые работают с аргументами в python

Я пишу декоратор функции, который будет применять преобразование к первому аргументу функции. Он отлично работает, если я только украшаю свои функции один раз, но если я их украшу дважды, я получаю сообщение об ошибке. Ниже приведен код, демонстрирующий проблему, это упрощенная версия кода, над которым я работаю. Я исключил код, который делает преобразование, чтобы не отвлекать от проблемы.

from inspect import getargspec
from functools import wraps

def dec(id):
    def _dec(fn):
        @wraps(fn)
        def __dec(*args, **kwargs):
            if len(args):
                return fn(args[0], *args[1:], **kwargs)
            else:
                first_arg = getargspec(fn).args[0]
                new_kwargs = kwargs.copy()
                del new_kwargs[first_arg]
                return fn(kwargs[first_arg], **new_kwargs)
        return __dec
    return _dec

@dec(1)
def functionWithOneDecorator(a, b, c):
    print "functionWithOneDecorator(a = %s, b = %s, c = %s)" % (a, b, c)

@dec(1)
@dec(2)
def functionWithTwoDecorators(a, b, c):
    print "functionWithTwoDecorators(a = %s, b = %s, c = %s)" % (a, b, c)

functionWithOneDecorator(1, 2, 3)
functionWithOneDecorator(1, b=2, c=3)
functionWithOneDecorator(a=1, b=2, c=3)
functionWithOneDecorator(c=3, b=2, a=1)

functionWithTwoDecorators(1, 2, 3)
functionWithTwoDecorators(1, b=2, c=3)
functionWithTwoDecorators(a=1, b=2, c=3)
functionWithTwoDecorators(c=3, b=2, a=1)

Когда я запускаю приведенный выше код, я получаю следующий вывод:

functionWithOneDecorator(a = 1, b = 2, c = 3)
functionWithOneDecorator(a = 1, b = 2, c = 3)
functionWithOneDecorator(a = 1, b = 2, c = 3)
functionWithOneDecorator(a = 1, b = 2, c = 3)
functionWithTwoDecorators(a = 1, b = 2, c = 3)
functionWithTwoDecorators(a = 1, b = 2, c = 3)
IndexError: list index out of range

Это происходит потому, что, когда второй декоратор проверяет функцию, которую он украшает, находит имена аргументов и терпит неудачу, потому что он украшает декоратор и принимает только * args и ** kwargs.

Я могу подумать о проблемах, которые будут работать в коде выше, но все равно сломается, если функция будет украшена моим декоратором, а другая - третьей стороной. Есть ли общий способ исправить это? или есть лучший способ добиться того же результата?

Обновление: Благодаря @Hernan для указания декоратора модуля. Это точно решает эту проблему. Теперь мой код выглядит следующим образом:

from decorator import decorator

def dec(id):
    @decorator
    def _dec(fn, *args, **kwargs):
        return fn(args[0], *args[1:], **kwargs)
    return _dec

@dec(1)
def functionWithOneDecorator(a, b, c):
    print "functionWithOneDecorator(a = %s, b = %s, c = %s)" % (a, b, c)

@dec(1)
@dec(2)
def functionWithTwoDecorators(a, b, c):
    print "functionWithTwoDecorators(a = %s, b = %s, c = %s)" % (a, b, c)

functionWithOneDecorator(1, 2, 3)
functionWithOneDecorator(1, b=2, c=3)
functionWithOneDecorator(a=1, b=2, c=3)
functionWithOneDecorator(c=3, b=2, a=1)

functionWithTwoDecorators(1, 2, 3)
functionWithTwoDecorators(1, b=2, c=3)
functionWithTwoDecorators(a=1, b=2, c=3)
functionWithTwoDecorators(c=3, b=2, a=1)    

Гораздо чище, и он работает!

+10
источник поделиться
1 ответ

Проблема заключается в том, что подпись вашей украшенной функции не является сигнатурой (getargspec) от оригинала. Это действительно хорошо объяснено в помощи decorator module, который вы можете решить свою проблему. В принципе, вы должны использовать декораторы, сохраняющие подпись, так что второй декоратор видит ту же подпись, что и первая.

+5
источник

Посмотрите другие вопросы по меткам или Задайте вопрос