Как я могу программно изменить argspec функции в декораторе python?
Для функции:
def func(f1, kw='default'):
pass
bare_argspec = inspect.getargspec(func)
@decorator
def func2(f1, kw='default'):
pass
decorated_argspec = inspect.getargspec(func2)
Как создать декоратор таким образом, чтобы bare_argspec == decorated_argspec
?
(Что касается того, почему структура, которая вызывает украшенную функцию, проверяет argspec, чтобы выбрать, что передать, поэтому декоратор должен сохранить тот же argspec, чтобы играть хорошо. Когда я задал этот вопрос на #python, я получил длинную речь о том, почему структура отстойная, чего я не ищу, я должен решить проблему здесь. Кроме того, меня тоже интересует ответ тоже)
Ответы
Ответ 1
Michele Simionato decorator module имеет декоратор, называемый декоратором, который сохраняет функцию argspecs.
import inspect
import decorator
def func(f1, kw='default'):
pass
bare_argspec = inspect.getargspec(func)
print(bare_argspec)
# ArgSpec(args=['f1', 'kw'], varargs=None, keywords=None, defaults=('default',))
@decorator.decorator
def mydecorator(func,*args,**kw):
result=func(*args,**kw)
return result
@mydecorator
def func2(f1, kw='default'):
pass
decorated_argspec = inspect.getargspec(func2)
print(decorated_argspec)
# ArgSpec(args=['f1', 'kw'], varargs=None, keywords=None, defaults=('default',))
assert(bare_argspec==decorated_argspec)
Ответ 2
Здесь decorator модуль:
from decorator import decorator
@decorator
def trace(func, *args, **kw):
print 'calling', func, 'with', args, kw
return func(*args, **kw)
Это делает trace
декоратором с теми же argspecs, что и украшенная функция. Пример:
>>> @trace
... def f(x, y=1, z=2, *args, **kw):
... pass
>>> f(0, 3)
calling f with (0, 3, 2), {}
>>> from inspect import getargspec
>>> print getargspec(f)
ArgSpec(args=['x', 'y', 'z'], varargs='args', keywords='kw', defaults=(1, 2))
Ответ 3
Достаточно ли functools.update_wrapper()
и/или functools.wraps()
?