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()