本文實例講述了Python 裝飾器@，對函數(shù)進行功能擴展操作。分享給大家供大家參考，具體如下：

裝飾器可以對原函數(shù)進行功能擴展，但還不需要修改原函數(shù)的內(nèi)容(開閉原則)，也不需要修改原函數(shù)的調(diào)用。

demo.py（裝飾器，@）：

# 閉包def w1(func):  def inner():    # 對原函數(shù)進行功能擴展    print("功能擴展")    func()    # return func() # 如果原函數(shù)需要返回值,可以return  return inner # 閉包@w1# 相當(dāng)于 f1 = w1(f1)def f1():  print('f1') # 原函數(shù)不需要修改f1() # 原函數(shù)的調(diào)用也不需要修改

demo.py（裝飾器通用格式,對不定長參數(shù)并且有返回值的函數(shù)進行裝飾）：

def set_func(func):  def call_func(*args, **kwargs):    print("裝飾器擴展的功能")    return func(*args, **kwargs) # 這里的*和*表示拆包。 不管有沒有返回值,return都沒問題。  return call_func@set_func # 相當(dāng)于 test1 = set_func(test1)# 對含有不定長參數(shù)并且有返回值的函數(shù)進行裝飾。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_funcdef add_2(func):  def call_func(*args, **kwargs):    print("裝飾器2 擴展的功能")    return func(*args, **kwargs)  return call_func@add_2@add_1# 先裝飾add_1，再裝飾add_2def test1():  print("------test1------")test1() # 在調(diào)用函數(shù)之前就已經(jīng)裝飾好了。# 裝飾器2 擴展的功能# 裝飾器1 擴展的功能# ------test1------

demo.py（用類充當(dāng)裝飾器）：

# 用類充當(dāng)裝飾器class Test(object):  def __init__(self, func):    self.func = func  def __call__(self, *args, **kwargs):    print("這里是裝飾器添加的功能.....")    return self.func(*args, **kwargs)@Test # 相當(dāng)于get_str = Test(get_str) # 實例化對象,調(diào)用__init__方法。def get_str():  return "haha"print(get_str())  # 實例對象(),會自動調(diào)用對象的__call__方法。

@functools.wraps修飾裝飾器的內(nèi)層函數(shù)。(修飾內(nèi)層函數(shù)后,被裝飾器裝飾的函數(shù)的__name__、__doc__不會被裝飾器改變)

demo.py（@functools.wraps修飾裝飾器的內(nèi)層函數(shù)）：

# coding:utf-8import functools # 導(dǎo)入# 自定義的裝飾器def login_required(func):  @functools.wraps(func)  # 裝飾器的內(nèi)層函數(shù),一般要加@functools.wraps裝飾器  def wrapper(*arg, **kwargs):    """wrapper的說明文檔"""    # 。。。    return func(*arg, **kwargs)  return wrapper# 使用自定義的裝飾器@login_requireddef demofunc():  """demofunc的說明文檔"""  passprint(demofunc.__name__)  # 不加@functools.wraps裝飾器時："wrapper"。 加裝飾器時："demofunc"print(demofunc.__doc__)  # 不加@functools.wraps裝飾器時："wrapper的說明文檔"。 加裝飾器時："demofunc的說明文檔"

更多關(guān)于Python相關(guān)內(nèi)容感興趣的讀者可查看本站專題：《Python面向?qū)ο蟪绦蛟O(shè)計入門與進階教程》、《Python數(shù)據(jù)結(jié)構(gòu)與算法教程》、《Python函數(shù)使用技巧總結(jié)》、《Python字符串操作技巧匯總》、《Python編碼操作技巧總結(jié)》及《Python入門與進階經(jīng)典教程》