Python函數(shù)裝飾器常見(jiàn)使用方法實(shí)例詳解

更新時(shí)間：2019年03月30日 12:25:53 作者：Ly_ox

這篇文章主要介紹了Python函數(shù)裝飾器常見(jiàn)使用方法,結(jié)合實(shí)例形式分析了Python函數(shù)裝飾器的概念、原理、用法及相關(guān)操作注意事項(xiàng),需要的朋友可以參考下

本文實(shí)例講述了Python函數(shù)裝飾器常見(jiàn)使用方法。分享給大家供大家參考，具體如下：

一、裝飾器

首先，我們要了解到什么是開(kāi)放封閉式原則？

軟件一旦上線(xiàn)后，對(duì)修改源代碼是封閉的，對(duì)功能的擴(kuò)張是開(kāi)放的，所以我們應(yīng)該遵循開(kāi)放封閉的原則。

也就是說(shuō)：我們必須找到一種解決方案，能夠在不修改一個(gè)功能源代碼以及調(diào)用方式的前提下，為其加上新功能。

總結(jié):原則如下：

1、不修改源代碼
2、不修改調(diào)用方式

目的：在遵循1和2原則的基礎(chǔ)上擴(kuò)展新功能。

二、什么是裝飾器？

器：指的是工具，

裝飾：指的是為被裝飾對(duì)象添加新功能。

完整的含義：裝飾器即在不修改裝飾對(duì)象源代碼與調(diào)用方式的前提下，為被裝飾器對(duì)象添加新功能的一種函數(shù)，這個(gè)函數(shù)的特殊之處就在于它的返回值也是一個(gè)函數(shù)。

一般而言，我們想要拓展原來(lái)函數(shù)的代碼，直接的辦法就是侵入代碼里面修改，例如：

import time
def index():
  start_time=time.time()
  time.sleep(3)
  print('hello word')
  stop_time=time.time()
  print('run time is %s'%(stop_time-start_time))
index()

輸出：

hello word
run time is 3.000171422958374

以上代碼就是讓你過(guò)三秒才打印'hello word'，下面我們要再添加一個(gè)新功能，和上面的功能一樣，但是要傳參數(shù)進(jìn)去，過(guò)5秒輸出結(jié)果。

修改1：

import time
def index():
  time.sleep(3)
  print('hello word')
def home(name):
  time.sleep(5)
  print('welcome %s to home page'%name)
def wrapper(func):
  start_time=time.time()
  func('haolo')
  stop_time=time.time()
  print('run time is %s'%(stop_time-start_time))
wrapper(home)

輸出：

welcome haolo to home page
run time is 5.000286102294922

這樣寫(xiě)感覺(jué)還是不怎么好，而且我們還修改了函數(shù)的調(diào)用方式，很不符合規(guī)矩。所以我們還是換一種方式來(lái)修改它。通過(guò)裝飾器的方式。

修改2

import time
def index():
  time.sleep(3)
  print('hello word')
def home(name):
  time.sleep(5)
  print('welcome to %s'%name)
def outter(func): # func為最原始的inde 和home
  def warpper():
    start_time=time.time()
    func('yuan')
    stop_time=time.time()
    print(stop_time-start_time)
  return warpper
home=outter(home) ###home這個(gè)變量名是新賦值的，把原來(lái)的home給覆蓋了。
home()

輸出：

welcome to yuan
5.000286102294922

這種方式雖然滿(mǎn)足了不修改源代碼和不修改調(diào)用方式的條件，但還是不能夠?qū)崿F(xiàn)兩個(gè)函數(shù)同時(shí)運(yùn)行的功能，說(shuō)到底還是不行，我們還得想個(gè)方式出來(lái)。就是讓他們兩個(gè)同時(shí)運(yùn)行。這時(shí)，我又想到了上節(jié)課所學(xué)的知識(shí)，就是*args和**kargs，用兩個(gè)函數(shù)通過(guò)可變參數(shù)形式來(lái)實(shí)現(xiàn)內(nèi)嵌函數(shù)的形式傳入，所以它支持運(yùn)行是構(gòu)建參數(shù)列表，這對(duì)于以上兩次不能解決的辦法是最有效的。下面我們來(lái)試試，看到底能不能成功。

方式3：

import time
def index():
  time.sleep(3)
  print('hello word')
def home(name):
  time.sleep(5)
  print('welcome %s to home page'%name)
def timmer(func):  #func為最原始的home
  def warpper(*args,**kwargs):
    start_time=time.time()
    res=func(*args,**kwargs) #調(diào)用了最原始的home
    stop_time=time.time()
    print(stop_time-start_time)
    return res
  return warpper
index=timmer(index) #為最新的index = wrapper
home=timmer(home) #為最新的home = wrapper
home(name='yuan') #wrapper=('yuan')
index() #wrapper

輸出：

welcome yuan to home page
5.000285863876343
hello word
3.000171661376953

看吧，很快就實(shí)現(xiàn)了兩個(gè)功能并用，而且我們還沒(méi)有修改原始代碼，還有調(diào)用方式。

其實(shí)很簡(jiǎn)單，我只是用了一個(gè)無(wú)參裝飾器的模板，這個(gè)模板可以說(shuō)是萬(wàn)能的，在以后很多的函數(shù)代碼都可以用這種方式來(lái)套用。

模板：

def outer(func):
  def inner(*args,**kwargs):
    res=func(*args,**kwargs)
    return res
  return inner

