Adds 401 to status_forcelist

plaidcloud/rpc/connection/jsonrpc.py

@@ -135,23 +135,15 @@
 class RPCRetry(Retry):
     def __init__(self, *args, check_allow_transmit=None, **kwargs):
         """
-
         Args:
             check_allow_transmit (callable, optional): Method to call to check if retries are still to be made
                 This can be used to prevent retry of RPC methods once a workflow has been cancelled and the RPC fails
         """
-        if version.parse(urllib3.__version__) >= version.parse("1.26.0"):
-            kwargs.update(dict(
-                allowed_methods=['POST'],
-                status_forcelist=[500, 502, 504],
-                backoff_factor=0.1,
-            ))
-        else:
-            kwargs.update(dict(
-                method_whitelist=['POST'],
-                status_forcelist=[500, 502, 504],
-                backoff_factor=0.1,
-            ))
+        kwargs.update(dict(
+            allowed_methods=['POST'],
+            status_forcelist=[401, 500, 502, 504],
+            backoff_factor=0.1,
+        ))
         if 'connect' not in kwargs:
             kwargs['connect'] = 5
         super(RPCRetry, self).__init__(*args, **kwargs)
@@ -168,6 +160,7 @@
         return True
 
     def increment(self, *args, **kwargs):
+        print(f'Hit Retry, Request History Looks Like: {self.history}')
         if not self.allow_transmit:
             raise Exception('No more retries, RPC method has been cancelled')
         return super(RPCRetry, self).increment(*args, **kwargs)
@@ -210,6 +203,7 @@
                 if response.get('ok'):
                     return response['result']
                 else:
+                    print(f'Retry Object History: {retry.}')
                     error = response['error']
                     if error.get('code') == WARNING_CODE:
                         raise Warning(error.get('message'))