将参数传递给装饰器函数

Question

谁能告诉我如何将参数传递给装饰器调用函数？

def doubleIt(Onefunc):
    def doubleIn():
        return Onefunc()*Onefunc()
    return doubleIn

@doubleIt
def Onefunc():  
    return 5

print(Onefunc()) # it prints out 25. 

但是，当我尝试将我的 Onefunc() 升级到：

@doubleIt
def Onefunc(x):
    return x

我面临以下错误：

TypeError                                 
Traceback (most recent call last)
<ipython-input-17-6e2b55c94c06> in <module>()
      9 
     10 
---> 11 print(Onefunc(5))
     12 

TypeError: doubleIn() takes 0 positional arguments but 1 was given

错误是不言自明的，但我不确定如何更新 doubleIn() 函数来处理它。

Answer 1

您需要传入可选的位置和关键字参数。

from functools import wraps

def doubleIt(Onefunc):

    @wraps(Onefunc)
    def doubleIn(*args, **kwargs):
        return Onefunc(*args, **kwargs) * Onefunc(*args, **kwargs)
    return doubleIn

@doubleIt
def Onefunc(x):
    return x

print(Onefunc(5))

Answer 2

如果您在Onefunc() 中设置参数，您还应该在doubleIn() 函数中传递一个参数：

def doubleIt(Onefunc):
    def doubleIn(x):
        return Onefunc(x)*Onefunc(x)
    return doubleIn

@doubleIt
def Onefunc(x):
    return x

print(Onefunc(5))