snowflake-connector-python 3.10.1__tar.gz → 3.12.0__tar.gz

{snowflake_connector_python-3.10.1 → snowflake_connector_python-3.12.0}/DESCRIPTION.md

@@ -8,6 +8,25 @@ Source code is also available at: https://github.com/snowflakedb/snowflake-conne
 
 # Release Notes
 
+- v3.12.0(July 24,2024)
+  - Set default connection timeout of 10 seconds and socket read timeout of 10 minutes for HTTP calls in file transfer.
+  - Optimized `to_pandas()` performance by fully parallel downloading logic.
+  - Fixed a bug that specifying client_session_keep_alive_heartbeat_frequency in snowflake-sqlalchemy could crash the connector.
+  - Fixed incorrect type hint of connection parameter `private_key`.
+  - Added support for connectivity to multiple domains.
+  - Bumped keyring dependency from >=23.1.0,<25.0.0 to >=23.1.0,<26.0.0.
+  - Disabled OOB Telemetry.
+
+- v3.11.0(June 17,2024)
+  - Added support for `token_file_path` connection parameter to read an OAuth token from a file when connecting to Snowflake.
+  - Added support for `debug_arrow_chunk` connection parameter to allow debugging raw arrow data in case of arrow data parsing failure.
+  - Added support for `disable_saml_url_check` connection parameter to disable SAML URL check in OKTA authentication.
+  - Fixed a bug that OCSP certificate signed using SHA384 algorithm cannot be verified.
+  - Fixed a bug that status code shown as uploaded when PUT command failed with 400 error.
+  - Fixed a bug that a PermissionError was raised when the current user does not have the right permission on parent directory of config file path.
+  - Fixed a bug that OCSP GET url is not encoded correctly when it contains a slash.
+  - Fixed a bug that an SSO URL didn't accept `:` in a query parameter, for instance, `https://sso.abc.com/idp/startSSO.ping?PartnerSpId=https://xyz.snowflakecomputing.com/`.
+
 - v3.10.1(May 21, 2024)
 
   - Removed an incorrect error log message that could occur during arrow data conversion.

{snowflake_connector_python-3.10.1 → snowflake_connector_python-3.12.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: snowflake-connector-python
-Version: 3.10.1
+Version: 3.12.0
 Summary: Snowflake Connector for Python
 Home-page: https://www.snowflake.com/
 Author: Snowflake, Inc
@@ -73,7 +73,7 @@ Provides-Extra: pandas
 Requires-Dist: pandas<3.0.0,>=1.0.0; extra == "pandas"
 Requires-Dist: pyarrow; extra == "pandas"
 Provides-Extra: secure-local-storage
-Requires-Dist: keyring<25.0.0,>=23.1.0; extra == "secure-local-storage"
+Requires-Dist: keyring<26.0.0,>=23.1.0; extra == "secure-local-storage"
 
 This package includes the Snowflake Connector for Python, which conforms to the Python DB API 2.0 specification:
 https://www.python.org/dev/peps/pep-0249/
@@ -85,6 +85,25 @@ Source code is also available at: https://github.com/snowflakedb/snowflake-conne
 
 # Release Notes
 
+- v3.12.0(July 24,2024)
+  - Set default connection timeout of 10 seconds and socket read timeout of 10 minutes for HTTP calls in file transfer.
+  - Optimized `to_pandas()` performance by fully parallel downloading logic.
+  - Fixed a bug that specifying client_session_keep_alive_heartbeat_frequency in snowflake-sqlalchemy could crash the connector.
+  - Fixed incorrect type hint of connection parameter `private_key`.
+  - Added support for connectivity to multiple domains.
+  - Bumped keyring dependency from >=23.1.0,<25.0.0 to >=23.1.0,<26.0.0.
+  - Disabled OOB Telemetry.
+
+- v3.11.0(June 17,2024)
+  - Added support for `token_file_path` connection parameter to read an OAuth token from a file when connecting to Snowflake.
+  - Added support for `debug_arrow_chunk` connection parameter to allow debugging raw arrow data in case of arrow data parsing failure.
+  - Added support for `disable_saml_url_check` connection parameter to disable SAML URL check in OKTA authentication.
+  - Fixed a bug that OCSP certificate signed using SHA384 algorithm cannot be verified.
+  - Fixed a bug that status code shown as uploaded when PUT command failed with 400 error.
+  - Fixed a bug that a PermissionError was raised when the current user does not have the right permission on parent directory of config file path.
+  - Fixed a bug that OCSP GET url is not encoded correctly when it contains a slash.
+  - Fixed a bug that an SSO URL didn't accept `:` in a query parameter, for instance, `https://sso.abc.com/idp/startSSO.ping?PartnerSpId=https://xyz.snowflakecomputing.com/`.
+
 - v3.10.1(May 21, 2024)
 
   - Removed an incorrect error log message that could occur during arrow data conversion.

{snowflake_connector_python-3.10.1 → snowflake_connector_python-3.12.0}/setup.cfg

@@ -96,7 +96,7 @@ pandas =
 	pandas>=1.0.0,<3.0.0
 	pyarrow
 secure-local-storage = 
-	keyring>=23.1.0,<25.0.0
+	keyring>=23.1.0,<26.0.0
 
 [egg_info]
 tag_build = 

{snowflake_connector_python-3.10.1 → snowflake_connector_python-3.12.0}/src/snowflake/connector/_query_context_cache.py

@@ -165,9 +165,11 @@ class QueryContextCache:
                             "id": qce.id,
                             "timestamp": qce.read_timestamp,
                             "priority": qce.priority,
-                            "context": {"base64Data": qce.context}
-                            if qce.context is not None
-                            else {},
+                            "context": (
+                                {"base64Data": qce.context}
+                                if qce.context is not None
+                                else {}
+                            ),
                         }
                         for qce in self._tree_set
                     ]

