反思功能，以確定哪些參數拆包（位置或關鍵字）用於

Question

我正在尋找一種方法來找出一些論據是否用於拆包，我發現這一點：

>>> 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是否改變任何關於此功能或其行爲？

Answer 1

As 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