將重試封裝爲`with`塊

Question

我正在尋求將數據庫事務的邏輯封裝到with塊中;將代碼封裝在事務中並處理各種異常（鎖定問題）。這很簡單，但是我想讓塊在某些例外之後封裝代碼塊的重試。我看不出一種方式將其整齊地打包到上下文管理器中。

是否可以在with語句中重複代碼？

我想使用它，就像這樣，這真的很乾淨。

def do_work(): 
    ... 
    # This is ideal! 
    with transaction(retries=3): 
     # Atomic DB statements 
     ... 
    ... 

我目前正在與一個裝飾處理這一點，但我更願意提供上下文管理器（或實際上兩者都有），所以我可以選擇包裹的幾行代碼在with塊代替內聯函數包裹在一個裝飾，這是我目前做的：

def do_work(): 
    ... 
    # This is not ideal! 
    @transaction(retries=3) 
    def _perform_in_transaction(): 
     # Atomic DB statements 
     ... 
    _perform_in_transaction() 
    ... 

Answer 1

由於裝飾只是函數本身，你可以做到以下幾點：

with transaction(_perform_in_transaction, retries=3) as _perf: 
    _perf() 

對於詳細信息，您需要實施transaction()作爲工廠方法，該工廠方法返回一個對象，其中__callable__()設置爲調用原始方法並重復執行，最多失敗次數爲retries; __enter__()和__exit__()將被定義爲正常的數據庫事務上下文管理器。

您可以選擇設置transaction()，這樣它自己就可以執行最多retries的傳遞方法，這可能需要大約與實現上下文管理器相同的工作量，但意味着實際使用量將減少到僅僅transaction(_perform_in_transaction, retries=3)（這實際上相當於提供的裝飾器例子delnan）。

Answer 2

是否可以重複with語句中的代碼？

No.

正如前面所指出的在郵件列表線程，你可以通過裝飾調用傳遞的功能降低有點重複：這發生在我

def do_work(): 
    ... 
    # This is not ideal! 
    @transaction(retries=3) 
    def _perform_in_transaction(): 
     # Atomic DB statements 
     ... 
    # called implicitly 
    ... 

Answer 3

方式這樣做只是爲了實現標準數據庫事務context manager，但允許它在構造函數中使用retries參數。然後，我會將它包裝在您的方法實現中。像這樣的：

class transaction(object): 
    def __init__(self, retries=0): 
     self.retries = retries 
    def __enter__(self): 
     return self 
    def __exit__(self, exc_type, exc_val, traceback): 
     pass 

    # Implementation... 
    def execute(self, query): 
     err = None 
     for _ in range(self.retries): 
      try: 
       return self._cursor.execute(query) 
      except Exception as e: 
       err = e # probably ought to save all errors, but hey 
     raise err 

with transaction(retries=3) as cursor: 
    cursor.execute('BLAH')