Python中為何建議在decorator函數中使用functools.wraps

以最常見的time_it裝飾器函數為例, 如下代碼中:

  • 第一個time_it函數沒有使用functools.wraps, 功能上這個裝飾器并沒有什么問題;

  • 第二個better_time_it在wrapper函數面前又加了一個@functools.wraps(func)函數, 其他代碼一致;

import time
import functools


def time_it(func):
    def wrapper(*args, **kwargs):
        """This is docs for wrapper"""
        start = time.time()
        result = func(*args, **kwargs)
        end = time.time()
        print('Running "%s" in %.3f seconds' % (func.__name__, (end - start)))
        return result

    return wrapper


def better_time_it(func):
    @functools.wraps(func)  # Pay attention here!
    def wrapper(*args, **kwargs):
        """This is docs for wrapper"""
        start = time.time()
        result = func(*args, **kwargs)
        end = time.time()
        print('Running "%s" in %.3f seconds' % (func.__name__, (end - start)))
        return result

    return wrapper

我們的測試代碼如下:

@time_it
def foo():
    """This is docs foo function"""
    print('Foo!')

@better_time_it
def yeah():
    """This is docs yeah function"""
    print('Yeah!')

if __name__ == '__main__':
    help(foo)
    help(yeah)

這里我用help函數查看function的doc strings,得到如下結果:

Help on function wrapper in module __main__:

wrapper(*args, **kwargs)
    This is docs for wrapper

Help on function yeah in module __main__:

yeah()
    This is docs yeah function
  • 可以看到使用time_it的foo函數的docs變成了wrapper函數的docs

  • 而使用better_time_it的yeah函數的docs才是我們期望得到的實際docs;
    不止help函數的結果會有變化,因為使用裝飾器實際上返回的并不是原函數本身,所以原函數相關的metadata信息已經變化了,如果不使用functools.wraps函數就會導致此類副作用,比如序列化和debugger時也會使結果混亂,所以養成良好的習慣,自定義decorator的時應該盡量使用functools.wraps

關于functools.wraps()函數的官方文檔中寫道:

The main intended use for this function is in decorator functions which wrap the decorated function and return the wrapper. If the wrapper function is not updated, the metadata of the returned function will reflect the wrapper definition rather than the original function definition, which is typically less than helpful.

