Adds sleep method to ApiKeys responses

Author: nunomaduro

algoliasearch/helpers.py

@@ -1,5 +1,7 @@
 import json
+import math
 import sys
+import time
 
 from typing import Optional, Iterable, List, Union, Iterator, Dict, Any
 
@@ -68,3 +70,10 @@ def is_async_available():
             pass
 
     return False
+
+
+def sleep_for(retries_count, before_retry):
+    # type: (int, int) -> None
+
+    factor = math.ceil(retries_count / 10)
+    time.sleep(factor * before_retry / 1000000.0)

algoliasearch/responses.py

@@ -1,9 +1,11 @@
 import abc
+import math
+import time
 
 from typing import List, Union
 
 from algoliasearch.exceptions import RequestException
-from algoliasearch.helpers import get_items
+from algoliasearch.helpers import get_items, sleep_for
 from algoliasearch.http.request_options import RequestOptions
 
 try:
@@ -98,13 +100,20 @@ def __init__(self, client, raw_response):
     def wait(self):
         # type: () -> AddApiKeyResponse
 
+        retries_count = 1
+
         while not self._done:
             try:
                 self._client._sync().get_api_key(self.raw_response['key'])
                 self._done = True
             except RequestException as e:
                 if e.status_code != 404:
                     raise e
+                retries_count += 1
+                sleep_for(
+                    retries_count,
+                    self._client._config.wait_task_time_before_retry
+                )
 
         return self
 
@@ -127,14 +136,22 @@ def __init__(self, client, raw_response, request_options):
     def wait(self):
         # type: () -> UpdateApiKeyResponse
 
+        retries_count = 1
+
         while not self._done:
             api_key = self._client._sync().get_api_key(
                 self.raw_response['key'],
                 self._request_options
             )
 
-            if self._have_changed(api_key):
-                break
+            self._done = self._have_changed(api_key)
+
+            if not self._done:
+                retries_count += 1
+                sleep_for(
+                    retries_count,
+                    self._client._config.wait_task_time_before_retry
+                )
 
         return self
 
@@ -171,12 +188,21 @@ def __init__(self, client, raw_response, key):
     def wait(self):
         # type: () -> DeleteApiKeyResponse
 
+        retries_count = 1
+
         while not self._done:
             try:
                 self._client._sync().get_api_key(self._key)
             except RequestException as e:
                 self._done = e.status_code == 404
 
+            if not self._done:
+                retries_count += 1
+                sleep_for(
+                    retries_count,
+                    self._client._config.wait_task_time_before_retry
+                )
+
         return self
 
     def __getitem__(self, key):
@@ -198,13 +224,20 @@ def __init__(self, client, raw_response, key):
     def wait(self):
         # type: () -> RestoreApiKeyResponse
 
+        retries_count = 1
+
         while not self._done:
             try:
                 self._client._sync().get_api_key(self._key)
                 self._done = True
             except RequestException as e:
                 if e.status_code != 404:
                     raise e
+                retries_count += 1
+                sleep_for(
+                    retries_count,
+                    self._client._config.wait_task_time_before_retry
+                )
 
         return self