快捷導(dǎo)航

老生常談Python進(jìn)階之裝飾器

更新時間：2017年05月11日 07:51:53 投稿：jingxian

下面小編就為大家?guī)硪黄仙Ｕ凱ython進(jìn)階之裝飾器。小編覺得挺不錯的，現(xiàn)在就分享給大家，也給大家做個參考。一起跟隨小編過來看看吧

函數(shù)也是對象

要理解Python裝飾器，首先要明白在Python中，函數(shù)也是一種對象，因此可以把定義函數(shù)時的函數(shù)名看作是函數(shù)對象的一個引用。既然是引用，因此可以將函數(shù)賦值給一個變量，也可以把函數(shù)作為一個參數(shù)傳遞或返回。同時，函數(shù)體中也可以再定義函數(shù)。

裝飾器本質(zhì)

可以通過編寫一個純函數(shù)的例子來還原裝飾器所要做的事。

def decorator(func):
  
  def wrap():
    print("Doing someting before executing func()")
    func()
    print("Doing someting after executing func()")

  return wrap


def fun_test():
  print("func")


fun_test = decorator(fun_test)
fun_test()

# Output:
# Doing someting before executing func()
# func
# Doing someting after executing func()

fun_test所指向的函數(shù)的引用傳遞給decorator()函數(shù)

decorator()函數(shù)中定義了wrap()子函數(shù)，這個子函數(shù)會調(diào)用通過func引用傳遞進(jìn)來的fun_test()函數(shù)，并在調(diào)用函數(shù)的前后做了一些其他的事情

decorator()函數(shù)返回內(nèi)部定義的wrap()函數(shù)引用

fun_test接收decorator()返回的函數(shù)引用，從而指向了一個新的函數(shù)對象

通過fun_test()調(diào)用新的函數(shù)執(zhí)行wrap()函數(shù)的功能，從而完成了對fun_test()函數(shù)的前后裝飾

Python中使用裝飾器

在Python中可以通過@符號來方便的使用裝飾器功能。

def decorator(func):
  
  def wrap():
    print("Doing someting before executing func()")
    func()
    print("Doing someting after executing func()")

  return wrap

@decorator
def fun_test():
  print("func")


fun_test()

# Output:
# Doing someting before executing func()
# func
# Doing someting after executing func()

裝飾的功能已經(jīng)實(shí)現(xiàn)了，但是此時執(zhí)行:

print(fun_test.__name__)

# Output:
# wrap

fun_test.__name__已經(jīng)變成了wrap，這是應(yīng)為wrap()函數(shù)已經(jīng)重寫了我們函數(shù)的名字和注釋文檔。此時可以通過functools.wraps來解決這個問題。wraps接受一個函數(shù)來進(jìn)行裝飾，并加入了復(fù)制函數(shù)名稱、注釋文檔、參數(shù)列表等等功能。這可以讓我們在裝飾器里面訪問在裝飾之前的函數(shù)的屬性。

更規(guī)范的寫法：

from functools import wraps

def decorator(func):
  @wraps(func)
  def wrap():
    print("Doing someting before executing func()")
    func()
    print("Doing someting after executing func()")

  return wrap


@decorator
def fun_test():
  print("func")


fun_test()
print(fun_test.__name__)

# Output:
# Doing someting before executing func()
# func
# Doing someting after executing func()
# fun_test

帶參數(shù)的裝飾器

通過返回一個包裹函數(shù)的函數(shù)，可以模仿wraps裝飾器，構(gòu)造出一個帶參數(shù)的裝飾器。

from functools import wraps

def loginfo(info='info1'):
  def loginfo_decorator(func):
    @wraps(func)
    def wrap_func(*args, **kwargs):
      print(func.__name__ + ' was called')
      print('info: %s' % info)
      
      return func(*args, **kwargs)
    return wrap_func
  return loginfo_decorator
  
@loginfo()
def func1():
  pass
  
func1()

# Output:
# func1 was called
# info: info1

@loginfo(info='info2')
def func2():
  pass

func2()
# Output:
# func2 was called
# info: info2

裝飾器類

通過編寫類的方法也可以實(shí)現(xiàn)裝飾器，并讓裝飾器具備繼承等面向?qū)ο笾懈鼘?shí)用的特性

首先編寫一個裝飾器基類：

from functools import wraps

class loginfo:
  def __init__(self, info='info1'):
    self.info = info
    
  def __call__(self, func):
    @wrap
    def wrap_func(*args, **kwargs):
      print(func.__name__ + ' was called')
      print('info: %s' % self.info)
      
      self.after()  # 調(diào)用after方法，可以在子類中實(shí)現(xiàn)
      return func(*args, **kwargs)
    return wrap_func

  def after(self):
    pass


@loginfo(info='info2')
def func1():
  pass
  
# Output:
# func1 was called
# info: info1

再通過繼承l(wèi)oginfo類，擴(kuò)展裝飾器的功能：

class loginfo_after(loginfo):
  def __init__(self, info2='info2', *args, **kwargs):
    self.info2 = info2
    super(loginfo_after, self).__init__(*args, **kwargs)

  def after(self):
    print('after: %s' % self.info2)


@loginfo_after()
def func2():
  pass

func2()
  
# Output:
# func2 was called
# info: info1
# after: info2

以上這篇老生常談Python進(jìn)階之裝飾器就是小編分享給大家的全部內(nèi)容了，希望能給大家一個參考，也希望大家多多支持腳本之家。

您可能感興趣的文章: