Change oauth redirect port range to port

Signed-off-by: Jacky Hu <[email protected]>

Author: jackyhu-db

src/databricks/sql/auth/auth.py

@@ -90,8 +90,9 @@ def get_python_sql_connector_auth_provider(hostname: str, **kwargs):
         tls_client_cert_file=kwargs.get("_tls_client_cert_file"),
         oauth_scopes=PYSQL_OAUTH_SCOPES,
         oauth_client_id=kwargs.get("oauth_client_id") or PYSQL_OAUTH_CLIENT_ID,
-        oauth_redirect_port_range=kwargs.get("oauth_redirect_port_range") if kwargs.get(
-            "oauth_client_id") else PYSQL_OAUTH_REDIRECT_PORT_RANGE,
+        oauth_redirect_port_range=[kwargs.get("oauth_redirect_port")]
+        if kwargs.get("oauth_client_id")
+        else PYSQL_OAUTH_REDIRECT_PORT_RANGE,
         oauth_persistence=kwargs.get("experimental_oauth_persistence"),
     )
     return get_auth_provider(cfg)

src/databricks/sql/client.py

@@ -61,10 +61,11 @@ def __init__(
                 Note this is beta (private preview)
 
             oauth_client_id: `str`, optional
-                custom oauth client_id. If not specified, it will use the client_id of databricks-sql-python
+                custom oauth client_id. If not specified, it will use the built-in client_id of databricks-sql-python.
 
-            oauth_redirect_port_range: `List[str]`, optional
-                oauth redirect port range. This is required when custom oauth client_id `oauth_client_id` is set
+            oauth_redirect_port: `int`, optional
+                port of the oauth redirect uri (localhost). This is required when custom oauth client_id
+                `oauth_client_id` is set
 
             experimental_oauth_persistence: configures preferred storage for persisting oauth tokens.
                 This has to be a class implementing `OAuthPersistence`.