adding port support into oauth code flow auth

snowflakedb · Jan 29, 2025 · 60b461a · 60b461a
1 parent 4d78183
commit 60b461a
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/src/snowflake/connector/connection.py b/src/snowflake/connector/connection.py
@@ -319,12 +319,12 @@ def _get_private_bytes_from_file(
         # SNOW-1825621: OAUTH implementation
     ),
     "oauth_authorization_url": (
-        "{host}/oauth/authorize",
+        "{host}:{port}/oauth/authorize",
         str,
         # SNOW-1825621: OAUTH implementation
     ),
     "oauth_token_request_url": (
-        "{host}/oauth/token-request",
+        "{host}:{port}/oauth/token-request",
         str,
         # SNOW-1825621: OAUTH implementation
     ),
@@ -1135,10 +1135,10 @@ def __open_connection(self):
                     client_id=self._oauth_client_id,
                     client_secret=self._oauth_client_secret,
                     authentication_url=self._oauth_authorization_url.format(
-                        host=self.host,
+                        host=self.host, port=self.port
                     ),
                     token_request_url=self._oauth_token_request_url.format(
-                        host=self.host,
+                        host=self.host, port=self.port
                     ),
                     redirect_uri="http://127.0.0.1:{port}/",
                     scope=self._oauth_scope,