apache-airflow-providers-exasol 4.8.2__py3-none-any.whl → 4.8.3__py3-none-any.whl

airflow/providers/exasol/__init__.py

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

airflow/providers/exasol/hooks/exasol.py

@@ -24,6 +24,7 @@ from typing import TYPE_CHECKING, Any, TypeVar, overload
 import pyexasol
 from deprecated import deprecated
 from pyexasol import ExaConnection, ExaStatement
+from sqlalchemy.engine import URL
 
 from airflow.exceptions import AirflowProviderDeprecationWarning
 from airflow.providers.common.sql.hooks.handlers import return_single_query_results
@@ -53,10 +54,12 @@ class ExasolHook(DbApiHook):
     conn_type = "exasol"
     hook_name = "Exasol"
     supports_autocommit = True
+    DEFAULT_SQLALCHEMY_SCHEME = "exa+websocket"  # sqlalchemy-exasol dialect
 
-    def __init__(self, *args, **kwargs) -> None:
+    def __init__(self, *args, sqlalchemy_scheme: str | None = None, **kwargs) -> None:
         super().__init__(*args, **kwargs)
         self.schema = kwargs.pop("schema", None)
+        self._sqlalchemy_scheme = sqlalchemy_scheme
 
     def get_conn(self) -> ExaConnection:
         conn = self.get_connection(self.get_conn_id())
@@ -74,6 +77,46 @@ class ExasolHook(DbApiHook):
         conn = pyexasol.connect(**conn_args)
         return conn
 
+    @property
+    def sqlalchemy_scheme(self) -> str:
+        """Sqlalchemy scheme either from constructor, connection extras or default."""
+        extra_scheme = self.connection is not None and self.connection_extra_lower.get("sqlalchemy_scheme")
+        sqlalchemy_scheme = self._sqlalchemy_scheme or extra_scheme or self.DEFAULT_SQLALCHEMY_SCHEME
+        if sqlalchemy_scheme not in ["exa+websocket", "exa+pyodbc", "exa+turbodbc"]:
+            raise ValueError(
+                f"sqlalchemy_scheme in connection extra should be one of 'exa+websocket', 'exa+pyodbc' or 'exa+turbodbc', "
+                f"but got '{sqlalchemy_scheme}'. See https://github.com/exasol/sqlalchemy-exasol?tab=readme-ov-file#using-sqlalchemy-with-exasol-db for more details."
+            )
+        return sqlalchemy_scheme
+
+    @property
+    def sqlalchemy_url(self) -> URL:
+        """
+        Return a Sqlalchemy.engine.URL object from the connection.
+
+        :return: the extracted sqlalchemy.engine.URL object.
+        """
+        connection = self.connection
+        query = connection.extra_dejson
+        query = {k: v for k, v in query.items() if k.lower() != "sqlalchemy_scheme"}
+        return URL.create(
+            drivername=self.sqlalchemy_scheme,
+            username=connection.login,
+            password=connection.password,
+            host=connection.host,
+            port=connection.port,
+            database=self.schema or connection.schema,
+            query=query,
+        )
+
+    def get_uri(self) -> str:
+        """
+        Extract the URI from the connection.
+
+        :return: the extracted uri.
+        """
+        return self.sqlalchemy_url.render_as_string(hide_password=False)
+
     def _get_pandas_df(
         self, sql, parameters: Iterable | Mapping[str, Any] | None = None, **kwargs
     ) -> pd.DataFrame:

