简体   繁体   中英

Python retry with dynamic parameters

Tried this in the retrying and tenacity python libraries to no avail.

Retries are typically used with a decorator, for example as shown in metacode below:

class FooBar:

   @retry(attempts=3)
   def dosomething():
      ...

I want the retry parameters to be configurable on the class

class FooBar:
   def __init__(retries=0):
       self.retries = retries

   @retry(attempts=self.retries)
   def dosomething():
      ...

Obviously this will break because the decorator cannot accedss object attributes (ie. cannot access self ). So figured this would work:

def dosomething():
   with retry(attempts=self.retries):
       ...

But neither library allows for retry to be called in a with block

>  with retry():
E  AttributeError: __enter__

What is the preferred way to wrap retry logic with dynamic parameters?

You needn't use deorators with the @ syntax - they can also be used as functions.

from tenacity import retry, stop_after_attempt

class CustomClass:
    def __init__(self, retries):
        decorator = retry(stop=stop_after_attempt(retries), reraise=True)
        self.method_with_retry = decorator(self.method)
    
    def method(self, x):
        print('Trying...')
        if x % 2:
            raise ValueError
        return x

CustomClass(3).method_with_retry(11)

Nested function can help this:

class FooBar:
   def __init__(retries=0):
       self.retries = retries

   def dosomething():
       @retry(attempts=self.retries)
       def _dosomething():
          ...
       return _dosomething()

The technical post webpages of this site follow the CC BY-SA 4.0 protocol. If you need to reprint, please indicate the site URL or the original address.Any question please contact:yoyou2525@163.com.

 
粤ICP备18138465号  © 2020-2024 STACKOOM.COM