File size: 1,133 Bytes
60a259e | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 | import asyncio
import time
from functools import wraps
def async_retry(timeout=60, max_retries=None):
def decorator(func):
@wraps(func)
async def wrapper(*args, **kwargs):
start_time = time.time()
attempts = 0
while True:
try:
return await func(*args, **kwargs)
except Exception as e:
attempts += 1
if max_retries is not None and attempts >= max_retries:
print(f"Reached maximum retries of {max_retries}.")
raise Exception(f"Failed after {max_retries} retries.") from e
if time.time() - start_time > timeout:
print(f"Function timeout after {timeout} seconds.")
raise TimeoutError(f"Function execution exceeded {timeout} seconds timeout.") from e
print(f"Attempt {attempts} failed: {e}. Retrying...")
await asyncio.sleep(1) # Sleep to avoid tight loop or provide backoff logic here
return wrapper
return decorator |