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

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