snowflake-connector-python 3.10.0__tar.gz → 3.11.0__tar.gz

{snowflake_connector_python-3.10.0 → snowflake_connector_python-3.11.0}/DESCRIPTION.md

@@ -8,6 +8,20 @@ Source code is also available at: https://github.com/snowflakedb/snowflake-conne
 
 # Release Notes
 
+- 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.
+
 - v3.10.0(April 29,2024)
 
   - Added support for structured types to fetch_pandas_all.

{snowflake_connector_python-3.10.0 → snowflake_connector_python-3.11.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: snowflake-connector-python
-Version: 3.10.0
+Version: 3.11.0
 Summary: Snowflake Connector for Python
 Home-page: https://www.snowflake.com/
 Author: Snowflake, Inc
@@ -85,6 +85,20 @@ Source code is also available at: https://github.com/snowflakedb/snowflake-conne
 
 # Release Notes
 
+- 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.
+
 - v3.10.0(April 29,2024)
 
   - Added support for structured types to fetch_pandas_all.

{snowflake_connector_python-3.10.0 → snowflake_connector_python-3.11.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.0 → snowflake_connector_python-3.11.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.0 → snowflake_connector_python-3.11.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.0 → snowflake_connector_python-3.11.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.0 → snowflake_connector_python-3.11.0}/src/snowflake/connector/connection.py

@@ -281,6 +281,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 +367,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 +722,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,
@@ -936,14 +947,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 +962,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 +1130,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
@@ -1182,6 +1195,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

{snowflake_connector_python-3.10.0 → snowflake_connector_python-3.11.0}/src/snowflake/connector/connection_diagnostic.py

@@ -175,7 +175,9 @@ 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}")

{snowflake_connector_python-3.10.0 → snowflake_connector_python-3.11.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.
@@ -630,9 +630,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()
@@ -758,9 +758,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 +809,7 @@ class SnowflakeCursor:
         _skip_upload_on_content_match: bool = False,
         file_stream: IO[bytes] | None = None,
         num_statements: int | None = None,
-    ) -> Self | None:
-        ...
+    ) -> Self | None: ...
 
     @overload
     def execute(
@@ -837,8 +838,7 @@ class SnowflakeCursor:
         _skip_upload_on_content_match: bool = False,
         file_stream: IO[bytes] | None = None,
         num_statements: int | None = None,
-    ) -> dict[str, Any] | None:
-        ...
+    ) -> dict[str, Any] | None: ...
 
     def execute(
         self,
@@ -1278,12 +1278,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.0 → snowflake_connector_python-3.11.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()

{snowflake_connector_python-3.10.0 → snowflake_connector_python-3.11.0}/src/snowflake/connector/file_transfer_agent.py

@@ -941,9 +941,9 @@ class SnowflakeFileTransferAgent:
             if len(response["src_locations"]) == len(self._encryption_material):
                 for idx, src_file in enumerate(self._src_files):
                     logger.debug(src_file)
-                    self._src_file_to_encryption_material[
-                        src_file
-                    ] = self._encryption_material[idx]
+                    self._src_file_to_encryption_material[src_file] = (
+                        self._encryption_material[idx]
+                    )
             elif len(self._encryption_material) != 0:
                 # some encryption material exists. Zero means no encryption
                 Error.errorhandler_wrapper(
@@ -1000,9 +1000,11 @@ class SnowflakeFileTransferAgent:
                         intermediate_stream=self._source_from_stream,
                         src_file_size=self._source_from_stream.seek(0, os.SEEK_END),
                         stage_location_type=self._stage_location_type,
-                        encryption_material=self._encryption_material[0]
-                        if len(self._encryption_material) > 0
-                        else None,
+                        encryption_material=(
+                            self._encryption_material[0]
+                            if len(self._encryption_material) > 0
+                            else None
+                        ),
                     )
                 )
                 self._source_from_stream.seek(0)
@@ -1035,9 +1037,11 @@ class SnowflakeFileTransferAgent:
                             src_file_name=file_name,
                             src_file_size=statinfo.st_size,
                             stage_location_type=self._stage_location_type,
-                            encryption_material=self._encryption_material[0]
-                            if len(self._encryption_material) > 0
-                            else None,
+                            encryption_material=(
+                                self._encryption_material[0]
+                                if len(self._encryption_material) > 0
+                                else None
+                            ),
                         )
                     )
         elif self._command_type == CMD_TYPE_DOWNLOAD:
@@ -1066,11 +1070,11 @@ class SnowflakeFileTransferAgent:
                         stage_location_type=self._stage_location_type,
                         local_location=self._local_location,
                         presigned_url=url,
-                        encryption_material=self._src_file_to_encryption_material[
-                            file_name
-                        ]
-                        if file_name in self._src_file_to_encryption_material
-                        else None,
+                        encryption_material=(
+                            self._src_file_to_encryption_material[file_name]
+                            if file_name in self._src_file_to_encryption_material
+                            else None
+                        ),
                     )
                 )
 

