快捷導(dǎo)航

python裝飾器三種裝飾模式的簡單分析

更新時間：2020年09月04日 10:32:12 作者：侯賽雷

這篇文章主要介紹了python裝飾器的三種裝飾模式，幫助大家更好的理解和學(xué)習(xí)python 裝飾器，感興趣的朋友可以了解下

學(xué)設(shè)計模式中有個裝飾模式,用java實現(xiàn)起來不是很難,但是遠遠沒有python簡單,難怪越來越火了!

這里就簡單討論下python的幾種裝飾模式:

一無參裝飾器:

# 裝飾器
import time
 
 
# 裝飾器,記錄函數(shù)運行時間
def decorator01(fun):
 def wapper():
  stime = time.time()
  fun()
  etime = time.time()
  print("fun run time is {TIME}".format(TIME=etime - stime))
 return wapper # 必須要返回一個函數(shù)的內(nèi)存地址
 
 
# 使用裝飾器裝飾某個函數(shù),等價于 test01=decorator01(test01),
# 即將test01實際引用變成wapper函數(shù)內(nèi)存地址,所以執(zhí)行test01實際是執(zhí)行wapper
@decorator01
def test01():
 time.sleep(2)
 print("test01 is running")
 
 
test01() # 不修改代碼和調(diào)用方式,實現(xiàn)添加記錄時間功能

二函數(shù)帶參裝飾器:

# 裝飾器
import time
 
 
# 裝飾器,記錄函數(shù)運行時間
def decorator01(fun):
 def wapper(*args, **kwargs): # 使用非固定參數(shù),無論參數(shù)是什么,都可以傳遞進來
  stime = time.time()
  fun(*args, **kwargs)
  etime = time.time()
  print("fun run time is {TIME}".format(TIME=etime - stime))
 return wapper # 必須要返回一個函數(shù)的內(nèi)存地址
 
 
# test01() = wapper(), 所以裝飾器加參數(shù)是給嵌套函數(shù)加參數(shù)
@decorator01
def test01(args1):
 time.sleep(2)
 print("參數(shù)是 {NAME} ".format(NAME=args1))
 
 
test01("侯征") # 不修改代碼和調(diào)用方式,實現(xiàn)添加記錄時間功能

三裝飾器本身帶參數(shù)的裝飾器:

# 裝飾器
import time
 
 
# 如果裝飾器有參數(shù),最外層是裝飾器的參數(shù)
def decorator01(*args, **kwargs):
 print("裝飾器參數(shù):", *args, **kwargs)
 def out(fun): #第二層才是接受的函數(shù)
  def wapper(*args, **kwargs): # 使用非固定參數(shù),無論參數(shù)是什么,都可以傳遞進來
   stime = time.time()
   fun(*args, **kwargs)
   etime = time.time()
   print("fun run time is {TIME}".format(TIME=etime - stime))
 
  return wapper # 必須要返回一個函數(shù)的內(nèi)存地址
 return out # 要返回裝飾函數(shù)的內(nèi)存地址
 
 
# 裝飾器本身帶參數(shù),此時 decorator01(arg)=out,即相當(dāng)于 @out裝飾test01,所以 test01=out(fun)=wapper
@decorator01(1)
def test01(args1):
 time.sleep(2)
 print("參數(shù)是 {NAME} ".format(NAME=args1))
 
 
test01("侯征") # 不修改代碼和調(diào)用方式,實現(xiàn)添加記錄時間功能

　　這種一開始挺難理解的,但是只要記住一點,@語法糖裝飾器的作用就是給被裝飾的函數(shù)重新賦一個函數(shù)的內(nèi)存地址,即裝飾器內(nèi)部定義的那個

和你直接fun01=decorator(fun),然后 fun01()是一樣的,只是這樣寫不用改變原來調(diào)用方式