保留签名的Python decorator(闭包版本)

wrapt 模块。

请记住 decorator的接口与标准python decorator不同。我强烈建议你读这本书 wrapt documentation . 不管怎样,这里有一个基本的decorator实现 wrapt.decorator :

import wrapt

def args_as_ints(f):
    time.sleep(1) # hard at work

    @wrapt.decorator
    def g(f, instance, *args, **kwargs):
        args = [int(x) for x in args]
        kwargs = dict((k, int(v)) for k, v in kwargs.items())
        return f(*args, **kwargs)

    return g(f)  # apply the decorator to the function

instance

但是,它确实保留了修饰函数的签名:

Help on function funny_function in module __main__:

funny_function(x, y, z=3)
    Computes x*y + 2*z

https://decorator.readthedocs.io/en/latest/tests.documentation.html#dealing-with-third-party-decorators

def args_as_ints(f):

    time.sleep(1)

    def g(*args, **kwargs):
        args = [int(x) for x in args]
        kwargs = dict((k, int(v)) for k, v in kwargs.items())
        return f(*args, **kwargs)

    return decorator.FunctionMaker.create(
        f, 'return decfunc(%(signature)s)',
        dict(decfunc=g, __wrapped__=f))