装饰函数以向函数添加自定义参数

Question

我有兴趣创建一个通用函数并用它装饰所有其他函数。新函数将包含两个函数的参数。示例

def decorator(func):
    Some chunk of code

@decorator
def func(a,b):
    print a**2,b**2
    return a**2,b**2

#Equivalent func of what I want out of the decorator
def equiv_func(a,b,var):
    print var # I want to modify all decorated functions to accept a new
              # argument and print it using "print var"
    print a**2, b**2
    return a**2, b**2

这是我想出来的，但是......

def decor(func):
    def modify(var,**kwargs):
       print var
       x,y=func(**kwargs)
       return x,y
    return modify

@decor
def func(a,b):
    print a**2,b**2
    return a**2,b**2

>>> func(var="hi",a=4,b=5)
hi
16 25

我使用 **kwargs 来解释 func 的参数，但这迫使我使用 func(var="hi",a=4,b=5) 而不是 func(a=4,b= 5,var="hi") 由于 **kwargs 的定位。

此外，这种解决方法可以防止我在需要时定义装饰器时使用 **kwargs，因为它已经用于将参数输入到 func 中。示例

def test(**kwargs):
    if 'test_ans' in kwargs:
        if kwargs['test_ans']=='y':
            return True
    else:
        return False

def decor(func):
    def modify(var,**kwargs):
       if test(**kwargs): # I need to use **kwargs here, but I also need it to define my  
                          # arguments in func.
           print var
       x,y=func(**kwargs)
       return x,y
    return modify

...
# The following is what I expect but due to **kwargs being used for two purposes, 
# it doesn't work
>>>func(var='hi',a=4,b=5,test_ans='y')
'hi'
16 25
>>>func(var='hi',a=4,b=5,test_ans='n')
16 25

我希望我的问题很清楚。这些例子只是为了说明我面临的限制。它们不是实际代码（例如，我不会编写使用 **kwargs 的测试函数） 提前感谢您的任何帮助，编码大师！

编辑： 或者简单地说，有没有其他方法可以在不使用装饰器中的kwargs的情况下将func(a,b)修改为func(a,b,c,d,e,...,**kwargs)？

Answer 1

试试这个。您可以使用func.func_code.co_varnames 访问函数接受的参数。一旦知道这一点，您就可以过滤掉其他所有内容。

编辑：这是有限的，如果myfunc 接受**kwargs 作为参数，您将无法有效地使用它。

def print_decorator(func):
    def wrapped(*args, **kwargs):
        # a tuple of the names of the parameters that func accepts
        func_params = func.func_code.co_varnames
        # grab all of the kwargs that are not accepted by func
        extra = set(kwargs.keys()) - set(func_params)

        for kw in extra:
            print kwargs.pop(kw)

        return func(*args, **kwargs)

    return wrapped

@print_decorator
def myfunc(a, b, c=None):
    print 'myfunc was passed: a={}, b={}, c={}'.format(a, b, c)
    return a, b, c

>>> myfunc(1, b=2, c=3, d=4, e=5)
4
5
myfunc was passed: a=1, b=2, c=3

Answer 2

你可以允许func被调用

func(a=4, b=5, var="hi") 

通过将var 设为关键字参数。

def decor(func):
    def modify(*args, **kwargs):
        var = kwargs.pop('var', None)
        print var
        x,y=func(*args, **kwargs)
        return x,y
    return modify

@decor
def func(a,b):
    print a**2,b**2
    return a**2,b**2

func(a=4, b=5, var="hi")
func(a=4, b=5)