{snowflake_connector_python-3.10.0 → snowflake_connector_python-3.11.0}/src/snowflake/connector/gcs_storage_client.py

@@ -203,9 +203,11 @@ class SnowflakeGCSRestClient(SnowflakeStorageClient):
                 encryption_metadata = EncryptionMetadata(
                     key=encryptiondata["WrappedContentKey"]["EncryptedKey"],
                     iv=encryptiondata["ContentEncryptionIV"],
-                    matdesc=response.headers[GCS_METADATA_MATDESC_KEY]
-                    if GCS_METADATA_MATDESC_KEY in response.headers
-                    else None,
+                    matdesc=(
+                        response.headers[GCS_METADATA_MATDESC_KEY]
+                        if GCS_METADATA_MATDESC_KEY in response.headers
+                        else None
+                    ),
                 )
 
         meta.gcs_file_header_digest = response.headers.get(GCS_METADATA_SFC_DIGEST)
@@ -359,9 +361,11 @@ class SnowflakeGCSRestClient(SnowflakeStorageClient):
                         encryption_metadata = EncryptionMetadata(
                             key=encryption_data["WrappedContentKey"]["EncryptedKey"],
                             iv=encryption_data["ContentEncryptionIV"],
-                            matdesc=response.headers[GCS_METADATA_MATDESC_KEY]
-                            if GCS_METADATA_MATDESC_KEY in response.headers
-                            else None,
+                            matdesc=(
+                                response.headers[GCS_METADATA_MATDESC_KEY]
+                                if GCS_METADATA_MATDESC_KEY in response.headers
+                                else None
+                            ),
                         )
                 meta.result_status = ResultStatus.UPLOADED
                 return FileHeader(

{snowflake_connector_python-3.10.0 → snowflake_connector_python-3.11.0}/src/snowflake/connector/nanoarrow_cpp/ArrowIterator/CArrowTableIterator.cpp

@@ -226,8 +226,6 @@ void CArrowTableIterator::convertIfNeeded(ArrowSchema* columnSchema,
         scale =
             std::stoi(std::string(scaleString.data, scaleString.size_bytes));
       }
-
-      logger->error(__FILE__, __func__, __LINE__, "scale: %d", scale);
       convertTimestampColumn_nanoarrow(&columnSchemaView, columnArray, scale,
                                        m_timezone);
       break;

{snowflake_connector_python-3.10.0 → snowflake_connector_python-3.11.0}/src/snowflake/connector/network.py

@@ -832,9 +832,9 @@ class SnowflakeRestful:
             retry_ctx = RetryCtx(
                 _include_retry_params=include_retry_params,
                 _include_retry_reason=include_retry_reason,
-                timeout=timeout
-                if timeout is not None
-                else self._connection.network_timeout,
+                timeout=(
+                    timeout if timeout is not None else self._connection.network_timeout
+                ),
                 backoff_generator=self._connection._backoff_generator,
             )
 

{snowflake_connector_python-3.10.0 → snowflake_connector_python-3.11.0}/src/snowflake/connector/ocsp_asn1crypto.py

@@ -50,8 +50,8 @@ class SnowflakeOCSPAsn1Crypto(SnowflakeOCSP):
     # map signature algorithm name to digest class
     SIGNATURE_ALGORITHM_TO_DIGEST_CLASS = {
         "sha256": hashes.SHA256,
-        "sha384": hashes.SHA3_384,
-        "sha512": hashes.SHA3_512,
+        "sha384": hashes.SHA384,
+        "sha512": hashes.SHA512,
     }
 
     def encode_cert_id_key(self, hkey):

{snowflake_connector_python-3.10.0 → snowflake_connector_python-3.11.0}/src/snowflake/connector/ocsp_snowflake.py

@@ -62,6 +62,7 @@ from . import constants
 from .backoff_policies import exponential_backoff
 from .cache import SFDictCache, SFDictFileCache
 from .telemetry import TelemetryField, generate_telemetry_data_dict
+from .url_util import url_encode_str
 
 
 class OCSPResponseValidationResult(NamedTuple):
@@ -436,8 +437,9 @@ class OCSPServer:
 
     def generate_get_url(self, ocsp_url, b64data):
         parsed_url = urlsplit(ocsp_url)
+        url_encoded_b64data = url_encode_str(b64data)
         if self.OCSP_RETRY_URL is None:
-            target_url = f"{ocsp_url}/{b64data}"
+            target_url = f"{ocsp_url}/{url_encoded_b64data}"
         else:
             # values of parsed_url.netloc and parsed_url.path based on oscp_url are as follows:
             # URL                                    NETLOC                         PATH
@@ -447,7 +449,9 @@ class OCSPServer:
             # "http://oneocsp.microsoft.com/ocsp"    "oneocsp.microsoft.com"        "/ocsp"
             # The check below is to treat first two urls same
             path = parsed_url.path if parsed_url.path != "/" else ""
-            target_url = self.OCSP_RETRY_URL.format(parsed_url.netloc + path, b64data)
+            target_url = self.OCSP_RETRY_URL.format(
+                parsed_url.netloc + path, url_encoded_b64data
+            )
 
         logger.debug("OCSP Retry URL is - %s", target_url)
         return target_url

{snowflake_connector_python-3.10.0 → snowflake_connector_python-3.11.0}/src/snowflake/connector/result_batch.py

@@ -366,9 +366,9 @@ class ResultBatch(abc.ABC):
                 )
                 time.sleep(sleep_timer)
 
