带装饰函数参数的python decorator

Question

当我用@包装函数时，如何使包装函数的外观和感觉完全像包装函数一样？ 特别是help(function) 。

一些代码：

>>> def wraps(f):
    def call(*args, **kw):
        print('in', f, args, kw) # example code. I need to transfer the arguments to another process and pickle them.
        return f(*args, **kw)
    return call

>>> def g():pass

>>> @wraps
def f(a, b = 1, g = g, *args, **kw):
    pass

>>> help(f)
Help on function call in module __main__:

call(*args, **kw) # this line bothers me. It should look different, look below

>>> def f(a, b = 1, g = g, *args, **kw):
    pass

>>> help(f)
Help on function f in module __main__:

f(a, b=1, g=<function g at 0x02EE14B0>, *args, **kw) # help(f) should look like this.

动机：当我输入帮助窗口，当我输入f( * plopp *我看到(a, b = 1, g = g, *args, **kw)时，看到参数也会很高兴。（在此IDLE Python Shell中的case

我看了一下inspect模块，它帮助我进行了很好的格式化。 问题仍然存在：我如何用参数做到这一点..

像def f(d = {}):一样传递的默认可变参数def f(d = {}):不需要工作，因为我将参数传递给另一个进程，无论如何身份都会丢失。

Answer 1

functools.wraps可用于复制函数的名称和docstring。 从头开始复制原始功能签名要困难得多。

但是，如果您使用第三方装饰器模块

import decorator


@decorator.decorator
def wraps(f):
    def call(*args, **kw):
        print('in', f, args, kw) 
        return f(*args, **kw)
    return call


def g():pass

@wraps
def f(a, b = 1, g = g, *args, **kw):
    pass

help(f)

产量

Help on function f in module __main__:

f(a, b=1, g=<function g>, *args, **kw)

Answer 2

使用functools.wraps ：

from functools import wraps

def wrapper(f):
    @wraps(f)
    def call(*args, **kw):
        print('in', f, args, kw)
        return f(*args, **kw)
    return call

@wrapper
def f(a, b = 1, g = g, *args, **kw):
    pass

help(f)
Help on function f in module __main__:

f(a, b=1, g=<function g at 0x7f5ad14a6048>, *args, **kw)

这会保留包装函数的__name__和__doc__属性。

Answer 3

我认为其他答案更可取，但如果由于某种原因您不想使用外部模块，您可以随时更改装饰器：

def wraps(f):
  def call(*args, **kw):
    print('in', f, args, kw)
    return f(*args, **kw)
call.__name__ = f.__name__
call.__doc__ = f.__doc__
return call