apache-airflow-providers-snowflake 5.5.2__py3-none-any.whl → 5.6.0__py3-none-any.whl

airflow/providers/snowflake/__init__.py

@@ -29,7 +29,7 @@ from airflow import __version__ as airflow_version
 
 __all__ = ["__version__"]
 
-__version__ = "5.5.2"
+__version__ = "5.6.0"
 
 if packaging.version.parse(packaging.version.parse(airflow_version).base_version) < packaging.version.parse(
     "2.7.0"

airflow/providers/snowflake/get_provider_info.py

@@ -28,8 +28,9 @@ def get_provider_info():
         "name": "Snowflake",
         "description": "`Snowflake <https://www.snowflake.com/>`__\n",
         "state": "ready",
-        "source-date-epoch": 1718605422,
+        "source-date-epoch": 1720423840,
         "versions": [
+            "5.6.0",
             "5.5.2",
             "5.5.1",
             "5.5.0",

airflow/providers/snowflake/hooks/snowflake.py

@@ -50,7 +50,8 @@ def _try_to_boolean(value: Any):
 
 
 class SnowflakeHook(DbApiHook):
-    """A client to interact with Snowflake.
+    """
+    A client to interact with Snowflake.
 
     This hook requires the snowflake_conn_id connection. The snowflake account, login,
     and, password field must be setup in the connection. Other inputs can be defined
@@ -128,6 +129,7 @@ class SnowflakeHook(DbApiHook):
                         "authenticator": "snowflake oauth",
                         "private_key_file": "private key",
                         "session_parameters": "session parameters",
+                        "client_request_mfa_token": "client request mfa token",
                     },
                     indent=1,
                 ),
@@ -155,6 +157,7 @@ class SnowflakeHook(DbApiHook):
         self.schema = kwargs.pop("schema", None)
         self.authenticator = kwargs.pop("authenticator", None)
         self.session_parameters = kwargs.pop("session_parameters", None)
+        self.client_request_mfa_token = kwargs.pop("client_request_mfa_token", None)
         self.query_ids: list[str] = []
 
     def _get_field(self, extra_dict, field_name):
@@ -181,7 +184,8 @@ class SnowflakeHook(DbApiHook):
 
     @cached_property
     def _get_conn_params(self) -> dict[str, str | None]:
