From 2fa6f42874c59d2941afc8fd2ecc7b445068709a Mon Sep 17 00:00:00 2001 From: Tomas Kukan Date: Mon, 14 Oct 2024 13:06:54 +0200 Subject: [PATCH] adjust logic --- kbcstorage/retry_requests.py | 6 ++---- 1 file changed, 2 insertions(+), 4 deletions(-) diff --git a/kbcstorage/retry_requests.py b/kbcstorage/retry_requests.py index 264e02a..b549671 100644 --- a/kbcstorage/retry_requests.py +++ b/kbcstorage/retry_requests.py @@ -3,16 +3,14 @@ MAX_RETRIES = 5 BACKOFF_FACTOR = 1.0 -RETRY_STATUS_CODES = {500, 502, 503, 504} - def _get_backoff_time(retry_count): return BACKOFF_FACTOR * (2 ** retry_count) def _retry_request(request_func, url, *args, **kwargs): response = request_func(url, *args, **kwargs) - for retry_count in range(1, MAX_RETRIES): - if response.status_code not in RETRY_STATUS_CODES: + for retry_count in range(MAX_RETRIES - 1): + if response.status_code == 501 or response.status_code < 500: return response time.sleep(_get_backoff_time(retry_count)) response = request_func(url, **kwargs)