這篇文章主要介紹了Python 裝飾器@，對函數進行功能擴展操作,結合實例形式分析了裝飾器的相關使用技巧,以及開閉原則下的函數功能擴展,需要的朋友可以參考下：
本文實例講述了Python 裝飾器@，對函數進行功能擴展操作。分享給大家供大家參考，具體如下：
裝飾器可以對原函數進行功能擴展，但還不需要修改原函數的內容(開閉原則)，也不需要修改原函數的調用。
demo.py（裝飾器，@）：
# 閉包
def w1(func):
def inner():
# 對原函數進行功能擴展
print("功能擴展")
func()
# return func() # 如果原函數需要返回值,可以return
return inner # 閉包
@w1
# 相當於 f1 = w1(f1)
def f1():
print('f1') # 原函數不需要修改
f1() # 原函數的調用也不需要修改
我是一名python開發工程師，整理了一套python的學習資料，從基礎的python腳本到web開發、爬蟲、
數據分析、數據可視化、機器學習、面試真題等。想要的可以進羣：688244617免費領取
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, *args, **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程序設計有所幫助。