Question

对于我正在编写的装饰器，我想操作函数的特定命名参数。考虑以下装饰者：

def square_param(param):
    def func_decorator(func):
        def func_caller(*args,**kwargs):
            kwargs[param] = kwargs[param] * kwargs[param]
            return func(*args,**kwargs)
    return func_caller
return func_decorator

应用于下一个功能：

@square_param('dividend')
def quotient(divisor=1,dividend=0):
    return dividend/divisor

如果将被除数称为关键字参数，例如：。

，这将有效

>>> quotient(dividend=2)
4

但是，当作为位置参数给出时，这将失败。

>>> quotient(3,4)
TypeError: quotient() got multiple values for keyword argument 'dividend'

使用Python 3，我可以通过强制参数为always given as a keyword来解决这个问题：

@square_param('dividend')
def quotient(divisor=1,*,dividend=0):
    return dividend/divisor

但是我想支持Python 2，我也希望对该功能施加一些限制。

有没有办法可以在我的装饰器中解决这个问题？

Answer 1

首先，您的square_param装饰器不起作用，因为它不返回功能。它必须是：

def square_param(param):
    def func_decorator(func):
        def func_caller(*args,**kwargs):
            kwargs[param] = kwargs[param] * kwargs[param]
            return func(*args,**kwargs)
        return func_caller
    return func_decorator

现在我接受了@Dirk的建议并查看了inspect模块。您可以通过首先检查参数是否是函数的位置参数之一来实现，如果已经指定了该位置参数，则可以执行此操作，然后修改该参数位置。如果参数作为关键字参数提供，您还需要确保只修改kwargs。

import inspect

def square_param(param):
    def func_decorator(func):
        def func_caller(*args,**kwargs):
            funparams = inspect.getargspec(func).args
            if param in funparams:
                i = funparams.index(param)
                if len(args) > i:
                    args = list(args)   # Make mutable
                    args[i] = args[i] * args[i]
            if param in kwargs:
                kwargs[param] = kwargs[param] * kwargs[param]
            return func(*args,**kwargs)
        return func_caller
    return func_decorator

Answer 2

即使不使用Inspect，我们也可以获得函数参数

>>> func = lambda x, y, args: (x, y, {})
>>> func.func_code.co_argcount
3
>>> func.func_code.co_varnames
('x', 'y', 'args')

Answer 3

这可能仅与切线相关，但我发现解决类似问题很有用。我想将*args和**kwargs合并到一个字典中，这样我的后续代码就可以处理而不必考虑args的输入方式了，我也不想改变现有的kwargs变量，否则我只会使用kwargs.update()。

all_args = {**kwargs, **{k: v for k, v in zip(list(inspect.signature(func).parameters), args)}}

# optionally delete `self`
del (all_args['self'])

更新：在此可行的同时，this answer具有更好的技术。部分：

bound_args = inspect.signature(f).bind(*args, **kwargs)
bound_args.apply_defaults()

如何在Python 2中将位置参数视为关键字参数

3 个答案: