RAI-17706 Added retry mechanism to the retry flow (#139)

* added retry mechanism to the retry flow

* wrapped _urlopen_with_retry into with to close the response once read

Author: Anatoli Kurtsevich

CHANGELOG.md

@@ -1,5 +1,9 @@
 # Changelog
 
+## v0.6.18
+
+* Added retry mechanism for the authentication flow.
+
 ## v0.6.17
 
 * Removed unnecessary dependencies of the SDK, bumped Pyarrow to v10.

railib/__init__.py

@@ -12,5 +12,5 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-__version_info__ = (0, 6, 17)
+__version_info__ = (0, 6, 18)
 __version__ = ".".join(map(str, __version_info__))

railib/rest.py

@@ -194,13 +194,15 @@ def _request_access_token(ctx: Context, url: str) -> AccessToken:
         data=data,
     )
     _print_request(req)
-    with urlopen(req) as rsp:
+    with _urlopen_with_retry(req, ctx.retries) as rsp:
         result = json.loads(rsp.read())
         token = result.get(ACCESS_KEY_TOKEN_KEY, None)
+
         if token is not None:
             expires_in = result.get(EXPIRES_IN_KEY, None)
             scope = result.get(SCOPE, None)
             return AccessToken(token, scope, expires_in)
+
     raise Exception("failed to get the access token")