Progress

Author: adamtheturtle

src/vws/vws.py

@@ -276,12 +276,18 @@ def get_target_record(self, target_id: str) -> Dict[str, Union[str, int]]:
 
         return dict(response.json()['target_record'])
 
-    def _wait_for_target_processed(self, target_id: str) -> None:
+    def _wait_for_target_processed(
+        self,
+        target_id: str,
+        seconds_between_requests: float,
+    ) -> None:
         """
         Wait indefinitely for a target to get past the processing stage.
 
         Args:
             target_id: The ID of the target to wait for.
+            seconds_between_requests: The number of seconds to wait between
+                requests made while polling the target status.
 
         Raises:
             ~vws.exceptions.AuthenticationFailure: The secret key is not
@@ -298,10 +304,7 @@ def _wait_for_target_processed(self, target_id: str) -> None:
             if report['status'] != 'processing':
                 return
 
-            # We wait 0.2 seconds rather than less than that to decrease the
-            # number of calls made to the API, to decrease the likelihood of
-            # hitting the request quota.
-            sleep(0.2)
+            sleep(seconds_between_requests)
 
     @timeout_decorator.timeout(seconds=60 * 5)
     def wait_for_target_processed(self, target_id: str) -> None:
@@ -322,6 +325,14 @@ def wait_for_target_processed(self, target_id: str) -> None:
             ~vws.exceptions.UnknownTarget: The given target ID does not match a
                 target in the database.
         """
+        # We wait 0.2 seconds rather than less than that to decrease the
+        # number of calls made to the API, to decrease the likelihood of
+        # hitting the request quota.
+        seconds_between_requests = 0.2
+        _wait_for_target_processed(
+            target_id=target_id,
+            seconds_between_requests=seconds_between_requests,
+        )
 
     def list_targets(self) -> List[str]:
         """