结合显式关键字参数和 **kwargs

Question

在 Python 中，我可以显式列出函数接受的仅限关键字的参数：

def foo(arg, *, option_a=False, option_b=False):
    return another_fn(arg, option_a=option_a, option_b=option_b)

虽然调用另一个函数的语法有点冗长，但我确实在函数调用中对命名参数进行了很好的自动完成：

我可以通过使用**kwargs来消除函数调用中的冗余：

def foo(arg, **kwargs):
    return another_fn(arg, **kargs)

但后来我失去了编辑器中的自动完成功能（并且函数签名不太清晰）：

我怎样才能两全其美？我想写一个显式的函数签名，但仍然在函数体中得到一个kwargs dict。

Answer 1

这是locals()的可能使用场景：

def foo(arg, arg2=None):
    return another_fn(**locals())

如果需要，也可以解压self：

class bar:
    def foo(self, arg):
        kwargs = locals()
        kwargs.pop('self')
        return another_fn(**kwargs)

警告：如果您在调用 locals() 之前在方法中定义了任何局部变量，它们最终会出现在输出中，您必须将它们过滤掉。

def foo(self, arg=None):
    baz = 'Oops'
    kwargs = locals() # {'self': <bar at 0x0000>, 'arg': None, 'baz': 'Oops'}
    return another_fn(**kwargs)

简单的解决方案是在执行任何其他操作之前调用locals()，为安全起见复制一份，并记住在实例方法中清理self：

class bar:
    def foo(self, arg=None):
        kwargs = locals().copy() # {'self': <bar at 0x0000>, 'arg': None}
        kwargs.pop('self')
        baz = 'Oops'
        return another_fn(**kwargs)

希望这会有所帮助！