-        """Fetch connection params as a dict.
+        """
+        Fetch connection params as a dict.
 
         This is used in ``get_uri()`` and ``get_connection()``.
         """
@@ -194,6 +198,7 @@ class SnowflakeHook(DbApiHook):
         role = self._get_field(extra_dict, "role") or ""
         insecure_mode = _try_to_boolean(self._get_field(extra_dict, "insecure_mode"))
         schema = conn.schema or ""
+        client_request_mfa_token = _try_to_boolean(self._get_field(extra_dict, "client_request_mfa_token"))
 
         # authenticator and session_parameters never supported long name so we don't use _get_field
         authenticator = extra_dict.get("authenticator", "snowflake")
@@ -216,6 +221,9 @@ class SnowflakeHook(DbApiHook):
         if insecure_mode:
             conn_config["insecure_mode"] = insecure_mode
 
+        if client_request_mfa_token:
+            conn_config["client_request_mfa_token"] = client_request_mfa_token
+
         # If private_key_file is specified in the extra json, load the contents of the file as a private key.
         # If private_key_content is specified in the extra json, use it as a private key.
         # As a next step, specify this private key in the connection configuration.
@@ -280,7 +288,9 @@ class SnowflakeHook(DbApiHook):
             **{
                 k: v
                 for k, v in conn_params.items()
-                if v and k not in ["session_parameters", "insecure_mode", "private_key"]
+                if v
+                and k
+                not in ["session_parameters", "insecure_mode", "private_key", "client_request_mfa_token"]
             }
         )
 
@@ -291,7 +301,8 @@ class SnowflakeHook(DbApiHook):
         return conn
 
     def get_sqlalchemy_engine(self, engine_kwargs=None):
-        """Get an sqlalchemy_engine object.
+        """
+        Get an sqlalchemy_engine object.
 
         :param engine_kwargs: Kwargs used in :func:`~sqlalchemy.create_engine`.
         :return: the created engine.
@@ -348,7 +359,8 @@ class SnowflakeHook(DbApiHook):
         return_last: bool = True,
         return_dictionaries: bool = False,
     ) -> tuple | list[tuple] | list[list[tuple] | tuple] | None:
-        """Run a command or list of commands.
+        """
+        Run a command or list of commands.
 
         Pass a list of SQL statements to the SQL parameter to get them to
         execute sequentially. The result of the queries is returned if the

airflow/providers/snowflake/transfers/copy_into_snowflake.py

@@ -152,7 +152,8 @@ class CopyFromExternalStageToSnowflakeOperator(BaseOperator):
     def _extract_openlineage_unique_dataset_paths(
         query_result: list[dict[str, Any]],
     ) -> tuple[list[tuple[str, str]], list[str]]:
-        """Extract and return unique OpenLineage dataset paths and file paths that failed to be parsed.
+        """
+        Extract and return unique OpenLineage dataset paths and file paths that failed to be parsed.
 
         Each row in the results is expected to have a 'file' field, which is a URI.
         The function parses these URIs and constructs a set of unique OpenLineage (namespace, name) tuples.

airflow/providers/snowflake/utils/openlineage.py

@@ -34,7 +34,8 @@ def fix_account_name(name: str) -> str:
 
 
 def fix_snowflake_sqlalchemy_uri(uri: str) -> str:
-    """Fix snowflake sqlalchemy connection URI to OpenLineage structure.
+    """
+    Fix snowflake sqlalchemy connection URI to OpenLineage structure.
 
     Snowflake sqlalchemy connection URI has following structure:
     'snowflake://<user_login_name>:<password>@<account_identifier>/<database_name>/<schema_name>?warehouse=<warehouse_name>&role=<role_name>'

{apache_airflow_providers_snowflake-5.5.2.dist-info → apache_airflow_providers_snowflake-5.6.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: apache-airflow-providers-snowflake
-Version: 5.5.2
+Version: 5.6.0
 Summary: Provider package apache-airflow-providers-snowflake for Apache Airflow
 Keywords: airflow-provider,snowflake,airflow,integration
 Author-email: Apache Software Foundation <dev@airflow.apache.org>
@@ -31,8 +31,8 @@ Requires-Dist: snowflake-sqlalchemy>=1.4.0
 Requires-Dist: apache-airflow-providers-common-sql ; extra == "common.sql"
 Requires-Dist: apache-airflow-providers-openlineage ; extra == "openlineage"
 Project-URL: Bug Tracker, https://github.com/apache/airflow/issues
-Project-URL: Changelog, https://airflow.apache.org/docs/apache-airflow-providers-snowflake/5.5.2/changelog.html
-Project-URL: Documentation, https://airflow.apache.org/docs/apache-airflow-providers-snowflake/5.5.2
+Project-URL: Changelog, https://airflow.apache.org/docs/apache-airflow-providers-snowflake/5.6.0/changelog.html
+Project-URL: Documentation, https://airflow.apache.org/docs/apache-airflow-providers-snowflake/5.6.0
 Project-URL: Slack Chat, https://s.apache.org/airflow-slack
 Project-URL: Source Code, https://github.com/apache/airflow
 Project-URL: Twitter, https://twitter.com/ApacheAirflow
@@ -84,7 +84,7 @@ Provides-Extra: openlineage
 
 Package ``apache-airflow-providers-snowflake``
 
-Release: ``5.5.2``
+Release: ``5.6.0``
 
 
 `Snowflake <https://www.snowflake.com/>`__
@@ -97,7 +97,7 @@ This is a provider package for ``snowflake`` provider. All classes for this prov
 are in ``airflow.providers.snowflake`` python package.
 
 You can find package information and changelog for the provider
-in the `documentation <https://airflow.apache.org/docs/apache-airflow-providers-snowflake/5.5.2/>`_.
+in the `documentation <https://airflow.apache.org/docs/apache-airflow-providers-snowflake/5.6.0/>`_.
 
 Installation
 ------------
@@ -144,4 +144,4 @@ Dependent package
 ==============================================================================================================  ===============
 
 The changelog for the provider package can be found in the
-`changelog <https://airflow.apache.org/docs/apache-airflow-providers-snowflake/5.5.2/changelog.html>`_.
+`changelog <https://airflow.apache.org/docs/apache-airflow-providers-snowflake/5.6.0/changelog.html>`_.

{apache_airflow_providers_snowflake-5.5.2.dist-info → apache_airflow_providers_snowflake-5.6.0.dist-info}/RECORD

@@ -1,20 +1,20 @@
 airflow/providers/snowflake/LICENSE,sha256=FFb4jd2AXnOOf7XLP04pQW6jbdhG49TxlGY6fFpCV1Y,13609
-airflow/providers/snowflake/__init__.py,sha256=Za2FwaQkAz1hqOofbW5IhxrSUCbcLOZECaMdJma376E,1496
-airflow/providers/snowflake/get_provider_info.py,sha256=uriVMER_1pjB3EV6Yw7YMy60iTEpYKxA7WHrhhUkgXk,4926
+airflow/providers/snowflake/__init__.py,sha256=L5RyPb469F362GFs1KFmcRDsiRSCMt8sF4P8ecoNIi0,1496
+airflow/providers/snowflake/get_provider_info.py,sha256=E2-2AsaJJc861WIkHv44ClpSlOh-e3uehaV0EjxSlgo,4947
 airflow/providers/snowflake/hooks/__init__.py,sha256=9hdXHABrVpkbpjZgUft39kOFL2xSGeG4GEua0Hmelus,785
-airflow/providers/snowflake/hooks/snowflake.py,sha256=KSV4uy27884NFiHI7A6Ol0GJBW_2JjrDqXqxSz0yi-k,20954
+airflow/providers/snowflake/hooks/snowflake.py,sha256=uCGBqsScQdDxYiN9EgtioSOtwr73A061Nnfjc4hw5Pk,21436
 airflow/providers/snowflake/hooks/snowflake_sql_api.py,sha256=ljm6v0CCeej7WyrK3IB6F6p954s5FRgEsocfUCQ83Is,14777
 airflow/providers/snowflake/operators/__init__.py,sha256=9hdXHABrVpkbpjZgUft39kOFL2xSGeG4GEua0Hmelus,785
 airflow/providers/snowflake/operators/snowflake.py,sha256=Evn8RfBTjLaSqrj7Dkhvu2-ewJoY6miTUFMiTTVd_dA,26383
 airflow/providers/snowflake/transfers/__init__.py,sha256=9hdXHABrVpkbpjZgUft39kOFL2xSGeG4GEua0Hmelus,785
-airflow/providers/snowflake/transfers/copy_into_snowflake.py,sha256=iV3JasS9U4pBHU_-kKFCAxiPRCBZWOzjMURaItAvgoc,12501
+airflow/providers/snowflake/transfers/copy_into_snowflake.py,sha256=ABkUcnpR1DaUH7CowxgcB7tT21nZslOnJ9xLWSegyuc,12510
 airflow/providers/snowflake/triggers/__init__.py,sha256=9hdXHABrVpkbpjZgUft39kOFL2xSGeG4GEua0Hmelus,785
 airflow/providers/snowflake/triggers/snowflake_trigger.py,sha256=YfMA7IXq3T7voUishTi171-8nIotPFzhYeFboSrHHAg,4223
 airflow/providers/snowflake/utils/__init__.py,sha256=9hdXHABrVpkbpjZgUft39kOFL2xSGeG4GEua0Hmelus,785
 airflow/providers/snowflake/utils/common.py,sha256=DG-KLy2KpZWAqZqm_XIECm8lmdoUlzwkXv9onmkQThc,1644
-airflow/providers/snowflake/utils/openlineage.py,sha256=PanpjG1YZlXfxqNjJCL7gNgRU1Gk-P0cj4j2qyf3mrA,2673
+airflow/providers/snowflake/utils/openlineage.py,sha256=b4qfJbMiTJNNhL0jMnRqff6DX5eyOaUYVnmJrEeqULY,2678
 airflow/providers/snowflake/utils/sql_api_generate_jwt.py,sha256=9mR-vHIquv60tfAni87f6FAjKsiRHUDDrsVhzw4M9vM,6762
-apache_airflow_providers_snowflake-5.5.2.dist-info/entry_points.txt,sha256=bCrl5J1PXUMzbgnrKYho61rkbL2gHRT4I6f_1jlxAX4,105
-apache_airflow_providers_snowflake-5.5.2.dist-info/WHEEL,sha256=EZbGkh7Ie4PoZfRQ8I0ZuP9VklN_TvcZ6DSE5Uar4z4,81
-apache_airflow_providers_snowflake-5.5.2.dist-info/METADATA,sha256=KgymXCPW7CrvbyfUWwXVghLvJNgn4XO3k8bD5uJdwDM,6938
-apache_airflow_providers_snowflake-5.5.2.dist-info/RECORD,,
+apache_airflow_providers_snowflake-5.6.0.dist-info/entry_points.txt,sha256=bCrl5J1PXUMzbgnrKYho61rkbL2gHRT4I6f_1jlxAX4,105
+apache_airflow_providers_snowflake-5.6.0.dist-info/WHEEL,sha256=EZbGkh7Ie4PoZfRQ8I0ZuP9VklN_TvcZ6DSE5Uar4z4,81
+apache_airflow_providers_snowflake-5.6.0.dist-info/METADATA,sha256=TAUl7RenaXd8PxNH6IKYY5IzoCqjHRgyg-iKUu7yezw,6938
+apache_airflow_providers_snowflake-5.6.0.dist-info/RECORD,,