13


0

Pythonデコレータの関数のargspecをプログラムで変更するにはどうすればよいですか?

与えられた機能:

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で提起したとき、私はフレームワークがなぜ悪いのかについて長いスピーチを得ましたが、それは私が探しているものではありません。ここで問題を解決する必要があります。 また、私は答えにも興味があります)

3 Answer


12


Michele Simionatoのhttp://pypi.python.org/pypi/decorator[decorator module]には、関数argspecsを保存するdecoratorと呼ばれるデコレーターがあります。

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)

これにより、装飾された関数と同じargspecsを持つ `trace`がデコレータになります。 例:

>>> @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))


0


`functools.update_wrapper()`および/または `functools.wraps()`で十分ですか?