-        self._metrics[
-            DownloadMetrics.download.value
-        ] = download_metric.get_timing_millis()
+        self._metrics[DownloadMetrics.download.value] = (
+            download_metric.get_timing_millis()
+        )
         return response
 
     @abc.abstractmethod
@@ -664,11 +664,21 @@ class ArrowResultBatch(ResultBatch):
     ) -> Iterator[dict | Exception] | Iterator[tuple | Exception] | Iterator[Table]:
         """Create an iterator for the ResultBatch. Used by get_arrow_iter."""
         if self._local:
-            return self._from_data(self._data, iter_unit)
+            try:
+                return self._from_data(self._data, iter_unit)
+            except Exception:
+                if connection and getattr(connection, "_debug_arrow_chunk", False):
+                    logger.debug(f"arrow data can not be parsed: {self._data}")
+                raise
         response = self._download(connection=connection)
         logger.debug(f"started loading result batch id: {self.id}")
         with TimerContextManager() as load_metric:
-            loaded_data = self._load(response, iter_unit)
+            try:
+                loaded_data = self._load(response, iter_unit)
+            except Exception:
+                if connection and getattr(connection, "_debug_arrow_chunk", False):
+                    logger.debug(f"arrow data can not be parsed: {response}")
+                raise
         logger.debug(f"finished loading result batch id: {self.id}")
         self._metrics[DownloadMetrics.load.value] = load_metric.get_timing_millis()
         return loaded_data

{snowflake_connector_python-3.10.0 → snowflake_connector_python-3.11.0}/src/snowflake/connector/result_set.py

@@ -174,12 +174,10 @@ class ResultSet(Iterable[list]):
         return self._create_iter(iter_unit=IterUnit.TABLE_UNIT, structure="arrow")
 
     @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:
         """Fetches a single Arrow Table from all of the ``ResultBatch``."""

{snowflake_connector_python-3.10.0 → snowflake_connector_python-3.11.0}/src/snowflake/connector/sf_dirs.py

@@ -14,8 +14,7 @@ from platformdirs import PlatformDirs
 
 class PlatformDirsProto(Protocol):
     @property
-    def user_config_path(self) -> pathlib.Path:
-        ...
+    def user_config_path(self) -> pathlib.Path: ...
 
 
 def _resolve_platform_dirs() -> PlatformDirsProto:

{snowflake_connector_python-3.10.0 → snowflake_connector_python-3.11.0}/src/snowflake/connector/storage_client.py

@@ -74,7 +74,7 @@ class SnowflakeStorageClient(ABC):
     ) -> None:
         self.meta = meta
         self.stage_info = stage_info
-        self.retry_count: dict[int | str, int] = defaultdict(lambda: 0)
+        self.retry_count: dict[int | str, int] = defaultdict(int)
         self.tmp_dir = tempfile.mkdtemp()
         self.data_file: str | None = None
         self.encryption_metadata: EncryptionMetadata | None = None
@@ -247,7 +247,7 @@ class SnowflakeStorageClient(ABC):
 
     def finish_upload(self) -> None:
         meta = self.meta
-        if self.successful_transfers == self.num_of_chunks:
+        if self.successful_transfers == self.num_of_chunks and self.num_of_chunks != 0:
             if self.num_of_chunks > 1:
                 self._complete_multipart_upload()
             meta.result_status = ResultStatus.UPLOADED

{snowflake_connector_python-3.10.0 → snowflake_connector_python-3.11.0}/src/snowflake/connector/telemetry.py

@@ -239,9 +239,9 @@ def generate_telemetry_data_dict(
         {
             TelemetryField.KEY_DRIVER_TYPE.value: CLIENT_NAME,
             TelemetryField.KEY_DRIVER_VERSION.value: SNOWFLAKE_CONNECTOR_VERSION,
-            TelemetryField.KEY_SOURCE.value: connection.application
-            if connection
-            else CLIENT_NAME,
+            TelemetryField.KEY_SOURCE.value: (
+                connection.application if connection else CLIENT_NAME
+            ),
             **from_dict,
         }
         if not is_oob_telemetry