Question

我正在尝试编写一个装饰器以重复N次错误函数，并且之间的睡眠时间不断增加。到目前为止，这是我的尝试：

def exponential_backoff(seconds=10, attempts=10):
    def our_decorator(func):
        def function_wrapper(*args, **kwargs):
            for s in range(0, seconds*attempts, attempts):
                sleep(s)
                try:
                    return func(*args, **kwargs)
                except Exception as e:
                    print(e)
        return function_wrapper
    return our_decorator

@exponential_backoff
def test():    
    for a in range(100):
        if a - random.randint(0,1) == 0:
            print('success count: {}'.format(a))
            pass
        else:
            print('error count {}'.format(a))
            'a' + 1

test()

我不断收到错误消息：

TypeError: our_decorator() missing 1 required positional argument: 'func'

Answer 1

您应该使用：

@exponential_backoff()
def test():
    ...

整体装饰器的设计并非使参数为可选，因此在使用时必须提供()。

如果想要使装饰器允许参数列表为可选的示例，请参见：

https://wrapt.readthedocs.io/en/latest/decorators.html#decorators-with-optional-arguments

您还可以考虑使用wrapt包使装饰器更轻松，更坚固。

Answer 2

您可以选择@Graham Dumpleton提供的解决方案，也可以像这样修改装饰器：

    selim@xt.com
    sergey@xt.com
    billy@xt.com

编辑我的答案是并不完全正确，请参阅@GrahamDumpleton的答案，该答案显示了如何使我的解决方案的尝试可行（即this link）。立即修复，谢谢@GrahamDumpleton！

Answer 3

了解什么是装饰器

@exponential_backoff
def test():
    pass

等于：

def test():
    pass

test = exponential_backoff(test)

在这种情况下，test是def our_decorator(func):。这就是为什么您在呼叫TypeError时得到test()的原因。

更进一步：

@exponential_backoff()
def test():
    pass

等于：

def test():
    pass

test = exponential_backoff()(test)

在这种情况下，现在您需要的是test。

此外，functools.wraps帮助您将原始功能的所有属性复制到修饰的功能。如函数名称或文档字符串：

from functools import wraps

def exponential_backoff(func):
#   @wraps(func)
    def function_wrapper(*args, **kwargs):
        pass
    return function_wrapper

@exponential_backoff
def test():
    pass

print(test)  # <function exponential_backoff.<locals>.function_wrapper at 0x7fcc343a4268>
# uncomment `@wraps(func)` line:
print(test)  # <function test at 0x7fcc343a4400>

python装饰器TypeError缺少1个必需的位置参数

3 个答案: