Python中的函數(shù)參數(shù)類型檢查
Python函數(shù)參數(shù)類型檢查
有一個很經(jīng)典的笑話：
三個月之前，只有我和上帝知道這代碼是干什么的。
現(xiàn)在，只有上帝知道了。
在Python中，不知道函數(shù)參數(shù)類型是一個很正常的事情，特別是在一個大項(xiàng)目里。
我見過有些項(xiàng)目里，每一個函數(shù)體的前十幾行都在檢查參數(shù)類型，這實(shí)在是太麻煩了。而且一旦參數(shù)有改動，這部分也需要改動。
下面我們用裝飾器來實(shí)現(xiàn)
函數(shù)參數(shù)的強(qiáng)制類型檢查。
首先，這個裝飾器，要接受類型參數(shù)，和指定函數(shù)參數(shù)的類型參數(shù)。也就是一個list和一個dict
from functools import wraps def typeassert(*type_args, **type_kwargs): ? ? def decorate(func): ? ? ? ? @wraps(func) ? ? ? ? def wrapper(*args, **kwargs): ? ? ? ? ? ? return func(*args, **kwargs) ? ? ? ? return wrapper ? ? return decorate
@wraps(func)的作用請看我的另一篇
那么，接下來，在裝飾器中，我們需要獲取函數(shù)參數(shù)列表，并且要和類型參數(shù)表映射。
這要借助Python的一個標(biāo)準(zhǔn)庫——inspect 這個庫一般用于Python代碼調(diào)試
from inspect import signature from functools import wraps def typeassert(*type_args, **type_kwargs): ? ? def decorate(func): ? ? ? ? sig = signature(func) ? ? ? ? bound_types = sig.bind_partial(*type_args, **type_kwargs).arguments ? ? ? ? @wraps(func) ? ? ? ? def wrapper(*args, **kwargs): ? ? ? ? ? ? return func(*args, **kwargs) ? ? ? ? return wrapper ? ? return decorate
上面的代碼中，我們使用inspect中的signature方法獲取了func的Signature對象，然后使用bind_partial方法創(chuàng)建了(*type_args, **type_kwargs)到func參數(shù)的映射(也就是一個字典)。
接下來就簡單了，我們只需要再獲取(*args, **kwargs)的類型，使用isintance函數(shù)進(jìn)行比較就好。
from inspect import signature
from functools import wraps
def typeassert(*type_args, **type_kwargs):
? ? def decorate(func):
? ? ? ? sig = signature(func)
? ? ? ? bound_types = sig.bind_partial(*type_args, **type_kwargs).arguments
? ? ? ? @wraps(func)
? ? ? ? def wrapper(*args, **kwargs):
? ? ? ? ? ? bound_values = sig.bind(*args, **kwargs)
? ? ? ? ? ? for name, value in bound_values.arguments.items():
? ? ? ? ? ? ? ? if name in bound_types:
? ? ? ? ? ? ? ? ? ? if not isinstance(value, bound_types[name]):
? ? ? ? ? ? ? ? ? ? ? ? raise TypeError('Argument {} must be {}'.format(name, bound_types[name]))
? ? ? ? ? ? return func(*args, **kwargs)
? ? ? ? return wrapper
? ? return decorate運(yùn)行如下代碼
@typeassert(int, int)
def add(x, y):
? ? return x+y
print(add("u", 2))能看到報(bào)錯如下
Traceback (most recent call last):
File "c:\Users\Chen\Desktop\typeassert.py", line 32, in <module>
print(add("u", 2))
File "c:\Users\Chen\Desktop\typeassert.py", line 22, in wrapper
'Argument {} must be {}'.format(name, bound_types[name])
TypeError: Argument x must be <class 'int'>
很貼心的提醒了我們哪一個參數(shù)應(yīng)該是什么類型。
你甚至可以自己改動這個裝飾器，讓它還能告訴你傳進(jìn)去了什么錯誤參數(shù)(特別是寫爬蟲的時候，參數(shù)很難掌握，一旦報(bào)錯，還得重跑一遍才知道為什么。)
你也可以指定某一個參數(shù)的類型，譬如
@typeassert(int, z=str) def display(x, y, z): ? ? print(x, y, z)
這時你會發(fā)現(xiàn)，y的類型就像原生的Python函數(shù)一樣，什么都行。而x必須是int，z必須是str。
總結(jié)
以上為個人經(jīng)驗(yàn)，希望能給大家一個參考，也希望大家多多支持腳本之家。
相關(guān)文章
這篇文章主要為大家介紹了簡單的Python解密rsa案例詳解，有需要的朋友可以借鑒參考下，希望能夠有所幫助，祝大家多多進(jìn)步，早日升職加薪2022-07-07
python使用 cx_Oracle 模塊進(jìn)行查詢操作示例
這篇文章主要介紹了python使用 cx_Oracle 模塊進(jìn)行查詢操作,結(jié)合實(shí)例形式分析了Python使用cx_Oracle模塊進(jìn)行數(shù)據(jù)庫的基本連接、查詢、輸出等相關(guān)操作技巧,需要的朋友可以參考下2019-11-11
這篇文章主要介紹了用python語言獲得圖片的Base64編碼的示例，大家參考使用吧2014-01-01
這篇文章主要介紹了Python查詢IP地址歸屬完整代碼，具有一定的參考價值，感興趣的小伙伴們可以參考一下2017-06-06
關(guān)于Python 解決Python3.9 pandas.read_excel(‘xxx.xlsx‘)報(bào)錯的問題
這篇文章主要介紹了關(guān)于Python 解決Python3.9 pandas.read_excel(‘xxx.xlsx‘)報(bào)錯的問題,本文給大家介紹的非常詳細(xì)，對大家的學(xué)習(xí)或工作具有一定的參考借鑒價值，需要的朋友可以參考下2020-11-11
python 利用pyttsx3文字轉(zhuǎn)語音過程詳解
這篇文章主要介紹了python 利用pyttsx3文字轉(zhuǎn)語音過程詳解,文中通過示例代碼介紹的非常詳細(xì)，對大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價值,需要的朋友可以參考下2019-09-09
在學(xué)習(xí)Python爬蟲部分，需要已經(jīng)學(xué)過Python基礎(chǔ)和前端的相關(guān)知識，本文對python爬蟲概念及原理給大家詳細(xì)介紹，需要的朋友跟隨小編一起看看吧2021-04-04
Python實(shí)現(xiàn)手繪圖效果實(shí)例分享
在本篇文章里小編給大家整理了關(guān)于Python實(shí)現(xiàn)手繪圖效果，有需要的朋友們可以學(xué)習(xí)下。2020-07-07
之前我們介紹過字典的實(shí)現(xiàn)原理，它底層是基于哈希表實(shí)現(xiàn)的，而集合也是如此。本次我們來聊一下 Python 的集合是怎么實(shí)現(xiàn)的，希望對大家有所幫助2022-12-12
Python descriptor(描述符)的實(shí)現(xiàn)
這篇文章主要介紹了Python descriptor(描述符)的實(shí)現(xiàn)，文中通過示例代碼介紹的非常詳細(xì)，對大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價值，需要的朋友們下面隨著小編來一起學(xué)習(xí)學(xué)習(xí)吧2020-11-11