一些關(guān)于python 裝飾器的個(gè)人理解

更新時(shí)間：2020年08月31日 14:16:03 作者：Wonz

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

裝飾器

本質(zhì)是一個(gè)接受參數(shù)為函數(shù)的函數(shù)。
作用：為一個(gè)已經(jīng)實(shí)現(xiàn)的方法添加額外的通用功能，比如日志記錄、運(yùn)行計(jì)時(shí)等。

舉例

1.不帶參數(shù)的裝飾器，不用@

# 不帶參數(shù)的裝飾器
def deco_test(func):
  def wrapper(*args, **kwargs):
    print("before function")
    f = func(*args, **kwargs)
    print("after function")
    return f
  return wrapper

def do_something(a,b,c):
  print(a)
  time.sleep(1)
  print(b)
  time.sleep(1)
  print(c)
  return a

if __name__ == '__main__':
  # 不用@
  f = deco_test(do_something)("1","2","3")

輸出：

before function
1
2
3
after function

個(gè)人理解：

相當(dāng)于在 do_something 函數(shù)外面套了兩個(gè)輸出： before function 和 after function 。

2.不帶參數(shù)的裝飾器，用 @

# 不帶參數(shù)的裝飾器
def deco_test(func):
  def wrapper(*args, **kwargs):
    print("before function")
    f = func(*args, **kwargs)
    print("after function")
    return f
  return wrapper

@deco_test
def do_something(a,b,c):
  print(a)
  time.sleep(1)
  print(b)
  time.sleep(1)
  print(c)
  return a

if __name__ == '__main__':
  # 使用@
  f = do_something("1","2","3")

輸出：

before function
1
2
3
after function

個(gè)人理解：

相當(dāng)于執(zhí)行 do_something 函數(shù)的時(shí)候，因?yàn)橛?@ 的原因，已經(jīng)知道有一層裝飾器 deco_test ，所以不需要再單獨(dú)寫(xiě) deco_test(do_something) 了。

3.帶參數(shù)的裝飾器

# 帶參數(shù)的裝飾器
def logging(level):
  def wrapper(func):
    def inner_wrapper(*args, **kwargs):
      print("[{level}]: enter function {func}()".format(level=level, func=func.__name__))
      f = func(*args, **kwargs)
      print("after function: [{level}]: enter function {func}()".format(level=level, func=func.__name__))
      return f
    return inner_wrapper
  return wrapper

@logging(level="debug")
def do_something(a,b,c):
  print(a)
  time.sleep(1)
  print(b)
  time.sleep(1)
  print(c)
  return a

if __name__ == '__main__':
  # 使用@
  f = do_something("1","2","3")

輸出：

[debug]: enter function do_something()
1
2
3
after function: [debug]: enter function do_something()

個(gè)人理解：

裝飾器帶了一個(gè)參數(shù) level = "debug" 。

最外層的函數(shù) logging() 接受參數(shù)并將它們作用在內(nèi)部的裝飾器函數(shù)上面。內(nèi)層的函數(shù) wrapper() 接受一個(gè)函數(shù)作為參數(shù)，然后在函數(shù)上面放置一個(gè)裝飾器。這里的關(guān)鍵點(diǎn)是裝飾器是可以使用傳遞給 logging() 的參數(shù)的。

4.類(lèi)裝飾器

# 類(lèi)裝飾器
class deco_cls(object):
  def __init__(self, func):
    self._func = func

  def __call__(self, *args, **kwargs):
    print("class decorator before function")
    f = self._func(*args, **kwargs)
    print("class decorator after function")
    return f

@deco_cls
def do_something(a,b,c):
  print(a)
  time.sleep(1)
  print(b)
  time.sleep(1)
  print(c)
  return a

if __name__ == '__main__':
  # 使用@
  f = do_something("1","2","3")

輸出：

class decorator before function
1
2
3
class decorator after function

個(gè)人理解：

使用一個(gè)裝飾器去包裝函數(shù)，返回一個(gè)可調(diào)用的實(shí)例。因此定義了一個(gè)類(lèi)裝飾器。

5.兩層裝飾器

# 不帶參數(shù)的裝飾器
def deco_test(func):
  def wrapper(*args, **kwargs):
    print("before function")
    f = func(*args, **kwargs)
    print("after function")
    return f
  return wrapper

# 帶參數(shù)的裝飾器
def logging(level):
  def wrapper(func):
    def inner_wrapper(*args, **kwargs):
      print("[{level}]: enter function {func}()".format(level=level, func=func.__name__))
      f = func(*args, **kwargs)
      print("after function: [{level}]: enter function {func}()".format(level=level, func=func.__name__))
      return f
    return inner_wrapper
  return wrapper

@logging(level="debug")
@deco_test
def do_something(a,b,c):
  print(a)
  time.sleep(1)
  print(b)
  time.sleep(1)
  print(c)
  return a

if __name__ == '__main__':
  # 使用@
  f = do_something("1","2","3")

輸出：