Python中的函數(shù)參數(shù)類型檢查
Python函數(shù)參數(shù)類型檢查
有一個很經(jīng)典的笑話：
三個月之前，只有我和上帝知道這代碼是干什么的。
現(xiàn)在，只有上帝知道了。
在Python中，不知道函數(shù)參數(shù)類型是一個很正常的事情，特別是在一個大項目里。
我見過有些項目里，每一個函數(shù)體的前十幾行都在檢查參數(shù)類型，這實在是太麻煩了。而且一旦參數(shù)有改動，這部分也需要改動。
下面我們用裝飾器來實現(xiàn)
函數(shù)參數(shù)的強制類型檢查。
首先，這個裝飾器，要接受類型參數(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的一個標準庫——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ù)進行比較就好。
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運行如下代碼
@typeassert(int, int)
def add(x, y):
? ? return x+y
print(add("u", 2))能看到報錯如下
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ù)應該是什么類型。
你甚至可以自己改動這個裝飾器，讓它還能告訴你傳進去了什么錯誤參數(shù)(特別是寫爬蟲的時候，參數(shù)很難掌握，一旦報錯，還得重跑一遍才知道為什么。)
你也可以指定某一個參數(shù)的類型，譬如
@typeassert(int, z=str) def display(x, y, z): ? ? print(x, y, z)
這時你會發(fā)現(xiàn)，y的類型就像原生的Python函數(shù)一樣，什么都行。而x必須是int，z必須是str。
總結
以上為個人經(jīng)驗，希望能給大家一個參考，也希望大家多多支持腳本之家。
相關文章
tkinter如何實現(xiàn)label超鏈接調(diào)用瀏覽器打開網(wǎng)址
這篇文章主要介紹了tkinter如何實現(xiàn)label超鏈接調(diào)用瀏覽器打開網(wǎng)址問題，具有很好的參考價值，希望對大家有所幫助。如有錯誤或未考慮完全的地方，望不吝賜教2023-01-01
這篇文章主要介紹了python通過實例講解反射機制,文中通過示例代碼介紹的非常詳細，對大家的學習或者工作具有一定的參考學習價值,需要的朋友可以參考下2019-10-10
anaconda虛擬環(huán)境python?sklearn庫的安裝過程
Anaconda是專注于數(shù)據(jù)分析的Python發(fā)行版本,包含了conda、Python等190多個科學包及其依賴項,這篇文章主要給大家介紹了關于anaconda虛擬環(huán)境python?sklearn庫的安裝過程,需要的朋友可以參考下2023-11-11
這篇文章主要為大家學習介紹了Flask如何實現(xiàn)定制日志并輸出到文件，文中的示例代碼簡介易懂，感興趣的小伙伴快跟隨小編一起學習一下吧2023-07-07
這篇文章主要介紹了tensorflow模型轉ncnn的操作方式，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧2020-05-05
這篇文章主要介紹了如何使用Python?繪制瀑布圖，我們一起了解瀑布圖的重要性，以及如何使用不同的繪圖庫繪制瀑布圖。瀑布圖是一種二維圖表，專門用于了解隨著時間或多個步驟或變量的增量正負變化的影響，下文更多詳細內(nèi)容需要的小伙伴可以參考一下2022-05-05
淺談flask中的before_request與after_request
這篇文章主要介紹了淺談flask中的before_request與after_request，小編覺得還是挺不錯的，具有一定借鑒價值，需要的朋友可以參考下2018-01-01
Python?PyJWT庫簡化JSON?Web?Token的生成與驗證
PyJWT庫為Python開發(fā)者提供了簡便的生成和驗證JWT的工具，本文將深入介紹PyJWT庫的核心概念、功能以及實際應用，通過豐富的示例代碼，幫助大家更全面地了解和應用這一強大的JWT庫2023-12-12
PyTorch是一個開源的Python機器學習庫，基于Torch，用于自然語言處理等應用程序。它具有強大的GPU加速的張量計算和自動求導系統(tǒng)的深度神經(jīng)網(wǎng)絡。本文將介紹如何通過PyTorch實現(xiàn)病蟲害圖像分類，感興趣的可以學習一下2021-12-12
這篇文章主要介紹了Python Socket編程的相關資料，幫助大家更好的理解和學習使用python，感興趣的朋友可以了解下2021-04-04