python如何定義帶參數(shù)的裝飾器
本文實例為大家分享了python定義帶參數(shù)裝飾器的具體代碼,供大家參考,具體內容如下
案例:
實現(xiàn)一個裝飾器,用它來檢查被裝飾函數(shù)的參數(shù)類型。
需求:
裝飾器可以通過函數(shù),指明函數(shù)參數(shù)類型,進行函數(shù)調用的時候,傳入?yún)?shù),檢測到不匹配時,拋出異常
如何解決這個問題?
先要獲取函數(shù)的簽名,并且獲得裝飾器中參數(shù),然后把函數(shù)簽名和裝飾器中參數(shù)對應綁定
把調用函數(shù)時候傳入的參數(shù)和函數(shù)簽名進行綁定
把實參和裝飾器中定義的數(shù)據(jù)進行類型比較,不匹配拋出異常
#!/usr/bin/python3
from inspect import signature
def check_type(*ty_args, **ty_kwargs):
def out_wrapper(func):
# 通過signature方法,獲取函數(shù)形參:name, age, height
sig = signature(func)
# 獲得裝飾器傳來的參數(shù), 函數(shù)簽名與之綁定,字典類型
bind_types = sig.bind_partial(*ty_args, **ty_kwargs).arguments
print(bind_types)
def wrapper(*args, **kwargs):
# 給執(zhí)行函數(shù)中具體的實參進行和形參進行綁定,形成字典的形式
func_type = sig.bind(*args, **kwargs).arguments.items()
print(func_type)
# 循環(huán)形參和實參字典的items()形式
for name, obj in func_type:
if name in bind_types:
if not isinstance(obj, bind_types[name]):
raise TypeError('%s must be %s' % (name, bind_types[name]))
func(*args, **kwargs)
return wrapper
return out_wrapper
# 通過裝飾器實現(xiàn)對函數(shù)參數(shù)進行類型檢查
@check_type(str, int, float)
def func(name, age, height):
print(name, age, height)
if __name__ == '__main__':
func('bei_men', 18, 1.75)
以上就是本文的全部內容,希望對大家的學習有所幫助,也希望大家多多支持腳本之家。
相關文章
Python使用pydub庫對mp3與wav格式進行互轉的方法
今天小編就為大家分享一篇Python使用pydub庫對mp3與wav格式進行互轉的方法,具有很好的參考價值,希望對大家有所幫助。一起跟隨小編過來看看吧2019-01-01
判斷python對象是否可調用的三種方式及其區(qū)別詳解
今天小編就為大家分享一篇判斷python對象是否可調用的三種方式及其區(qū)別詳解,具有很好的參考價值,希望對大家有所幫助。一起跟隨小編過來看看吧2019-01-01