{snowflake_connector_python-3.10.1 → snowflake_connector_python-3.12.0}/src/snowflake/connector/auth/_auth.py

@@ -417,27 +417,35 @@ class Auth:
         else:
             logger.debug(
                 "token = %s",
-                "******"
-                if ret["data"] and ret["data"].get("token") is not None
-                else "NULL",
+                (
+                    "******"
+                    if ret["data"] and ret["data"].get("token") is not None
+                    else "NULL"
+                ),
             )
             logger.debug(
                 "master_token = %s",
-                "******"
-                if ret["data"] and ret["data"].get("masterToken") is not None
-                else "NULL",
+                (
+                    "******"
+                    if ret["data"] and ret["data"].get("masterToken") is not None
+                    else "NULL"
+                ),
             )
             logger.debug(
                 "id_token = %s",
-                "******"
-                if ret["data"] and ret["data"].get("idToken") is not None
-                else "NULL",
+                (
+                    "******"
+                    if ret["data"] and ret["data"].get("idToken") is not None
+                    else "NULL"
+                ),
             )
             logger.debug(
                 "mfa_token = %s",
-                "******"
-                if ret["data"] and ret["data"].get("mfaToken") is not None
-                else "NULL",
+                (
+                    "******"
+                    if ret["data"] and ret["data"].get("mfaToken") is not None
+                    else "NULL"
+                ),
             )
             if not ret["data"]:
                 Error.errorhandler_wrapper(

{snowflake_connector_python-3.10.1 → snowflake_connector_python-3.12.0}/src/snowflake/connector/auth/okta.py

@@ -315,7 +315,9 @@ class AuthByOkta(AuthByPlugin):
             host=conn._rest._host,
             port=conn._rest._port,
         )
-        if not _is_prefix_equal(post_back_url, full_url):
+        if not getattr(conn, "_disable_saml_url_check", False) and not _is_prefix_equal(
+            post_back_url, full_url
+        ):
             Error.errorhandler_wrapper(
                 conn._rest._connection,
                 None,

{snowflake_connector_python-3.10.1 → snowflake_connector_python-3.12.0}/src/snowflake/connector/config_manager.py

@@ -320,8 +320,15 @@ class ConfigManager:
         ):
             if sliceoptions.only_in_slice:
                 del read_config_file[section]
