本文例項講述了Python 裝飾器@，對函式進行功能擴充套件操作。分享給大家供大家參考，具體如下：

裝飾器可以對原函式進行功能擴充套件，但還不需要修改原函式的內容(開閉原則)，也不需要修改原函式的呼叫。

demo.py（裝飾器，@）：

# 閉包
def w1(func):
  def inner():
    # 對原函式進行功能擴充套件
    print("功能擴充套件")
    func()
    # return func() # 如果原函式需要返回值,可以return
  return inner # 閉包
@w1
# 相當於 f1 = w1(f1)
def f1():
  print('f1') # 原函式不需要修改
f1() # 原函式的呼叫也不需要修改

demo.py（裝飾器通用格式,對不定長引數並且有返回值的函式進行裝飾）：

def set_func(func):
  def call_func(*args,**kwargs):
    print("裝飾器擴充套件的功能")
    return func(*args,**kwargs) # 這裡的*和*表示拆包。 不管有沒有返回值,return都沒問題。
  return call_func
@set_func # 相當於 test1 = set_func(test1)
# 對含有不定長引數並且有返回值的函式進行裝飾。
def test1(num,*args,**kwargs):
  print("-----test1----%d" % num)
  return "ok"
ret = test1(100)
print(ret)

demo.py（多個裝飾器的裝飾順序）：

def add_1(func):
  def call_func(*args,**kwargs):
    print("裝飾器1 擴充套件的功能")
    return func(*args,**kwargs)
  return call_func
def add_2(func):
  def call_func(*args,**kwargs):
    print("裝飾器2 擴充套件的功能")
    return func(*args,**kwargs)
  return call_func
@add_2
@add_1
# 先裝飾add_1，再裝飾add_2
def test1():
  print("------test1------")
test1() # 在呼叫函式之前就已經裝飾好了。
# 裝飾器2 擴充套件的功能
# 裝飾器1 擴充套件的功能
# ------test1------

demo.py（用類充當裝飾器）：

# 用類充當裝飾器
class Test(object):
  def __init__(self,func):
    self.func = func
  def __call__(self,**kwargs):
    print("這裡是裝飾器新增的功能.....")
    return self.func(*args,**kwargs)
@Test # 相當於get_str = Test(get_str) # 例項化物件,呼叫__init__方法。
def get_str():
  return "haha"
print(get_str())  # 例項物件(),會自動呼叫物件的__call__方法。

@functools.wraps修飾裝飾器的內層函式。(修飾內層函式後,被裝飾器裝飾的函式的__name__、__doc__不會被裝飾器改變)

demo.py（@functools.wraps修飾裝飾器的內層函式）：

# coding:utf-8
import functools # 匯入
# 自定義的裝飾器
def login_required(func):
  @functools.wraps(func)
  # 裝飾器的內層函式,一般要加@functools.wraps裝飾器
  def wrapper(*arg,**kwargs):
    """wrapper的說明文件"""
    # 。。。
    return func(*arg,**kwargs)
  return wrapper
# 使用自定義的裝飾器
@login_required
def demofunc():
  """demofunc的說明文件"""
  pass
print(demofunc.__name__)  # 不加@functools.wraps裝飾器時："wrapper"。 加裝飾器時："demofunc"
print(demofunc.__doc__)  # 不加@functools.wraps裝飾器時："wrapper的說明文件"。 加裝飾器時："demofunc的說明文件"

更多關於Python相關內容感興趣的讀者可檢視本站專題：《Python面向物件程式設計入門與進階教程》、《Python資料結構與演算法教程》、《Python函式使用技巧總結》、《Python字串操作技巧彙總》、《Python編碼操作技巧總結》及《Python入門與進階經典教程》

希望本文所述對大家Python程式設計有所幫助。