Python裝飾器神器@wraps(func)

遇到問題:

因?yàn)楸救酥笆怯玫膗nittest框架寫的自動化腳本挪挤,已經(jīng)預(yù)先寫好了log日志模塊的裝飾器。后需將unittest替換成pytest蛾派,用到了pytest中功能強(qiáng)大的fixture瓤荔,但是讀取coftest.py中的fixture后,pytest的測試用例中填寫fixture的方法名骨坑,執(zhí)行,卻發(fā)現(xiàn)自己寫log裝飾器報missing 1 required positional argument

  • 同時使用fixture和裝飾器代碼:


    image.png
  • 報錯:


    image.png

解決問題途徑:

發(fā)現(xiàn)同樣的情況下柬采,加上@allure.step("")裝飾器卻不會報該錯欢唾,查看allure的源碼

解決問題:

allure源碼中,發(fā)現(xiàn)裝飾器中加了一個@wraps(func)粉捻,解決上述問題

以下介紹wraps相關(guān):

Python中被裝飾后的函數(shù)礁遣,函數(shù)名等函數(shù)屬性會發(fā)生改變(相當(dāng)于另一個函數(shù)了),所以杀迹,Python的functools包中提供了一個叫wraps的裝飾器來解決該問題亡脸。它能使其保留原有函數(shù)的結(jié)構(gòu)押搪。

  • 不加@wraps()的裝飾器:
import time


class MyLog(object):

    def __call__(self, func):
        def my_log(*args, **kwargs):
            """這里是my_log的docstring"""
            start = time.time()
            res = func(*args, **kwargs)
            end = time.time()
            print('|運(yùn)行時長:{:.4f}'.format(end - start))
            return res
        return my_log


@MyLog()
def is_test():
    """這里是is_test的docstring"""
    print("一個測試方法")


print("__name__: ", is_test.__name__)
print("__doc__:  ", is_test.__doc__)

運(yùn)行結(jié)果:


image.png

看運(yùn)行結(jié)果树酪,會發(fā)現(xiàn),被裝飾器裝飾后的函數(shù)大州,name和docstring都發(fā)生了變化

  • 加了@wraps()的裝飾器:
import time
from functools import wraps


class MyLog(object):

    def __call__(self, func):
        @wraps(func)
        def my_log(*args, **kwargs):
            """這里是my_log的docstring"""
            start = time.time()
            res = func(*args, **kwargs)
            end = time.time()
            print('|運(yùn)行時長:{:.4f}'.format(end - start))
            return res
        return my_log


@MyLog()
def is_test():
    """這里是is_test的docstring"""
    print("一個測試方法")


print("__name__: ", is_test.__name__)
print("__doc__:  ", is_test.__doc__)

運(yùn)行結(jié)果:


image.png

看運(yùn)行結(jié)果续语,會發(fā)現(xiàn),裝飾器加了@wraps后厦画,被裝飾器裝飾后的函數(shù)疮茄,還是保持原有的狀態(tài)

  • 裝飾器的作用: 在不改變原有功能代碼的基礎(chǔ)上,添加額外的功能。
  • @wraps(func)的作用: 不改變使用裝飾器原有函數(shù)的結(jié)構(gòu)(如name, doc)
