如何實(shí)現(xiàn)一個(gè)python函數(shù)裝飾器(Decorator)
裝飾器本質(zhì)上是一個(gè) Python 函數(shù)或類，它可以讓其他函數(shù)或類在不需要做任何代碼修改的前提下增加額外功能，裝飾器的返回值也是一個(gè)函數(shù)/類對(duì)象。它經(jīng)常用于為已有函數(shù)/類添加記錄日志、計(jì)時(shí)統(tǒng)計(jì)、性能測(cè)試等。
首先定義一個(gè)倒計(jì)時(shí)函數(shù)，這個(gè)函數(shù)的功能非常簡(jiǎn)單，就是把n從當(dāng)前值減少到0。
def countdown(n):
while n > 0:
print('time' + str(n))
n -= 1
print(countdown.__name__)
程序輸出:
countdown
1.為函數(shù)增加一個(gè)日志裝飾器
假設(shè)現(xiàn)在要增強(qiáng)countdown的功能，在函數(shù)調(diào)用前后自動(dòng)打印日志，又不想修改函數(shù)自身的功能。這種在代碼運(yùn)行期間動(dòng)態(tài)增加功能的方式，稱之為裝飾器(Decorator)。
能打印日志的decorator，可以定義如下：
def log(func):
def wrapper(*args, **kw):
print('call %s().' % func.__name__)
return func(*args, **kw)
return wrapper
然后我們借助Python的@語法，把decorator置于函數(shù)的定義處：
@log
def countdown(n):
while n > 0:
print('time:' + str(n))
n -= 1
countdown(10)
程序輸出：
call countdown().
time:10
time:9
time:8
time:7
time:6
time:5
time:4
time:3
time:2
time:1
但此時(shí)我們?cè)俅蛴『瘮?shù)的name:
print(countdown.__name__)
程序輸出:
wrapper
我們發(fā)現(xiàn)函數(shù)的元數(shù)據(jù)信息變了，這顯然不是我們想要的結(jié)果。
2. 在裝飾器中拷貝元數(shù)據(jù)
為了把函數(shù)的元數(shù)據(jù)信息都保留下來，我們可以直接使用Python提供的functools庫(kù)中的@wraps裝飾器。
from functools import wraps
def log(func):
@wraps(func)
def wrapper(*args, **kw):
print('call %s().' % func.__name__)
return func(*args, **kw)
return wrapper
@log
def countdown(n):
while n > 0:
print('time:' + str(n))
n -= 1
print(countdown.__name__)
程序輸出：
countdown
3.為函數(shù)增加一個(gè)計(jì)時(shí)裝飾器
添加函數(shù)裝飾器的方法已經(jīng)講清楚了，現(xiàn)在再實(shí)現(xiàn)一個(gè)完整的函數(shù)計(jì)時(shí)耗時(shí)裝飾器。
import time
from functools import wraps
def TimeCost(func):
@wraps(func)
def wrapper(*arg, **kwargs):
start = time.time()
result = func(*args, **kwargs)
end = time.time()
print(func.__name__, end - start)
return result
return wrapper
@TimeCost
def countdown(n):
while n > 0:
print('time:' + str(n))
n -= 1
countdown(10000)
函數(shù)輸出：
('countdown', 0.0004801750183105469)
參考資料：
https://www.liaoxuefeng.com/wiki/1016959663602400/1017451662295584
以上就是如何實(shí)現(xiàn)一個(gè)python函數(shù)裝飾器(Decorator)的詳細(xì)內(nèi)容，更多關(guān)于python函數(shù)裝飾器的資料請(qǐng)關(guān)注腳本之家其它相關(guān)文章！
相關(guān)文章
python簡(jiǎn)單獲取數(shù)組元素個(gè)數(shù)的方法
這篇文章主要介紹了python簡(jiǎn)單獲取數(shù)組元素個(gè)數(shù)的方法,實(shí)例分析了Python中l(wèi)en方法的相關(guān)使用技巧,非常簡(jiǎn)單實(shí)用,需要的朋友可以參考下2015-07-07
生成器指的是生成器對(duì)象,可以由生成器表達(dá)式得到,也可以使用yield關(guān)鍵字得到一個(gè)生成器函數(shù),調(diào)用這個(gè)函數(shù)得到一個(gè)生成器對(duì)象,下面這篇文章主要給大家介紹了關(guān)于Python中生成器的相關(guān)資料,需要的朋友可以參考下2022-04-04
詳解Python3 定義一個(gè)跨越多行的字符串的多種方法
這篇文章主要介紹了詳解Python3 定義一個(gè)跨越多行的字符串的多種方法，文中通過示例代碼介紹的非常詳細(xì)，對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值，需要的朋友們下面隨著小編來一起學(xué)習(xí)學(xué)習(xí)吧2020-09-09
淺談Python中的zip()與*zip()函數(shù)詳解
這篇文章主要介紹了淺談Python中的zip()與*zip()函數(shù)詳解，小編覺得挺不錯(cuò)的，現(xiàn)在分享給大家，也給大家做個(gè)參考。一起跟隨小編過來看看吧2018-02-02
在使用Pandas處理數(shù)據(jù)時(shí),常見的讀取數(shù)據(jù)的方式時(shí)從Excel或CSV文件中獲取,這篇文章主要給大家介紹了關(guān)于如何使用pandas生成、讀取csv文件的相關(guān)資料,需要的朋友可以參考下2021-07-07
原來我一直安裝 Python 庫(kù)的姿勢(shì)都不對(duì)呀
平常我都是直接執(zhí)行 pip install 安裝的第三方庫(kù)，很多教程也是這么介紹的，一直以來我都認(rèn)為這是標(biāo)準(zhǔn)的、正確的安裝 Python 第三方庫(kù)的姿勢(shì)。下面小編給大家分享一篇教程，一起看看吧2019-11-11
利用Tensorflow構(gòu)建和訓(xùn)練自己的CNN來做簡(jiǎn)單的驗(yàn)證碼識(shí)別方式
今天小編就為大家分享一篇利用Tensorflow構(gòu)建和訓(xùn)練自己的CNN來做簡(jiǎn)單的驗(yàn)證碼識(shí)別方式，具有很好的參考價(jià)值，希望對(duì)大家有所幫助。一起跟隨小編過來看看吧2020-01-01
python數(shù)據(jù)結(jié)構(gòu):數(shù)據(jù)類型
這篇文章主要介紹了python數(shù)據(jù)結(jié)構(gòu)中的數(shù)據(jù)類型，在?Python?以及其他所有面向?qū)ο缶幊陶Z言中，類都是對(duì)數(shù)據(jù)的構(gòu)成(狀態(tài))以及數(shù)據(jù)?能做什么(行為)的描述，下面我們就來你看看python數(shù)據(jù)結(jié)構(gòu)中的數(shù)據(jù)類型商務(wù)詳細(xì)介紹，需要的小伙伴可以參考一下2021-12-12
這篇文章主要介紹了Pytest測(cè)試框架基本使用方法詳解,文中通過示例代碼介紹的非常詳細(xì)，對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值,需要的朋友可以參考下2020-11-11
這篇文章主要介紹了Python深入學(xué)習(xí)之特殊方法與多范式,特殊方法就像PHP里的魔術(shù)方法,多范式是指可以用面向?qū)ο?、面向過程、函數(shù)式、聲明式等方式編寫程序,需要的朋友可以參考下2014-08-08