Merge pull request #259 from dharmeshkakadia/client_specified_retry

Allow clients to control Retry specification

Author: jamalex

notion/client.py

@@ -27,18 +27,21 @@
 from .utils import extract_id, now
 
 
-def create_session():
+def create_session(client_specified_retry=None):
     """
     retry on 502
     """
     session = Session()
-    retry = Retry(
-        5,
-        backoff_factor=0.3,
-        status_forcelist=(502,),
-        # CAUTION: adding 'POST' to this list which is not technically idempotent
-        method_whitelist=("POST", "HEAD", "TRACE", "GET", "PUT", "OPTIONS", "DELETE"),
-    )
+    if client_specified_retry:
+        retry = client_specified_retry
+    else:
+        retry = Retry(
+            5,
+            backoff_factor=0.3,
+            status_forcelist=(502,),
+            # CAUTION: adding 'POST' to this list which is not technically idempotent
+            method_whitelist=("POST", "HEAD", "TRACE", "GET", "PUT", "OPTIONS", "DELETE"),
+        )
     adapter = HTTPAdapter(max_retries=retry)
     session.mount("https://", adapter)
     return session
@@ -58,8 +61,9 @@ def __init__(
         start_monitoring=False,
         enable_caching=False,
         cache_key=None,
+        client_specified_retry=None,
     ):
-        self.session = create_session()
+        self.session = create_session(client_specified_retry)
         self.session.cookies = cookiejar_from_dict({"token_v2": token_v2})
         if enable_caching:
             cache_key = cache_key or hashlib.sha256(token_v2.encode()).hexdigest()