最后編輯于
?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請聯(lián)系作者
  • 序言:七十年代末根暑,一起剝皮案震驚了整個濱河市力试,隨后出現(xiàn)的幾起案子,更是在濱河造成了極大的恐慌排嫌,老刑警劉巖畸裳,帶你破解...
    沈念sama閱讀 218,451評論 6 506
  • 序言:濱河連續(xù)發(fā)生了三起死亡事件,死亡現(xiàn)場離奇詭異淳地,居然都是意外死亡怖糊,警方通過查閱死者的電腦和手機(jī)帅容,發(fā)現(xiàn)死者居然都...
    沈念sama閱讀 93,172評論 3 394
  • 文/潘曉璐 我一進(jìn)店門,熙熙樓的掌柜王于貴愁眉苦臉地迎上來伍伤,“玉大人并徘,你說我怎么就攤上這事∪呕辏” “怎么了麦乞?”我有些...
    開封第一講書人閱讀 164,782評論 0 354
  • 文/不壞的土叔 我叫張陵,是天一觀的道長劝评。 經(jīng)常有香客問我路幸,道長,這世上最難降的妖魔是什么付翁? 我笑而不...
    開封第一講書人閱讀 58,709評論 1 294
  • 正文 為了忘掉前任简肴,我火速辦了婚禮,結(jié)果婚禮上百侧,老公的妹妹穿的比我還像新娘砰识。我一直安慰自己,他們只是感情好佣渴,可當(dāng)我...
    茶點(diǎn)故事閱讀 67,733評論 6 392
  • 文/花漫 我一把揭開白布辫狼。 她就那樣靜靜地躺著,像睡著了一般辛润。 火紅的嫁衣襯著肌膚如雪膨处。 梳的紋絲不亂的頭發(fā)上,一...
    開封第一講書人閱讀 51,578評論 1 305
  • 那天砂竖,我揣著相機(jī)與錄音真椿,去河邊找鬼。 笑死乎澄,一個胖子當(dāng)著我的面吹牛突硝,可吹牛的內(nèi)容都是我干的。 我是一名探鬼主播置济,決...
    沈念sama閱讀 40,320評論 3 418
  • 文/蒼蘭香墨 我猛地睜開眼解恰,長吁一口氣:“原來是場噩夢啊……” “哼!你這毒婦竟也來了浙于?” 一聲冷哼從身側(cè)響起护盈,我...
    開封第一講書人閱讀 39,241評論 0 276
  • 序言:老撾萬榮一對情侶失蹤,失蹤者是張志新(化名)和其女友劉穎羞酗,沒想到半個月后腐宋,有當(dāng)?shù)厝嗽跇淞掷锇l(fā)現(xiàn)了一具尸體,經(jīng)...
    沈念sama閱讀 45,686評論 1 314
  • 正文 獨(dú)居荒郊野嶺守林人離奇死亡,尸身上長有42處帶血的膿包…… 初始之章·張勛 以下內(nèi)容為張勛視角 年9月15日...
    茶點(diǎn)故事閱讀 37,878評論 3 336
  • 正文 我和宋清朗相戀三年脏款,在試婚紗的時候發(fā)現(xiàn)自己被綠了围苫。 大學(xué)時的朋友給我發(fā)了我未婚夫和他白月光在一起吃飯的照片。...
    茶點(diǎn)故事閱讀 39,992評論 1 348
  • 序言:一個原本活蹦亂跳的男人離奇死亡撤师,死狀恐怖剂府,靈堂內(nèi)的尸體忽然破棺而出,到底是詐尸還是另有隱情剃盾,我是刑警寧澤腺占,帶...
    沈念sama閱讀 35,715評論 5 346
  • 正文 年R本政府宣布,位于F島的核電站痒谴,受9級特大地震影響衰伯,放射性物質(zhì)發(fā)生泄漏。R本人自食惡果不足惜积蔚,卻給世界環(huán)境...
    茶點(diǎn)故事閱讀 41,336評論 3 330
  • 文/蒙蒙 一意鲸、第九天 我趴在偏房一處隱蔽的房頂上張望。 院中可真熱鬧尽爆,春花似錦怎顾、人聲如沸。這莊子的主人今日做“春日...
    開封第一講書人閱讀 31,912評論 0 22
  • 文/蒼蘭香墨 我抬頭看了看天上的太陽。三九已至幅狮,卻和暖如春募强,著一層夾襖步出監(jiān)牢的瞬間,已是汗流浹背崇摄。 一陣腳步聲響...
    開封第一講書人閱讀 33,040評論 1 270
  • 我被黑心中介騙來泰國打工擎值, 沒想到剛下飛機(jī)就差點(diǎn)兒被人妖公主榨干…… 1. 我叫王不留,地道東北人配猫。 一個月前我還...
    沈念sama閱讀 48,173評論 3 370
  • 正文 我出身青樓幅恋,卻偏偏與公主長得像,于是被迫代替她去往敵國和親泵肄。 傳聞我的和親對象是個殘疾皇子,可洞房花燭夜當(dāng)晚...
    茶點(diǎn)故事閱讀 44,947評論 2 355

推薦閱讀更多精彩內(nèi)容