Effective Python 读书笔记: 第42条: 用functools.wraps定义函数修饰器

# -*- encoding: utf-8 -*-

import functools

'''
第42条: 用functools.wraps定义函数修饰器

关键:
1 装饰其
作用: 装饰函数，能在原始函数执行前后添加另外地代码
问题: 打印被装饰器修饰的函数，返回的是内部的wrapper函数
解决方法: 用functools.wraps装饰器来装饰wrapper函数

2 装饰器样例
def trace(func):
    @functools.wraps(func)
    def wrapper(*args, **kwargs):
        result = func(*args, **kwargs)
        info = "%s(%r, %r) -> %r" % (
            func.__name__, args, kwargs, result
        )
        print info
        # 注意: 有返回值的必须要返回函数结果
        return result
    return wrapper
    
3 总结
装饰器本质上是以函数为参数，在运行原始函数的前后添加其他代码。
通过对装饰器内部的wrapper函数加上functools.wraps可以解决名称变化的问题

参考:
Effectiv Python 编写高质量Python代码的59个有效方法
'''

def trace(func):
    @functools.wraps(func)
    def wrapper(*args, **kwargs):
        result = func(*args, **kwargs)
        info = "%s(%r, %r) -> %r" % (
            func.__name__, args, kwargs, result
        )
        print info
        '''
        注意: 有返回值的必须要返回函数结果
        '''
        return result
    return wrapper


@trace
def fibonacci(n):
    if n in (0, 1):
        return n
    result = fibonacci(n - 2) + fibonacci(n - 1)
    return result

def process():
    result = fibonacci(3)
    print result
    print fibonacci


if __name__ == "__main__":
    process()