声明装饰器

我们有一些如下所示的代码：

from third_party_library import foo

for n in range(3):
    try:
        foo(args)
        break
    except:
        print "Retry %i / 3" % n

我想使用装饰器，让我们的代码更加简洁，如下所示：

from third_party_library import foo

@retry(3)
foo(args)

这会产生语法错误。我是否遗漏了一些东西，或者Python只是不允许在语句上使用装饰器？

装饰器只能应用于函数和类定义例如：

@decorator
def func():
    ...

@decorator
class MyClass(object):
    ...

您不能在该语言的其他任何地方使用它们。

为了做你想做的事，你可以做一个正常的retry函数和传递foo and args作为参数。实现看起来像这样：

def retry(times, func, *args, **kwargs):
    for n in xrange(times):
        try:
            func(*args, **kwargs)
            break
        except Exception:  # Try to catch something more specific
            print "Retry %i / %i" % (n, times)