python 之裝飾器(decorator)

ckxllf發表於2019-12-17

  定義:

  本質是函式(裝飾其他函式),就是為其他函式新增附加功能。

  原則:

  不能修改被裝飾的函式的原始碼

  不能修改被裝飾函式的呼叫方式

  先看一個完整的裝飾器的例子:

  # Author: Mr.Xue

  # 2019.10.23

  """

  實現裝飾的知識儲備:

  1、函式即‘變數’

  2、高階函式

  3、巢狀函式

  高階函式+巢狀函式--->裝飾器

  """

  import time

  def timmer(func):

  def warpper(*args, **kwargs):

  start_time = time.time()

  func()

  stop_time = time.time()

  print('the func run time is %s' % (stop_time - start_time))

  return warpper

  @timmer

  def test1():

  time.sleep(3)

  print("in the func test1")

  test1()

  1、函式即“變數”

  函式在記憶體中的儲存機制:在記憶體中儲存函式體,函式名作為門牌號

  注:

  python中的回收機制:不管是函式還是變數,一但"門牌號"不存在了,記憶體空間就會被回收

  2、高階函式

  符合下面兩個條件任何一個的就是高階函式:

  把一個函式名作為實參傳給另一個函式

  返回值中包含函式名

  '''

  第一種高階函式:

  把一個函式名作為實參傳給另一個函式

  --> 實現了不修改被裝飾的函式bar程式碼的情況下為其新增新功能

  '''

  import time

  def bar():

  time.sleep(3)

  print('in the bar')

  def test(func):

  start_time = time.time()

  print(func)

  func() # run function bar

  stop_time = time.time()

  print("the func run time is %s" %(stop_time-start_time))

  test(bar)

  #x = bar # 函式即'變數'

  #x()

  '''

  第二種高階函式:

  返回值中包含函式名

  --> 不修改函式的呼叫方式

  '''

  import time

  def bar():

  time.sleep(3)

  print('in the bar')

  def test2(func):

  print(func) # func的記憶體地址

  return func #返回func的記憶體地址

  print(test2(bar))

  bar = test2(bar)

  bar()

  3、巢狀函式

  在一個函式的函式體中還有一個用def宣告函式

  def foo():

  print("in the foo")

  def bar():

  print("in the bar")

  bar()

  foo()

  好啦,理解了上面的三個知識點之後,我們來一步步寫一個裝飾器,首先我們先定義兩個函式test1、test2,再給兩個函式新增新的功能,計算這兩個函式的執行時間,怎麼做呢,先定義一個高階函式(上面提到的第一種高階函式),將這兩個函式當作引數傳進去。

  import time

  def deco(func):

  start_time = time.time()

  func() # run function bar

  stop_time = time.time()

  print("the func run time is %s" %(stop_time-start_time))

  def test1():

  time.sleep(3)

  print('in the test1')

  def test2():

  time.sleep(3)

  print('in the test2')

  deco(test1)

  deco(test2)

  現在我們完成了裝飾器的原則之一,再不修改原始碼的基礎上給函式新增了新的功能。

  看上去好像不難呀,接下來,只要我們改進程式碼,完成不修改函式的呼叫方式的這個原則,是不是就完成了呢,好,那我們將上面的高階函式改寫成第二種的高階函式

  import time

  def deco(func):

  start_time = time.time()

  return func # 返回func函式的記憶體地址

  stop_time = time.time()

  print("the func run time is %s" %(stop_time-start_time))

  def test1():

  time.sleep(3)

  print('in the test1')

  def test2():

  time.sleep(3)

  print('in the test2')

  test1 = deco(test1)

  test2 = deco(test2)

  test1() #沒有修改呼叫方式,新功能卻不見了

  test2() #沒有修改呼叫方式

  利用了第二種高階函式之後,我們完成了裝飾器的另外一個原則,沒有改變函式的呼叫方式,但是發現新功能卻沒有加上去,這時候只用高階函式來實現裝飾器好像有點困難啊,怎麼辦呢?好像進入死迴圈了呀。

  不知道大家注意到沒有,在上面那個裝飾器的例子裡面,有一行寫著,“高階函式+巢狀函式—>裝飾器”,哦,那是不是我們再搞個巢狀函式就可以了呢,按照這個思路,我們繼續往下走。

  import time

  def timer(func):

  def deco():

  start_time = time.time()

  func() # 呼叫func函式

  stop_time = time.time()

  print("the func run time is %s" %(stop_time-start_time))

  return deco

  def test1():

  time.sleep(3)

  print('in the test1')

  def test2():

  time.sleep(3)

  print('in the test2')

  test1 = timer(test1)

  test2 = timer(test2)

  test1() #沒有修改呼叫方式,新功能也加上去了

  test2() #沒有修改呼叫方式,新功能也加上去了

  哇哦,現在就已經完成了裝飾器了,只不過呼叫方式看上去比較lower而已,只要修改成標準寫法就ok了。

  怎麼做呢,在要被裝飾的函式頭部,加上"@裝飾器名字",就好啦。

  import time

  def timer(func):

  def deco():

  start_time = time.time()

  func() # 呼叫func函式

  stop_time = time.time()

  print("the func run time is %s" %(stop_time-start_time))

  return deco

  @timer # test1 = timer(test1)

  def test1():

  time.sleep(3)

  print('in the test1')

  @timer

  def test2():

  time.sleep(3)

  print('in the test2')

  test1()

  test2()

  只是現在我們寫的這個裝飾器和上面那個還有一點區別,我們的裝飾器沒有引數,而上面的那個卻有,接下來,我們繼續完善完善。 鄭州人流多少錢

  給我們的內嵌函式加上兩個非固定引數,這樣就可以接收引數了,如下:

  import time

  def timer(func):

  def deco(*args, **kwargs):

  start_time = time.time()

  func(*args, **kwargs) # 呼叫func函式

  stop_time = time.time()

  print("the func run time is %s" %(stop_time-start_time))

  return deco

  @timer # test1 = timer(test1)

  def test1():

  time.sleep(3)

  print('in the test1')

  @timer

  def test2(name, age):

  time.sleep(3)

  print('in the test2', name, age)

  test1()

  test2("xue", 24)

  ok, all finish!

  現在我們用裝飾器寫一個網站頁面的需求:比如說,一共20個網站,其中除了首頁不需要驗證可以直接顯示外,其他的都需要驗證使用者名稱密碼才可以顯示。

  # Author: Mr.Xue

  # 2019.10.24

  import time

  user, passwd = 'xue', 'abc123'

  def auth(auth_type):

  print("auth_type: ", auth_type)

  def outer_warpper(func):

  def warpper(*args, **kwargs):

  print("warpper func args:", *args, **kwargs)

  if auth_type == 'local':

  username = input("Username: ")

  password = input("Password: ")

  if user == username and passwd == password:

  print("\033[32;1mUser has passed authentication\033[0m")

  res = func(*args, **kwargs)

  return res

  else:

  print("\033[31;1mInvalid username or password\033[0m")

  elif auth_type == 'ldap':

  print('bu hui gao mao xian')

  return warpper

  return outer_warpper

  def index():

  print("in the index")

  @auth(auth_type='local') #home = warpper(home) -> warpper

  def home():

  print("in the home")

  return 'from home'

  @auth(auth_type='ldap')

  def bbs():

  print('in the bbs')

  index()

  print(home())

  bbs()


來自 “ ITPUB部落格 ” ,連結:http://blog.itpub.net/69945560/viewspace-2669064/,如需轉載,請註明出處,否則將追究法律責任。

相關文章