Make retry backoff and status codes customizable

Author: mmmeeedddsss

gql/transport/requests.py

@@ -1,7 +1,7 @@
 import io
 import json
 import logging
-from typing import Any, Dict, Optional, Tuple, Type, Union
+from typing import Any, Collection, Dict, Optional, Tuple, Type, Union
 
 import requests
 from graphql import DocumentNode, ExecutionResult, print_ast
@@ -31,6 +31,7 @@ class RequestsHTTPTransport(Transport):
     """
 
     file_classes: Tuple[Type[Any], ...] = (io.IOBase,)
+    _default_retry_codes = (429, 500, 502, 503, 504)
 
     def __init__(
         self,
@@ -42,6 +43,8 @@ def __init__(
         timeout: Optional[int] = None,
         verify: Union[bool, str] = True,
         retries: int = 0,
+        retry_backoff_factor: float = 0.1,
+        retry_status_forcelist: Collection[int] = _default_retry_codes,
         method: str = "POST",
         **kwargs: Any,
     ):
@@ -61,6 +64,13 @@ def __init__(
             the server's TLS certificate, or a string, in which case it must be a path
             to a CA bundle to use. (Default: True).
         :param retries: Pre-setup of the requests' Session for performing retries
+        :param retry_backoff_factor: A backoff factor to apply between attempts after 
+            the second try. urllib3 will sleep for: 
+            {backoff factor} * (2 ** ({number of previous retries}))
+        :param retry_status_forcelist: A set of integer HTTP status codes that we 
+            should force a retry on. A retry is initiated if the request method is 
+            in allowed_methods and the response status code is in status_forcelist.
+            (Default: [429, 500, 502, 503, 504])
         :param method: HTTP method used for requests. (Default: POST).
         :param kwargs: Optional arguments that ``request`` takes.
             These can be seen at the `requests`_ source code or the official `docs`_
@@ -76,6 +86,8 @@ def __init__(
         self.default_timeout = timeout
         self.verify = verify
         self.retries = retries
+        self.retry_backoff_factor = retry_backoff_factor
+        self.retry_status_forcelist = retry_status_forcelist
         self.method = method
         self.kwargs = kwargs
 
@@ -95,8 +107,8 @@ def connect(self):
                 adapter = HTTPAdapter(
                     max_retries=Retry(
                         total=self.retries,
-                        backoff_factor=0.1,
-                        status_forcelist=[500, 502, 503, 504],
+                        backoff_factor=self.retry_backoff_factor,
+                        status_forcelist=self.retry_status_forcelist,
                         allowed_methods=None,
                     )
                 )