{apache_airflow_providers_exasol-4.8.2.dist-info → apache_airflow_providers_exasol-4.8.3.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: apache-airflow-providers-exasol
-Version: 4.8.2
+Version: 4.8.3
 Summary: Provider package apache-airflow-providers-exasol for Apache Airflow
 Keywords: airflow-provider,exasol,airflow,integration
 Author-email: Apache Software Foundation <dev@airflow.apache.org>
@@ -26,8 +26,8 @@ Requires-Dist: pyexasol>=0.26.0
 Requires-Dist: pandas>=2.1.2; python_version <"3.13"
 Requires-Dist: pandas>=2.2.3; python_version >="3.13"
 Project-URL: Bug Tracker, https://github.com/apache/airflow/issues
-Project-URL: Changelog, https://airflow.apache.org/docs/apache-airflow-providers-exasol/4.8.2/changelog.html
-Project-URL: Documentation, https://airflow.apache.org/docs/apache-airflow-providers-exasol/4.8.2
+Project-URL: Changelog, https://airflow.apache.org/docs/apache-airflow-providers-exasol/4.8.3/changelog.html
+Project-URL: Documentation, https://airflow.apache.org/docs/apache-airflow-providers-exasol/4.8.3
 Project-URL: Mastodon, https://fosstodon.org/@airflow
 Project-URL: Slack Chat, https://s.apache.org/airflow-slack
 Project-URL: Source Code, https://github.com/apache/airflow
@@ -58,9 +58,8 @@ Project-URL: YouTube, https://www.youtube.com/channel/UCSXwxpWZQ7XZ1WL3wqevChA/
 
 Package ``apache-airflow-providers-exasol``
 
-Release: ``4.8.2``
+Release: ``4.8.3``
 
-Release Date: ``|PypiReleaseDate|``
 
 `Exasol <https://www.exasol.com/>`__
 
@@ -72,12 +71,12 @@ This is a provider package for ``exasol`` provider. All classes for this provide
 are in ``airflow.providers.exasol`` python package.
 
 You can find package information and changelog for the provider
-in the `documentation <https://airflow.apache.org/docs/apache-airflow-providers-exasol/4.8.2/>`_.
+in the `documentation <https://airflow.apache.org/docs/apache-airflow-providers-exasol/4.8.3/>`_.
 
 Installation
 ------------
 
-You can install this package on top of an existing Airflow 2 installation (see ``Requirements`` below
+You can install this package on top of an existing Airflow installation (see ``Requirements`` below
 for the minimum Airflow version supported) via
 ``pip install apache-airflow-providers-exasol``
 
@@ -116,5 +115,5 @@ Dependent package
 ============================================================================================================  ==============
 
 The changelog for the provider package can be found in the
-`changelog <https://airflow.apache.org/docs/apache-airflow-providers-exasol/4.8.2/changelog.html>`_.
+`changelog <https://airflow.apache.org/docs/apache-airflow-providers-exasol/4.8.3/changelog.html>`_.
 

{apache_airflow_providers_exasol-4.8.2.dist-info → apache_airflow_providers_exasol-4.8.3.dist-info}/RECORD

@@ -1,11 +1,11 @@
 airflow/providers/exasol/LICENSE,sha256=gXPVwptPlW1TJ4HSuG5OMPg-a3h43OGMkZRR1rpwfJA,10850
-airflow/providers/exasol/__init__.py,sha256=D_gu1Fw81Rv2BJuOoaF_8tg4YRj5Yrj1J62OdSP2W-k,1495
+airflow/providers/exasol/__init__.py,sha256=lT_eVcX_xGi9aBt2QKp3ok1GCXEHH20W40bAJW9u8cM,1495
 airflow/providers/exasol/get_provider_info.py,sha256=uupz_ak2p7d1WTyVzKAFgaWsvNlCMYuMT892FXqytWA,1980
 airflow/providers/exasol/hooks/__init__.py,sha256=mlJxuZLkd5x-iq2SBwD3mvRQpt3YR7wjz_nceyF1IaI,787
-airflow/providers/exasol/hooks/exasol.py,sha256=Rw6t3gMg52QFSKa_A3Ta4X_vnqJRfiPyBUtfQMuR6W0,12872
+airflow/providers/exasol/hooks/exasol.py,sha256=O2AZUERAD9Jxtt0_i1uN-LT_dNpvtih6gP4mpZBPMYA,14768
 airflow/providers/exasol/operators/__init__.py,sha256=mlJxuZLkd5x-iq2SBwD3mvRQpt3YR7wjz_nceyF1IaI,787
 airflow/providers/exasol/operators/exasol.py,sha256=A4P3Qo65RJzrSVEgSNwPbJ2BSA9OsitmFm8kutyxVfw,2500
-apache_airflow_providers_exasol-4.8.2.dist-info/entry_points.txt,sha256=8pvKoFs3wCXkl3x0a5dzqvTonx17I8zD8oodM_M386Q,102
-apache_airflow_providers_exasol-4.8.2.dist-info/WHEEL,sha256=G2gURzTEtmeR8nrdXUJfNiB3VYVxigPQ-bEQujpNiNs,82
-apache_airflow_providers_exasol-4.8.2.dist-info/METADATA,sha256=DdXq3CDTlQjwrdTjdY90SgIfwnVs38RGNdcbNGua4qQ,5465
-apache_airflow_providers_exasol-4.8.2.dist-info/RECORD,,
+apache_airflow_providers_exasol-4.8.3.dist-info/entry_points.txt,sha256=8pvKoFs3wCXkl3x0a5dzqvTonx17I8zD8oodM_M386Q,102
+apache_airflow_providers_exasol-4.8.3.dist-info/WHEEL,sha256=G2gURzTEtmeR8nrdXUJfNiB3VYVxigPQ-bEQujpNiNs,82
+apache_airflow_providers_exasol-4.8.3.dist-info/METADATA,sha256=whQBES0jSDha54nXPu8CYx-R8ARVRcj6nTxWOxP-HWc,5427
+apache_airflow_providers_exasol-4.8.3.dist-info/RECORD,,