1 裝飾器背景知識

1.1 基本概念

裝飾器（Decorator）是 Python 中一種函數(shù)或類，用來修飾其他函數(shù)或類。裝飾器可以改變被裝飾函數(shù)的行為，或者在調(diào)用被裝飾函數(shù)之前和之后增加額外的操作。裝飾器的語法是使用 @ 語法符，在函數(shù)定義之前增加裝飾器函數(shù)的名稱。

@decorator_func
def my_func():
    pass

1.2 應(yīng)用場景

• 代碼重用：裝飾器可以讓我們在不更改原函數(shù)代碼的情況下，為其添加額外的功能。
• 日志記錄：裝飾器可以記錄函數(shù)的調(diào)用日志，幫助我們追蹤程序的運(yùn)行情況。
• 權(quán)限控制：裝飾器可以用來實(shí)現(xiàn)函數(shù)級別的權(quán)限控制，只允許特定的用戶訪問特定的函數(shù)。
• 緩存：裝飾器可以用來緩存函數(shù)的返回值，避免重復(fù)計(jì)算。
• 類型檢查：裝飾器可以用來在函數(shù)調(diào)用前檢查參數(shù)的類型是否符合要求。
• 裝飾器可以讓你在函數(shù)或類的定義中添加額外的邏輯，而不更改它們的實(shí)現(xiàn)。

2 簡單的裝飾器代碼

def decorator_func(func):
    def wrapper():
        print("Before calling the function")
        func()
        print("After calling the function")
    return wrapper
@decorator_func
def my_func():
    print("Inside the function")
my_func()
# Output: Before calling the function
#         Inside the function
#         After calling the function

上面展示了最簡單的裝飾器示例代碼。在代碼中，我們建立了一個(gè)名為decorator_func的裝飾器和一個(gè)名為my_func函數(shù)。

• 裝飾器外部的return必須為裝飾器的內(nèi)部函數(shù)，不含括號。通過代碼結(jié)構(gòu)可以看出，裝飾器本身也是一個(gè)閉包。
• 在定義裝飾器decorator_func時(shí)，括號中的’func’指代被裝飾器裝飾的函數(shù)，在這段代碼中指代的就是my_func函數(shù)。
• 在被裝飾器裝時(shí)候，函數(shù)的實(shí)際執(zhí)行執(zhí)行順序變成了內(nèi)部函數(shù)wrapper所指定的順序。即先執(zhí)行print(“Before calling the function”)；再執(zhí)行func()指代的my_func函數(shù)；最后執(zhí)行print(“After calling the function”)。
• 本段代碼的最終輸出為：Before calling the function；Inside the function；After calling the function

3 使用裝飾器記錄函數(shù)執(zhí)行次數(shù)

def cal_times(func):
    l=[]
    def wrapper(*var):
        l.append('1')
        func(*var)
        print("函數(shù)執(zhí)行了%s次"%(len(l)))
    return wrapper
@cal_times
def my_func(i):
    print('%s的平方是%s'%(i,i**2))
my_func(5)
my_func(6)

在my_func(i)函數(shù)中，我們增加了形參的輸入，因此，在裝飾器中，也要為之做出更改。此處裝飾器中的wrapper函數(shù)我們使用*var傳參，這種設(shè)計(jì)方式的優(yōu)點(diǎn)是可以讓這個(gè)裝飾器適用于任何函數(shù)。

再加入了cal_times裝飾器后，函數(shù)每運(yùn)行一次，都會(huì)使列表l添加一個(gè)1，這樣可以計(jì)算函數(shù)的運(yùn)行次數(shù)。這段代碼的運(yùn)行結(jié)果如下：

4 帶參數(shù)的裝飾器

裝飾器與函數(shù)一樣，也可以帶入?yún)?shù)，我們在第二節(jié)的基礎(chǔ)上，對代碼做出如下修改：

def decorator_func(param1, param2):
    def decorator(func):
        def wrapper():
            print("Before calling the function with params:", param1,param2)
            func()
            print("After calling the function")
        return wrapper
    return decorator
@decorator_func("hello", "world")
def my_func():
    print("Inside the function")
my_func()

這段代碼使用了裝飾器來在my_func函數(shù)調(diào)用前后打印額外的信息，并且裝飾器函數(shù)decorator_func接受兩個(gè)參數(shù)，在調(diào)用wrapper函數(shù)時(shí)使用這兩個(gè)參數(shù)。

最終，代碼將輸出：

Before calling the function with params: hello world； Inside the function；After calling the function。

如果同時(shí)還有字典類型的參數(shù)傳入，可以使用（*var,**_var）進(jìn)行解決

5 裝飾器處理有返回值的函數(shù)

前面我們定義的函數(shù)都是執(zhí)行某種功能，不涉及到return的相關(guān)操作。當(dāng)涉及到處理有返回值的函數(shù)時(shí)，對于內(nèi)部函數(shù)我們應(yīng)該使用一個(gè)變量將函數(shù)的運(yùn)行結(jié)果保存起來，并放在內(nèi)層函數(shù)的return中。為了實(shí)現(xiàn)這一功能，我們將第三部分的代碼做出如下修改：

def cal_times(func):
    l=[]
    def wrapper(*var):
        l.append('1')
        result = func(*var)
        print("函數(shù)執(zhí)行了%s次"%(len(l)))
        return result
    return wrapper
@cal_times
def my_func(i):
    print('%s的平方是%s'%(i,i**2))
    return i**2
a = my_func(5)
b = my_func(6)
print(a,b)

對于這個(gè)裝飾器，我們在內(nèi)部函數(shù)wrapper使用result保存運(yùn)行結(jié)果，并將result return，這樣a與b就可以被正常的賦值，運(yùn)行結(jié)果如下圖。

而如果不執(zhí)行保存result并return，a和b將不會(huì)得到任何值：

到此這篇關(guān)于Python裝飾器使用方法全面梳理的文章就介紹到這了,更多相關(guān)Python裝飾器內(nèi)容請搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家！

最新評論