SNOW-648958 Support for Okta OIE (#1233)

sfc-gh-hchaturvedi · web-flow · commit 979219d50a9a · 2022-08-23T14:59:49.000-07:00
diff --git a/src/snowflake/connector/auth_okta.py b/src/snowflake/connector/auth_okta.py
@@ -200,7 +200,7 @@ def _step3(self, headers, token_url, user, password):
             socket_timeout=self._rest._connection.login_timeout,
             catch_okta_unauthorized_error=True,
         )
-        one_time_token = ret.get("cookieToken")
+        one_time_token = ret.get("sessionToken", ret.get("cookieToken"))
         if not one_time_token:
             Error.errorhandler_wrapper(
                 self._rest._connection,

Original file line number	Diff line number	Diff line change
`@@ -200,7 +200,7 @@ def _step3(self, headers, token_url, user, password):`
`200`	`200`	`socket_timeout=self._rest._connection.login_timeout,`
`201`	`201`	`catch_okta_unauthorized_error=True,`
`202`	`202`	`)`
`203`		`- one_time_token = ret.get("cookieToken")`
	`203`	`+ one_time_token = ret.get("sessionToken", ret.get("cookieToken"))`
`204`	`204`	`if not one_time_token:`
`205`	`205`	`Error.errorhandler_wrapper(`
`206`	`206`	`self._rest._connection,`