首頁 > 軟體

python裝飾器底層原理詳解

2022-07-28 18:01:51

1 python裝飾器的作用

被裝飾物件加上裝飾器(戴了個帽子),被裝飾物件獲得了更強大的功能。

2 python裝飾器的原理

  • python裝飾器本身是一個函數
  • 這個函數的引數是一個函數物件
  • 這個函數的返回值也是一個函數物件,這個函數的功能更強
  • 大python裝飾器是python的一個語法糖(更簡便的語法)

3 python裝飾器的實現

3.1 最簡陋的裝飾器

# 自定義裝飾器
def super_(func):
    def wrapper():
        print('把內褲穿到外面來,變身超人')
        func()
        print('會飛!')
    return wrapper
@super_
def man():
    print('會走')
# @super_原始語法結構
# man = super_(man)
man()

執行結果如下:

3.2 給有返回值的函數加上裝飾器

'''
    給有返回值的函數加上裝飾器
'''
import time
def decorate(func):
    def wrapper():
        print('開始執行時間:' + time.strftime('%Y-%m-%d %H:%M:%S'))
        result = func()
        print('結束執行時間:' + time.strftime('%Y-%m-%d %H:%M:%S'))
        return result
    return wrapper
@decorate
def normal_func():
    time.sleep(1)
    print('normal_func執行中......')
    return 2 + 2
# @decorate原始語法結構
# normal_func = decorate(normal_func)
print(normal_func())

執行結果如下:

3.3 給有返回值和引數的函數加上裝飾器

'''
    給有返回值和引數的函數加上裝飾器
'''
import time
def decorate(func):
    def wrapper(*args, **kwargs):
        print('開始執行時間:' + time.strftime('%Y-%m-%d %H:%M:%S'))
        result = func(*args, **kwargs)
        print('結束執行時間:' + time.strftime('%Y-%m-%d %H:%M:%S'))
        return result
    return wrapper
@decorate
def normal_func1(a, b):
    time.sleep(1)
    print('normal_func1執行中......')
    return a + b
@decorate
def normal_func2(a, b, c):
    time.sleep(1)
    print('normal_func2執行中......')
    return a + b + c
# @decorate原始語法結構
# normal_func1 = decorate(normal_func1)
# normal_func2 = decorate(normal_func2)
print(normal_func1(1, 2))
print(normal_func2(1, 2, 3))

執行結果如下:

可變引數原理:

1 定義時使用可變引數:在函數定義時使用args,在函數呼叫時,所有未匹配到的位置引數,會被放到args這個元組當中。
在函數定義時使用**kwargs,在函數呼叫時,所有未匹配到的關鍵字引數,會被放到kwargs這個字典當中。
2 呼叫時使用可變引數:在函數呼叫時使用args,是把args這個元組解包,元組內的每個元素作為函數的位置引數傳遞。
在函數呼叫時使用**kwargs,是把kwargs這個字典解包,字典內的每個元素作為函數的關鍵字引數傳遞。

def test1(a, b, c, d):
    print(a+b+c+d)

print('傳統呼叫'.center(60, '='))
test1(1, 2, 3, 4)    # 位置引數
test1(b=2, c=3, d=4, a=1)    # 關鍵字引數

# 在函數定義時使用*args,在函數呼叫時,所有未匹配到的位置引數,會被放到args這個元組當中
# 在函數定義時使用**kwargs,在函數呼叫時,所有未匹配到的關鍵字引數,會被放到kwargs這個字典當中
def test2(*args, **kwargs):
    print(args)
    print(type(args))
    print(kwargs)
    print(type(kwargs))
print('定義時使用可變引數'.center(60, '='))
test2(1, 2, 3, 4, b=5, c=6)
# 在函數呼叫時使用*args,是把args這個元組解包,元組內的每個元素作為函數的位置引數傳遞。
# 在函數呼叫時使用**kwargs,是把kwargs這個字典解包,字典內的每個元素作為函數的關鍵字引數傳遞。
print('呼叫時使用可變引數'.center(60, '='))
test1(*(1, 2, 3, 4))
test1(**{'a':1, 'b':2, 'c':3, 'd':4})
test1(*(1, 2), **{'c':3, 'd':4})

執行結果如下:

3.4 讓我還是那個我

'''
    讓我還是那個我
'''
import time
from functools import wraps
def decorate(func):
    @wraps(func)    # 把wrapper的內建屬性轉換成func的內建屬性(name/doc)
    def wrapper(*args, **kwargs):
        print('開始執行時間:' + time.strftime('%Y-%m-%d %H:%M:%S'))
        result = func(*args, **kwargs)
        print('結束執行時間:' + time.strftime('%Y-%m-%d %H:%M:%S'))
        return result
    # wrapper.__name__ = func.__name__
    # wrapper.__doc__ = func.__doc__
    return wrapper
@decorate
def normal_func(a, b):
    '''
        這是一個測試函數
    '''
    time.sleep(1)
    print('normal_func1執行中......')
    return a + b
print(normal_func.__name__)
print(normal_func.__doc__)

4 python裝飾器在自動化測試框架中的應用

4.1 從一個需求開始

對自動化測試指令碼需要增加紀錄檔列印功能:

  • 測試用例執行前列印:測試用例【xxx】開始執行
  • 測試用例執行完列印:測試用例【xxx】執行完畢
  • 測試用例執行完列印:測試用例【xxx】執行耗時:xx秒
  • 測試用例執行完列印分割線
import time
from functools import wraps

def log_decorator(func):
    @wraps(func)    # 把wrapper的內建屬性轉換成func的內建屬性(name/doc)
    def wrapper(*args, **kwargs):
        print('測試用例[{}]開始執行'.format(func.__name__))
        time_start = time.time()
        result = func(*args, **kwargs)
        time_end = time.time()
        print('測試用例[{}]執行完畢'.format(func.__name__))
        print('測試用例[{}]執行耗時:%.2f秒'.format(func.__name__) % (time_end - time_start))
        print('分割線'.center(60, '='))
        return result
    return wrapper

到此這篇關於python裝飾器底層原理講解的文章就介紹到這了,更多相關python裝飾器底層原理講解內容請搜尋it145.com以前的文章或繼續瀏覽下面的相關文章希望大家以後多多支援it145.com!


IT145.com E-mail:sddin#qq.com