python - 模拟函数参数分配给 *args 和 **kwargs

Question

有没有一种简单的方法可以“模拟”参数对函数参数的赋值？

假设我有一个函数

def fn(a, b, c='default', *args, **kwargs):
    print(f'a={a}, b={b}, c={c}')
    print(f'args={args}')
    print(f'kwargs={kwargs}')

我这样调用它:

>>> args = (1, 2, 3, 4)
>>> kwargs = {'extra_kw': 5}
>>> fn(*args, **kwargs)
a=1, b=2, c=3
args=(4,)
kwargs={'extra_kw': 5}

我想提取函数 fn 中的 args 和 kwargs，而不调用 fn。

基本上，我正在尝试编写一个函数，它将一个函数和一个参数列表和字典作为输入，并返回额外的 args 和 kwargs:

def extract_extra_args_and_kwargs(f, *args, **kwargs):
    ...

该示例的预期结果为:

>>> extract_extra_args_and_kwargs(fn, args, kwargs)
((4,), {'extra_kw': 5})

我试过使用 inspect.signature，但考虑到 Python 本身必须在某处解析此赋值，我希望我可以直接访问相应的函数。

Answer 1

请注意，自 Python 3.5 以来，inspect.getcallargs 已被弃用，根据 documentation .

继续使用 inspect.signature.bind:

from inspect import signature

def extract_extra_args_and_kwargs(f, *args, **kwargs):
    sig = signature(f)
    bound = sig.bind(*args, **kwargs)
    return bound.arguments['args'], bound.arguments['kwargs']

def fn(a, b, c='default', *args, **kwargs):
    print(f'a={a}, b={b}, c={c}')
    print(f'args={args}')
    print(f'kwargs={kwargs}')
args = (1, 2, 3, 4)
kwargs = {'extra_kw': 5}
print(extract_extra_args_and_kwargs(fn, *args, **kwargs))

这个输出:

((4,), {'extra_kw': 5})