現(xiàn)在又有問(wèn)題來(lái)了，我們調(diào)裝飾器的時(shí)候，每調(diào)一次，又要把裝飾器對(duì)象傳進(jìn)來(lái)，調(diào)一次又傳一次，這樣不會(huì)覺(jué)得很麻煩嗎？那么我們又想到了一種方法，就是裝飾器語(yǔ)法糖，在被裝飾對(duì)象的上面加@timmer 用它來(lái)取代 index=timmer(index)

并且把返回值正常的返回給它。

import time
def timmer(func):  #func為最原始的home
  def warpper(*args,**kwargs):
    start_time=time.time()
    res=func(*args,**kwargs) #調(diào)用了最原始的home
    stop_time=time.time()
    print(stop_time-start_time)
    return res
  return warpper
@timmer #就是取代底下的 index=timmer(index)
def index():
  time.sleep(3)
  print('hello word')
@timmer #就是取代底下的home=timmer(home)  home(name='yuan')
def home(name):
  time.sleep(5)
  print('welcome %s to home page'%name)
index()
home('liuyuan')

輸出：

hello word
3.000171661376953
welcome liuyuan to home page
5.000286102294922

注意：這里的timmer函數(shù)就是最原始的裝飾器，它的參數(shù)就是一個(gè)函數(shù)，然后返回值也是一個(gè)函數(shù)。其中作為參數(shù)的這個(gè)函數(shù)index()和hemo(name)就是在返回函數(shù)的wrapper()的內(nèi)部執(zhí)行。然后再這兩個(gè)函數(shù)的前面加上@timmer，index()和home(name)函數(shù)就相當(dāng)于被注入了計(jì)時(shí)功能，現(xiàn)在只需要調(diào)用index()和home('yuan')，它就已經(jīng)變身為'新功能更多的函數(shù)了。'

所以這里的裝飾器就像一個(gè)注入的符號(hào)：有了它，拓展了原來(lái)函數(shù)的功能既不需要侵入函數(shù)內(nèi)更改代碼，也不需要重復(fù)執(zhí)行原函數(shù)。

用裝飾器來(lái)實(shí)現(xiàn)認(rèn)證功能：

import time
current_user={
  'username':None,
  #'login_time':None
}
def auth(func):
  # func = index
  def wrapper(*args,**kwargs):
    if current_user['username']:
      print('已經(jīng)登錄過(guò)了')
      res=func(*args,**kwargs)
      return res
    uname = input('輸入用戶(hù)名：').strip()
    pwd = input('密碼：').strip()
    if uname == 'yuan' and pwd == '123':
      print('登錄成功')
      current_user['username']=uname
      res = func(*args,**kwargs)
      return res
    else:
      print('用戶(hù)名或密碼錯(cuò)誤')
  return wrapper
@auth #index = auth(index)
def index():
  time.sleep(1)
  print('welcom to index page')
@auth
def home(name):
  time.sleep(2)
  print('welcome %s to home page'%name)
input()
home('yuan')

有參數(shù)的裝飾器來(lái)用于用戶(hù)認(rèn)證

import time
current_user={
  'username':None,
  # 'login_time':None
}
def auth(func):
  # func=index
  def wrapper(*args,**kwargs):
    if current_user['username']:
      print('已經(jīng)登陸過(guò)了')
      res=func(*args,**kwargs)
      return res
    uname=input('用戶(hù)名>>: ').strip()
    pwd=input('密碼>>: ').strip()
    if uname == 'yuan' and pwd == '123':
      print('登陸成功')
      current_user['username']=uname
      res=func(*args,**kwargs)
      return res
    else:
      print('用戶(hù)名或密碼錯(cuò)誤')
  return wrapper
def timmer(func):
  def wrapper(*args,**kwargs):
    start_time=time.time()
    res=func(*args,**kwargs)
    stop_time=time.time()
    print(stop_time-start_time)
    return res
  return wrapper
@timmer # timmer 統(tǒng)計(jì)的是auth+index的執(zhí)行時(shí)間
@auth
def index():
  time.sleep(1)
  print('welcome to index page')
  return 122
index()

疊加多個(gè)裝飾器：

import time
current_user={
  'username':None,
  # 'login_time':None
}
def auth(func):
  # func=index
  def wrapper(*args,**kwargs):
    if current_user['username']:
      print('已經(jīng)登陸過(guò)了')
      res=func(*args,**kwargs)
      return res
    uname=input('用戶(hù)名>>: ').strip()
    pwd=input('密碼>>: ').strip()
    if uname == 'egon' and pwd == '123':
      print('登陸成功')
      current_user['username']=uname
      res=func(*args,**kwargs)
      return res
    else:
      print('用戶(hù)名或密碼錯(cuò)誤')
  return wrapper
def timmer(func):
  def wrapper(*args,**kwargs):
    start_time=time.time()
    res=func(*args,**kwargs)
    stop_time=time.time()
    print(stop_time-start_time)
    return res
  return wrapper
@timmer # timmer 統(tǒng)計(jì)的是auth+index的執(zhí)行時(shí)間
@auth
def index():
  time.sleep(1)
  print('welcome to index page')
  return 122
index()

關(guān)于Python相關(guān)內(nèi)容感興趣的讀者可查看本站專(zhuān)題：《Python函數(shù)使用技巧總結(jié)》、《Python面向?qū)ο蟪绦蛟O(shè)計(jì)入門(mén)與進(jìn)階教程》、《Python數(shù)據(jù)結(jié)構(gòu)與算法教程》、《Python字符串操作技巧匯總》、《Python編碼操作技巧總結(jié)》及《Python入門(mén)與進(jìn)階經(jīng)典教程》

希望本文所述對(duì)大家Python程序設(shè)計(jì)有所幫助。

您可能感興趣的文章: