Python基礎(chǔ)【裝飾器】

裝飾器：

裝飾器：用來(lái)修改函數(shù)功能的函數(shù)

• 可以在不改變?cè)瘮?shù)的基礎(chǔ)上添加功能
  實(shí)現(xiàn)裝飾器的方法：從函數(shù)中返回函數(shù)，將原函數(shù)作為一個(gè)參數(shù)傳給另一個(gè)函數(shù)

  代碼：裝飾器pro_print在函數(shù)執(zhí)行前輸出提示"welcome to class"

  def pro_print(fun): # 裝飾器函數(shù)
  def wrapper(*args,**kwargs): # 接收各種類(lèi)型的不定長(zhǎng)參數(shù)
  print("welcome to class")
  fun()
  return wrapper
  @pro_print # 語(yǔ)法糖，在函數(shù)前使用用于裝飾函數(shù)
  def fun():
  print('hello,python')
  fun()

  ---

  但裝飾器會(huì)覆蓋原有函數(shù)的函數(shù)名，提示文檔等信息

  def pro_print(fun): # 裝飾器函數(shù)
  def wrapper(*args,**kwargs): # 接收各種類(lèi)型的不定長(zhǎng)參數(shù)
  '''執(zhí)行函數(shù)前提示welcom to class'''
  print("welcome to class")
  fun()
  return wrapper
  @pro_print # 語(yǔ)法糖，在函數(shù)前使用用于裝飾函數(shù)
  def fun():
  '''輸出hello,python'''
  print('hello,python')
  fun()
  print(fun.__name__)
  print(fun.__doc__)

  測(cè)試結(jié)果：

  

---

導(dǎo)入functools中的wraps函數(shù)可以解決這個(gè)問(wèn)題

from functools import wraps
def pro_print(fun): # 裝飾器函數(shù)
@wraps(fun)
def wrapper(*args,**kwargs): # 接收各種類(lèi)型的不定長(zhǎng)參數(shù)
'''執(zhí)行函數(shù)前提示welcom to class'''
print("welcome to class")
fun()
return wrapper
@pro_print # 語(yǔ)法糖，在函數(shù)前使用用于裝飾函數(shù)
def fun():
'''輸出hello,python'''
print('hello,python')
fun()
print(fun.__name__)

測(cè)試結(jié)果：