[英]Introspect function to determine which argument unpacking (positional or keyword) was used
我正在尋找一種方法來確定是否有一些參數用於解包,我發現了這個:
>>> def func_has_positional_args(func):
std_args = func.func_code.co_argcount
wildcard_args = len(func.func_code.co_varnames) - std_args
if wildcard_args == 2:
return True # yes, has both positional and keyword args
elif wildcard_args == 0:
return False # has neither positional, nor keyword args
else:
raise NotImplementedError('Unable to tell')
>>> func_has_keyword_args = func_has_positional_args
>>> def test1(a, b, *args, **kwargs): pass
>>> func_has_positional_args(test1), func_has_keyword_args(test1)
(True, True)
>>> def test2(a, b): pass
>>> func_has_positional_args(test2), func_has_keyword_args(test2)
(False, False)
>>> def test3(a, b, *args): pass
>>> func_has_positional_args(test3)
Traceback (most recent call last):
File "<pyshell#52>", line 1, in <module>
func_has_positional_args(test3)
File "<pyshell#41>", line 9, in func_has_positional_args
raise NotImplementedError('Unable to tell')
NotImplementedError: Unable to tell
所以我能夠告訴,如果沒有位置,也沒有關鍵字參數解包。 我也能夠判斷是否存在兩者,但是如果只有一個“通配符”,我無法區分實現了哪個“通配符”類型參數。
你能幫我實現以下結果嗎?
# Already satisfied with above code:
assert func_has_positional_args(test1) == True
assert func_has_keyword_args(test1) == True
assert func_has_positional_args(test2) == False
assert func_has_keyword_args(test2) == False
# Missing functionality (tests are failing):
assert func_has_positional_args(test3) == True
assert func_has_keyword_args(test3) == False
此外,Python 3是否會針對此功能或其行為進行更改?
正如mgilson所評論的那樣,使用inspect.getargspec
(更好的是Python 3.x中的inspect.getfullargspec
)。
import inspect
def func_has_positional_args(func):
spec = inspect.getfullargspec(func)
return bool(spec.varargs) # varargs: name of the * argument or None
def func_has_keyword_args(func):
spec = inspect.getfullargspec(func)
return bool(spec.varkw) # varkw: name of the ** argument or None
例:
>>> def test1(a, b, *args, **kwargs): pass
...
>>> def test2(a, b): pass
...
>>> def test3(a, b, *args): pass
...
>>> func_has_positional_args(test1)
True
>>> func_has_keyword_args(test1)
True
>>> func_has_positional_args(test2)
False
>>> func_has_keyword_args(test2)
False
>>> func_has_positional_args(test3)
True
>>> func_has_keyword_args(test3)
False
聲明:本站的技術帖子網頁,遵循CC BY-SA 4.0協議,如果您需要轉載,請注明本站網址或者原文地址。任何問題請咨詢:yoyou2525@163.com.