13

给定一个函数：

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 上提出这个问题时，我很长关于框架为什么糟糕的演讲，这不是我想要的；我必须在这里解决问题。另外，我也只是对答案感兴趣）

4

3 回答 3

12

Michele Simionato 的装饰器模块有一个名为 decorator 的装饰器，它保留函数 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)
于 2010-09-16T18:09:45.473 回答
2

有装饰器模块：

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))
于 2010-09-16T18:11:04.333 回答
0

是否functools.update_wrapper()和/或functools.wraps()足够好？

于 2010-09-16T18:09:10.733 回答