diff --git a/kbcstorage/retry_requests.py b/kbcstorage/retry_requests.py index 604c890..423230a 100644 --- a/kbcstorage/retry_requests.py +++ b/kbcstorage/retry_requests.py @@ -4,9 +4,11 @@ MAX_RETRIES = 11 BACKOFF_FACTOR = 1.0 + 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(MAX_RETRIES - 1): @@ -16,14 +18,18 @@ def _retry_request(request_func, url, *args, **kwargs): response = request_func(url, **kwargs) return response + def get(url, *args, **kwargs): return _retry_request(requests.get, url, *args, **kwargs) + def post(url, *args, **kwargs): return _retry_request(requests.post, url, *args, **kwargs) + def put(url, *args, **kwargs): return _retry_request(requests.put, url, *args, **kwargs) + def delete(url, *args, **kwargs): - return _retry_request(requests.delete, url, *args, **kwargs) \ No newline at end of file + return _retry_request(requests.delete, url, *args, **kwargs)