我可以将异常作为参数传递给python中的函数吗？

Question

我是新来的蟒蛇。我试图创建一个重试装饰器，当应用于一个函数时，它将继续重试，直到满足一些条件（为了简单起见，重试10次）。

def retry(): 
    def wrapper(func): 
     for i in range(0,10): 
      try: 
       func() 
       break 
      except: 
       continue 
    return wrapper 

现在，将重试任何异常。我如何改变它，使其重试特定的例外情况。 e.g，我想用它喜欢：

@retry(ValueError, AbcError) 
def myfunc(): 
    //do something 

我想myfunc将仅重试它抛出ValueError或AbcError。

Answer 1

您可以提供异常的tuple到except ..块捕捉：

from functools import wraps 

def retry(*exceptions, **params): 
    if not exceptions: 
     exceptions = (Exception,) 
    tries = params.get('tries', 10) 

    def decorator(func): 
     @wraps(func) 
     def wrapper(*args, **kw): 
      for i in range(tries): 
       try: 
        return func(*args, **kw) 
       except exceptions: 
        pass 
     return wrapper 
    return decorator 

包罗万象的*exceptions参数将总是导致一个元组。我添加了一个tries关键字一样，所以你可以配置重过数量：

@retry(ValueError, TypeError, tries=20) 
def foo(): 
    pass 

演示：

>>> @retry(NameError, tries=3) 
... def foo(): 
...  print 'Futzing the foo!' 
...  bar 
... 
>>> foo() 
Futzing the foo! 
Futzing the foo! 
Futzing the foo! 

Answer 2

您可以检查错误类：

except Exception as e: 
    for et in error_types: #(or args) 
     if isinstance(e, et): 
      continue 
    raise e #re-raise 

Answer 3

from functools import wraps 

class retry(object): 
    def __init__(self, *exceptions): 
     self.exceptions = exceptions 

    def __call__(self, f): 
     @wraps(f) # required to save the original context of the wrapped function 
     def wrapped(*args, **kwargs): 
      for i in range(0,10): 
       try: 
        f(*args, **kwargs) 
       except self.exceptions: 
        continue 
     return wrapped 

用法：

@retry(ValueError, Exception) 
def f(): 
    print('In f') 
    raise ValueError 


>>> f() 
In f 
In f 
In f 
In f 
In f 
In f 
In f 
In f 
In f 
In f