日本免费高清视频-国产福利视频导航-黄色在线播放国产-天天操天天操天天操天天操|www.shdianci.com

學無先后，達者為師

網(wǎng)站首頁 編程語言 正文

Python中的函數(shù)參數(shù)類型檢查_python

作者：子休_ ? 更新時間： 2023-01-18 編程語言

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。

總結(jié)

原文鏈接：https://www.jianshu.com/p/7a2c9133a002

欄目分類
最近更新