我试图在
python中将可选参数传递给我的类装饰器.
在我现在的代码下面
在我现在的代码下面
class Cache(object):
def __init__(self,function,max_hits=10,timeout=5):
self.function = function
self.max_hits = max_hits
self.timeout = timeout
self.cache = {}
def __call__(self,*args):
# Here the code returning the correct thing.
@Cache
def double(x):
return x * 2
@Cache(max_hits=100,timeout=50)
def double(x):
return x * 2
具有覆盖默认值的参数的第二个装饰器(max_hits = 10,__init__函数中的timeout = 5)不起作用,我得到异常TypeError：__init __()至少有两个参数(3个给定).我尝试了许多解决方案并阅读有关它的文章,但在这里我仍然无法使其工作.
有什么想法来解决吗？谢谢！
解决方法
@Cache(max_hits = 100,timeout = 50)调用__init __(max_hits = 100,timeout = 50),所以你不满足函数参数.
您可以通过一个检测函数是否存在的包装方法来实现您的装饰器.如果它找到一个函数,它可以返回Cache对象.否则,它可以返回将用作装饰器的包装器函数.
class _Cache(object):
def __init__(self,*args):
# Here the code returning the correct thing.
# wrap _Cache to allow for deferred calling
def Cache(function=None,timeout=5):
if function:
return _Cache(function)
else:
def wrapper(function):
return _Cache(function,max_hits,timeout)
return wrapper
@Cache
def double(x):
return x * 2
@Cache(max_hits=100,timeout=50)
def double(x):
return x * 2