@@ -84,8 +84,8 @@ def make_request(self, fn, endpoint, as_json, **kwargs):
8484 if 0 < r .status_code < 500 :
8585 node .healthy = True
8686
87- # We should raise a custom exception if status code is not 200 or 201
88- if r .status_code not in [ 200 , 201 ] :
87+ # We should raise a custom exception if status code is not 20X
88+ if 200 <= r .status_code < 300 :
8989 error_message = r .json ().get ('message' , 'API error.' )
9090 # Raised exception will be caught and retried only if it's a 50X
9191 raise ApiCall .get_exception (r .status_code )(r .status_code , error_message )
@@ -104,16 +104,16 @@ def get(self, endpoint, params=None, as_json=True):
104104 params = params or {}
105105 return self .make_request (requests .get , endpoint , as_json ,
106106 params = params ,
107- timeout = self .config .timeout_seconds )
107+ timeout = self .config .connection_timeout_seconds )
108108
109109 def post (self , endpoint , body ):
110110 return self .make_request (requests .post , endpoint , True ,
111- data = body , timeout = self .config .timeout_seconds )
111+ data = body , timeout = self .config .connection_timeout_seconds )
112112
113113 def put (self , endpoint , body ):
114114 return self .make_request (requests .put , endpoint , True ,
115- data = body , timeout = self .config .timeout_seconds )
115+ data = body , timeout = self .config .connection_timeout_seconds )
116116
117117 def delete (self , endpoint ):
118118 return self .make_request (requests .delete , endpoint , True ,
119- timeout = self .config .timeout_seconds )
119+ timeout = self .config .connection_timeout_seconds )
0 commit comments