最后編輯于
?著作權歸作者所有,轉載或內容合作請聯系作者
平臺聲明:文章內容(如有圖片或視頻亦包括在內)由作者上傳并發布,文章內容僅代表作者本人觀點,簡書系信息發布平臺,僅提供信息存儲服務。
  • 序言:七十年代末,一起剝皮案震驚了整個濱河市,隨后出現的幾起案子,更是在濱河造成了極大的恐慌,老刑警劉巖,帶你破解...
    沈念sama閱讀 228,646評論 6 533
  • 序言:濱河連續發生了三起死亡事件,死亡現場離奇詭異,居然都是意外死亡,警方通過查閱死者的電腦和手機,發現死者居然都...
    沈念sama閱讀 98,595評論 3 418
  • 文/潘曉璐 我一進店門,熙熙樓的掌柜王于貴愁眉苦臉地迎上來,“玉大人,你說我怎么就攤上這事。” “怎么了?”我有些...
    開封第一講書人閱讀 176,560評論 0 376
  • 文/不壞的土叔 我叫張陵,是天一觀的道長。 經常有香客問我,道長,這世上最難降的妖魔是什么? 我笑而不...
    開封第一講書人閱讀 63,035評論 1 314
  • 正文 為了忘掉前任,我火速辦了婚禮,結果婚禮上,老公的妹妹穿的比我還像新娘。我一直安慰自己,他們只是感情好,可當我...
    茶點故事閱讀 71,814評論 6 410
  • 文/花漫 我一把揭開白布。 她就那樣靜靜地躺著,像睡著了一般。 火紅的嫁衣襯著肌膚如雪。 梳的紋絲不亂的頭發上,一...
    開封第一講書人閱讀 55,224評論 1 324
  • 那天,我揣著相機與錄音,去河邊找鬼。 笑死,一個胖子當著我的面吹牛,可吹牛的內容都是我干的。 我是一名探鬼主播,決...
    沈念sama閱讀 43,301評論 3 442
  • 文/蒼蘭香墨 我猛地睜開眼,長吁一口氣:“原來是場噩夢啊……” “哼!你這毒婦竟也來了?” 一聲冷哼從身側響起,我...
    開封第一講書人閱讀 42,444評論 0 288
  • 序言:老撾萬榮一對情侶失蹤,失蹤者是張志新(化名)和其女友劉穎,沒想到半個月后,有當地人在樹林里發現了一具尸體,經...
    沈念sama閱讀 48,988評論 1 335
  • 正文 獨居荒郊野嶺守林人離奇死亡,尸身上長有42處帶血的膿包…… 初始之章·張勛 以下內容為張勛視角 年9月15日...
    茶點故事閱讀 40,804評論 3 355
  • 正文 我和宋清朗相戀三年,在試婚紗的時候發現自己被綠了。 大學時的朋友給我發了我未婚夫和他白月光在一起吃飯的照片。...
    茶點故事閱讀 42,998評論 1 370
  • 序言:一個原本活蹦亂跳的男人離奇死亡,死狀恐怖,靈堂內的尸體忽然破棺而出,到底是詐尸還是另有隱情,我是刑警寧澤,帶...
    沈念sama閱讀 38,544評論 5 360
  • 正文 年R本政府宣布,位于F島的核電站,受9級特大地震影響,放射性物質發生泄漏。R本人自食惡果不足惜,卻給世界環境...
    茶點故事閱讀 44,237評論 3 347
  • 文/蒙蒙 一、第九天 我趴在偏房一處隱蔽的房頂上張望。 院中可真熱鬧,春花似錦、人聲如沸。這莊子的主人今日做“春日...
    開封第一講書人閱讀 34,665評論 0 26
  • 文/蒼蘭香墨 我抬頭看了看天上的太陽。三九已至,卻和暖如春,著一層夾襖步出監牢的瞬間,已是汗流浹背。 一陣腳步聲響...
    開封第一講書人閱讀 35,927評論 1 287
  • 我被黑心中介騙來泰國打工, 沒想到剛下飛機就差點兒被人妖公主榨干…… 1. 我叫王不留,地道東北人。 一個月前我還...
    沈念sama閱讀 51,706評論 3 393
  • 正文 我出身青樓,卻偏偏與公主長得像,于是被迫代替她去往敵國和親。 傳聞我的和親對象是個殘疾皇子,可洞房花燭夜當晚...
    茶點故事閱讀 47,993評論 2 374

推薦閱讀更多精彩內容

  • 包(lib)、模塊(module) 在Python中,存在包和模塊兩個常見概念。 模塊:編寫Python代碼的py...
    清清子衿木子水心閱讀 3,825評論 0 27
  • 直接上正文 函數是Python內建支持的一種封裝,我們通過把大段代碼拆成函數,通過一層一層的函數調用,就可以把復雜...
    OzanShareing閱讀 460評論 0 0
  • 半夜+一個清晨看完這本書,感覺就是一個字 堵 。小人物的悲喜串成了整個時代,他們的命運如同巨大機器下的齒輪,千...
    Jasmyn1閱讀 612評論 3 3
  • 最近找了一份實習,工作的性質偏銷售,跟我的專業并不相關,但是公司是一家世界五百強,總體來說還可以。 但是身邊的同事...
    北海道的北閱讀 476評論 0 0
  • 第三章 查案2 按第五遍門鈴的時候,門突然被打開了,出現在門內的是安瑜。劉劍看到安瑜明顯愣了一下,接著變得緊張起來...
    白寧紀事閱讀 251評論 0 2