python functools.wraps 实例

#!/usr/bin/env python
# coding:utf-8


def logged(func):
    def with_logging(*args, **kwargs):
        ‘‘‘i am wraps‘s doc‘‘‘
        print func.__name__ + " was called"
        return func(*args, **kwargs)

    return with_logging


@logged
def f(x):
    """i am original doc"""
    return x + x * x

print f.__name__    # 期望的是原本的名字f，结果换成了装饰器里的新函数的名字with_logging
print f.__doc__    # 期望的是原本的注释，加了个装饰器成了新加装饰器里的注释了

from functools import wraps


def logged(func):
    @wraps(func)
    def with_logging(*args, **kwargs):
        ‘‘‘i am wraps‘s doc‘‘‘
        print func.__name__ + " was called"
        return func(*args, **kwargs)

    return with_logging


@logged
def f(x):
    """i am original doc"""
    return x + x * x


print f.__name__  # prints ‘f‘
print f.__doc__  # prints ‘i am original doc‘