-            if not filep.exists():
+            try:
+                if not filep.exists():
+                    continue
+            except PermissionError:
+                LOGGER.debug(
+                    f"Fail to read configuration file from {str(filep)} due to no permission on its parent directory"
+                )
                 continue
+
             if (
                 sliceoptions.check_permissions  # Skip checking if this file couldn't hold sensitive information
                 # Same check as openssh does for permissions

{snowflake_connector_python-3.10.1 → snowflake_connector_python-3.12.0}/src/snowflake/connector/connection.py

@@ -52,6 +52,7 @@ from .compat import IS_LINUX, IS_WINDOWS, quote, urlencode
 from .config_manager import CONFIG_MANAGER, _get_default_connection_params
 from .connection_diagnostic import ConnectionDiagnostic
 from .constants import (
+    _DOMAIN_NAME_MAP,
     ENV_VAR_PARTNER,
     PARAMETER_AUTOCOMMIT,
     PARAMETER_CLIENT_PREFETCH_THREADS,
@@ -107,6 +108,7 @@ from .sqlstate import SQLSTATE_CONNECTION_NOT_EXISTS, SQLSTATE_FEATURE_NOT_SUPPO
 from .telemetry import TelemetryClient, TelemetryData, TelemetryField
 from .telemetry_oob import TelemetryService
 from .time_util import HeartBeatTimer, get_time_millis
+from .url_util import extract_top_level_domain_from_hostname
 from .util_text import construct_hostname, parse_account, split_statements
 
 DEFAULT_CLIENT_PREFETCH_THREADS = 4
@@ -181,7 +183,7 @@ DEFAULT_CONFIGURATION: dict[str, tuple[Any, type | tuple[type, ...]]] = {
     "backoff_policy": (DEFAULT_BACKOFF_POLICY, Callable),
     "passcode_in_password": (False, bool),  # Snowflake MFA
     "passcode": (None, (type(None), str)),  # Snowflake MFA
-    "private_key": (None, (type(None), str, RSAPrivateKey)),
+    "private_key": (None, (type(None), bytes, RSAPrivateKey)),
     "private_key_file": (None, (type(None), str)),
     "private_key_file_pwd": (None, (type(None), str, bytes)),
     "token": (None, (type(None), str)),  # OAuth or JWT Token
@@ -190,6 +192,10 @@ DEFAULT_CONFIGURATION: dict[str, tuple[Any, type | tuple[type, ...]]] = {
     "password_callback": (None, (type(None), Callable)),
     "auth_class": (None, (type(None), AuthByPlugin)),
     "application": (CLIENT_NAME, (type(None), str)),
+    # internal_application_name/version is used to tell the server the type of client connecting to
+    # Snowflake. There are some functionalities (e.g., MFA, Arrow result format) that
+    # Snowflake server doesn't support for new client types, which requires developers to
+    # add the new client type to the server to support these features.
     "internal_application_name": (CLIENT_NAME, (type(None), str)),
     "internal_application_version": (CLIENT_VERSION, (type(None), str)),
     "insecure_mode": (False, bool),  # Error security fix requirement
@@ -281,6 +287,14 @@ DEFAULT_CONFIGURATION: dict[str, tuple[Any, type | tuple[type, ...]]] = {
         True,
         bool,
     ),  # Disable console login and fall back to getting SSO URL from GS
+    "debug_arrow_chunk": (
+        False,
+        bool,
+    ),  # log raw arrow chunk for debugging purpuse in case there is malformed arrow data
+    "disable_saml_url_check": (
+        False,
+        bool,
+    ),  # disable saml url check in okta authentication
 }
 
 APPLICATION_RE = re.compile(r"[\w\d_]+")
@@ -359,6 +373,7 @@ class SnowflakeConnection:
           This parameter is only effective when the result format is JSON.
         server_session_keep_alive: When true, the connector does not destroy the session on the Snowflake server side
           before the connector shuts down. Default value is false.
+        token_file_path: The file path of the token file. If both token and token_file_path are provided, the token in token_file_path will be used.
     """
 
     OCSP_ENV_LOCK = Lock()
@@ -713,9 +728,11 @@ class SnowflakeConnection:
                 account=self.account,
                 host=self.host,
                 connection_diag_log_path=self.connection_diag_log_path,
-                connection_diag_allowlist_path=self.connection_diag_allowlist_path
-                if self.connection_diag_allowlist_path is not None
-                else self.connection_diag_whitelist_path,
+                connection_diag_allowlist_path=(
+                    self.connection_diag_allowlist_path
+                    if self.connection_diag_allowlist_path is not None
+                    else self.connection_diag_whitelist_path
+                ),
                 proxy_host=self.proxy_host,
                 proxy_port=self.proxy_port,
                 proxy_user=self.proxy_user,
@@ -920,7 +937,7 @@ class SnowflakeConnection:
                 os.environ["SF_OCSP_RESPONSE_CACHE_SERVER_URL"],
             )
 
-        if self.host.endswith(".privatelink.snowflakecomputing.com"):
+        if ".privatelink.snowflakecomputing." in self.host:
             SnowflakeConnection.setup_ocsp_privatelink(self.application, self.host)
         else:
             if "SF_OCSP_RESPONSE_CACHE_SERVER_URL" in os.environ:
@@ -936,14 +953,14 @@ class SnowflakeConnection:
 
         if self._validate_default_parameters:
             # Snowflake will validate the requested database, schema, and warehouse
-            self._session_parameters[
-                PARAMETER_CLIENT_VALIDATE_DEFAULT_PARAMETERS
-            ] = True
+            self._session_parameters[PARAMETER_CLIENT_VALIDATE_DEFAULT_PARAMETERS] = (
+                True
+            )
 
         if self.client_session_keep_alive is not None:
-            self._session_parameters[
-                PARAMETER_CLIENT_SESSION_KEEP_ALIVE
-            ] = self._client_session_keep_alive
+            self._session_parameters[PARAMETER_CLIENT_SESSION_KEEP_ALIVE] = (
+                self._client_session_keep_alive
+            )
 
         if self.client_session_keep_alive_heartbeat_frequency is not None:
             self._session_parameters[
@@ -951,9 +968,9 @@ class SnowflakeConnection:
             ] = self._validate_client_session_keep_alive_heartbeat_frequency()
 
         if self.client_prefetch_threads:
-            self._session_parameters[
-                PARAMETER_CLIENT_PREFETCH_THREADS
-            ] = self._validate_client_prefetch_threads()
+            self._session_parameters[PARAMETER_CLIENT_PREFETCH_THREADS] = (
+                self._validate_client_prefetch_threads()
+            )
 
         # Setup authenticator
         auth = Auth(self.rest)
@@ -1119,11 +1136,13 @@ class SnowflakeConnection:
                     warnings.warn(
                         "'{}' connection parameter should be of type '{}', but is a '{}'".format(
                             name,
-                            str(tuple(e.__name__ for e in accepted_types)).replace(
-                                "'", ""
-                            )
-                            if isinstance(accepted_types, tuple)
-                            else accepted_types.__name__,
+                            (
+                                str(tuple(e.__name__ for e in accepted_types)).replace(
+                                    "'", ""
+                                )
+                                if isinstance(accepted_types, tuple)
+                                else accepted_types.__name__
+                            ),
                             type(value).__name__,
                         ),
                         # Raise warning from where class was initiated
@@ -1165,6 +1184,10 @@ class SnowflakeConnection:
             if "protocol" not in kwargs:
                 self._protocol = "https"
 
+        logger.info(
+            f"Connecting to {_DOMAIN_NAME_MAP.get(extract_top_level_domain_from_hostname(self._host), 'GLOBAL')} Snowflake domain"
+        )
+
         # If using a custom auth class, we should set the authenticator
         # type to be the same as the custom auth class
         if self._auth_class:
@@ -1182,6 +1205,12 @@ class SnowflakeConnection:
             ]:
                 self._authenticator = auth_tmp
 
+        # read OAuth token from
+        token_file_path = kwargs.get("token_file_path")
+        if token_file_path:
+            with open(token_file_path) as f:
+                self._token = f.read()
+
         if not (self._master_token and self._session_token):
             if not self.user and self._authenticator != OAUTH_AUTHENTICATOR:
                 # OAuth Authentication does not require a username
@@ -1268,6 +1297,7 @@ class SnowflakeConnection:
         _update_current_object: bool = True,
         _no_retry: bool = False,
         timeout: int | None = None,
+        dataframe_ast: str | None = None,
     ) -> dict[str, Any]:
         """Executes a query with a sequence counter."""
         logger.debug("_cmd_query")
@@ -1277,6 +1307,8 @@ class SnowflakeConnection:
             "sequenceId": sequence_counter,
             "querySubmissionTime": get_time_millis(),
         }
+        if dataframe_ast is not None:
+            data["dataframeAst"] = dataframe_ast
         if statement_params is not None:
             data["parameters"] = statement_params
         if is_internal:
@@ -1651,6 +1683,12 @@ class SnowflakeConnection:
         """Validate and return heartbeat frequency in seconds."""
         real_max = int(self.rest.master_validity_in_seconds / 4)
         real_min = int(real_max / 4)
+
+        # ensure the type is integer
+        self._client_session_keep_alive_heartbeat_frequency = int(
+            self.client_session_keep_alive_heartbeat_frequency
+        )
+
         if self.client_session_keep_alive_heartbeat_frequency is None:
             # This is an unlikely scenario but covering it just in case.
             self._client_session_keep_alive_heartbeat_frequency = real_min
@@ -1659,10 +1697,6 @@ class SnowflakeConnection:
         elif self.client_session_keep_alive_heartbeat_frequency < real_min:
             self._client_session_keep_alive_heartbeat_frequency = real_min
 
-        # ensure the type is integer
-        self._client_session_keep_alive_heartbeat_frequency = int(
-            self.client_session_keep_alive_heartbeat_frequency
-        )
         return self.client_session_keep_alive_heartbeat_frequency
 
     def _validate_client_prefetch_threads(self) -> int:

{snowflake_connector_python-3.10.1 → snowflake_connector_python-3.12.0}/src/snowflake/connector/connection_diagnostic.py

@@ -23,6 +23,7 @@ import OpenSSL
 
 from .compat import IS_WINDOWS, urlparse
 from .cursor import SnowflakeCursor
+from .url_util import extract_top_level_domain_from_hostname
 from .vendored import urllib3
 
 logger = getLogger(__name__)
@@ -89,15 +90,21 @@ class ConnectionDiagnostic:
         self.__append_message(
             host_type, f"Host based on specified account: {self.host}"
         )
-        if ".com.snowflakecomputing.com" in self.host:
-            self.host = host.split(".com.snow", 1)[0] + ".com"
+
+        top_level_domain = extract_top_level_domain_from_hostname(host)
+        if (
+            f".{top_level_domain}.snowflakecomputing.{top_level_domain}" in self.host
+        ):  # repeated domain name pattern
+            self.host = (
+                host.split(f".{top_level_domain}.snow", 1)[0] + f".{top_level_domain}"
+            )
             logger.warning(
-                f"Account should not have snowflakecomputing.com in it. You provided {host}.  "
+                f"Account should not have snowflakecomputing.{top_level_domain} in it. You provided {host}.  "
                 f"Continuing with fixed host."
             )
             self.__append_message(
                 host_type,
-                f"We removed extra .snowflakecomputing.com and will continue with host: "
+                f"We removed extra .snowflakecomputing.{top_level_domain} and will continue with host: "
                 f"{self.host}",
             )
         else:
@@ -175,13 +182,15 @@ class ConnectionDiagnostic:
                 )
                 self.full_connection_diag_allowlist_path = None
 
-        self.allowlist_sql: str = "select /* snowflake-connector-python:connection_diagnostics */ system$allowlist();"
+        self.allowlist_sql: str = (
+            "select /* snowflake-connector-python:connection_diagnostics */ system$allowlist();"
+        )
 
         if self.__is_privatelink():
             self.ocsp_urls.append(f"ocsp.{self.host}")
             self.allowlist_sql = "select system$allowlist_privatelink();"
         else:
-            self.ocsp_urls.append("ocsp.snowflakecomputing.com")
+            self.ocsp_urls.append(f"ocsp.snowflakecomputing.{top_level_domain}")
 
         self.allowlist_retrieval_success: bool = False
         self.cursor: SnowflakeCursor | None = None

{snowflake_connector_python-3.10.1 → snowflake_connector_python-3.12.0}/src/snowflake/connector/constants.py

@@ -36,6 +36,11 @@ DBAPI_TYPE_BINARY = 1
 DBAPI_TYPE_NUMBER = 2
 DBAPI_TYPE_TIMESTAMP = 3
 
+_DEFAULT_HOSTNAME_TLD = "com"
+_CHINA_HOSTNAME_TLD = "cn"
+_TOP_LEVEL_DOMAIN_REGEX = r"\.[a-zA-Z]{1,63}$"
+_SNOWFLAKE_HOST_SUFFIX_REGEX = r"snowflakecomputing(\.[a-zA-Z]{1,63}){1,2}$"
+
 
 class FieldType(NamedTuple):
     name: str
@@ -399,6 +404,7 @@ class IterUnit(Enum):
     TABLE_UNIT = "table"
 
 
+# File Transfer
 # Amazon S3 multipart upload limits
 # https://docs.aws.amazon.com/AmazonS3/latest/userguide/qfacts.html
 S3_DEFAULT_CHUNK_SIZE = 8 * 1024**2
@@ -410,8 +416,15 @@ S3_MAX_PARTS = 10000
 S3_CHUNK_SIZE = 8388608  # boto3 default
 AZURE_CHUNK_SIZE = 4 * megabyte
 
+# https://requests.readthedocs.io/en/latest/user/advanced/#timeouts
+REQUEST_CONNECTION_TIMEOUT = 10
+REQUEST_READ_TIMEOUT = 600
+
 DAY_IN_SECONDS = 60 * 60 * 24
 
 # TODO: all env variables definitions should be here
 ENV_VAR_PARTNER = "SF_PARTNER"
 ENV_VAR_TEST_MODE = "SNOWFLAKE_TEST_MODE"
+
+
+_DOMAIN_NAME_MAP = {_DEFAULT_HOSTNAME_TLD: "GLOBAL", _CHINA_HOSTNAME_TLD: "CHINA"}

{snowflake_connector_python-3.10.1 → snowflake_connector_python-3.12.0}/src/snowflake/connector/cursor.py

@@ -149,9 +149,11 @@ class ResultMetadata(NamedTuple):
     def from_column(cls, col: dict[str, Any]):
         """Initializes a ResultMetadata object from the column description in the query response."""
         type_code = FIELD_NAME_TO_ID[
-            col["extTypeName"].upper()
-            if col.get("extTypeName")
-            else col["type"].upper()
+            (
+                col["extTypeName"].upper()
+                if col.get("extTypeName")
+                else col["type"].upper()
+            )
         ]
 
         return cls(
@@ -546,12 +548,10 @@ class SnowflakeCursor:
         return None
 
     @overload
-    def callproc(self, procname: str) -> tuple:
-        ...
+    def callproc(self, procname: str) -> tuple: ...
 
     @overload
-    def callproc(self, procname: str, args: T) -> T:
-        ...
+    def callproc(self, procname: str, args: T) -> T: ...
 
     def callproc(self, procname: str, args=tuple()):
         """Call a stored procedure.
@@ -601,6 +601,7 @@ class SnowflakeCursor:
         _no_results: bool = False,
         _is_put_get=None,
         _no_retry: bool = False,
+        dataframe_ast: str | None = None,
     ) -> dict[str, Any]:
         del self.messages[:]
 
@@ -630,9 +631,9 @@ class SnowflakeCursor:
                 if str(result_format_val).upper() == "ARROW":
                     self.check_can_use_arrow_resultset()
                 elif result_format_val is None:
-                    statement_params[
-                        PARAMETER_PYTHON_CONNECTOR_QUERY_RESULT_FORMAT
-                    ] = "JSON"
+                    statement_params[PARAMETER_PYTHON_CONNECTOR_QUERY_RESULT_FORMAT] = (
+                        "JSON"
+                    )
 
         self._sequence_counter = self._connection._next_sequence_counter()
         self._request_id = uuid.uuid4()
@@ -704,6 +705,7 @@ class SnowflakeCursor:
                 _no_results=_no_results,
                 _no_retry=_no_retry,
                 timeout=real_timeout,
+                dataframe_ast=dataframe_ast,
             )
         finally:
             try:
@@ -758,9 +760,11 @@ class SnowflakeCursor:
         if params is not None and len(params) == 0:
             self._log_telemetry_job_data(
                 TelemetryField.EMPTY_SEQ_INTERPOLATION,
-                TelemetryData.TRUE
-                if self.connection._interpolate_empty_sequences
-                else TelemetryData.FALSE,
+                (
+                    TelemetryData.TRUE
+                    if self.connection._interpolate_empty_sequences
+                    else TelemetryData.FALSE
+                ),
             )
         if logger.getEffectiveLevel() <= logging.DEBUG:
             logger.debug(
@@ -807,8 +811,8 @@ class SnowflakeCursor:
         _skip_upload_on_content_match: bool = False,
         file_stream: IO[bytes] | None = None,
         num_statements: int | None = None,
-    ) -> Self | None:
-        ...
+        _dataframe_ast: str | None = None,
+    ) -> Self | None: ...
 
     @overload
     def execute(
@@ -837,8 +841,8 @@ class SnowflakeCursor:
         _skip_upload_on_content_match: bool = False,
         file_stream: IO[bytes] | None = None,
         num_statements: int | None = None,
-    ) -> dict[str, Any] | None:
-        ...
+        _dataframe_ast: str | None = None,
+    ) -> dict[str, Any] | None: ...
 
     def execute(
         self,
@@ -866,6 +870,7 @@ class SnowflakeCursor:
         _skip_upload_on_content_match: bool = False,
         file_stream: IO[bytes] | None = None,
         num_statements: int | None = None,
+        _dataframe_ast: str | None = None,
     ) -> Self | dict[str, Any] | None:
         """Executes a command/query.
 
@@ -900,6 +905,7 @@ class SnowflakeCursor:
             file_stream: File-like object to be uploaded with PUT
             num_statements: Query level parameter submitted in _statement_params constraining exact number of
             statements being submitted (or 0 if submitting an uncounted number) when using a multi-statement query.
+            _dataframe_ast: Base64-encoded dataframe request abstract syntax tree.
 
         Returns:
             The cursor itself, or None if some error happened, or the response returned
@@ -941,6 +947,7 @@ class SnowflakeCursor:
             "_no_results": _no_results,
             "_is_put_get": _is_put_get,
             "_no_retry": _no_retry,
+            "dataframe_ast": _dataframe_ast,
         }
 
         if self._connection.is_pyformat:
@@ -1278,12 +1285,10 @@ class SnowflakeCursor:
         return self._result_set._fetch_arrow_batches()
 
     @overload
-    def fetch_arrow_all(self, force_return_table: Literal[False]) -> Table | None:
-        ...
+    def fetch_arrow_all(self, force_return_table: Literal[False]) -> Table | None: ...
 
     @overload
-    def fetch_arrow_all(self, force_return_table: Literal[True]) -> Table:
-        ...
+    def fetch_arrow_all(self, force_return_table: Literal[True]) -> Table: ...
 
     def fetch_arrow_all(self, force_return_table: bool = False) -> Table | None:
         """

{snowflake_connector_python-3.10.1 → snowflake_connector_python-3.12.0}/src/snowflake/connector/errors.py

@@ -153,9 +153,9 @@ class Error(BASE_EXCEPTION_CLASS):
             and not connection._telemetry.is_closed
         ):
             # Send with in-band telemetry
-            telemetry_data[
-                TelemetryField.KEY_TYPE.value
-            ] = TelemetryField.SQL_EXCEPTION.value
+            telemetry_data[TelemetryField.KEY_TYPE.value] = (
+                TelemetryField.SQL_EXCEPTION.value
+            )
             telemetry_data[TelemetryField.KEY_SOURCE.value] = connection.application
             telemetry_data[TelemetryField.KEY_EXCEPTION.value] = self.__class__.__name__
             ts = get_time_millis()