我可以將異常作爲參數傳遞給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