apache-airflow-providers-common-sql 1.14.2rc1__py3-none-any.whl → 1.15.0rc1__py3-none-any.whl

airflow/providers/common/sql/__init__.py

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

airflow/providers/common/sql/get_provider_info.py

@@ -28,8 +28,9 @@ def get_provider_info():
         "name": "Common SQL",
         "description": "`Common SQL Provider <https://en.wikipedia.org/wiki/SQL>`__\n",
         "state": "ready",
-        "source-date-epoch": 1720422590,
+        "source-date-epoch": 1722663378,
         "versions": [
+            "1.15.0",
             "1.14.2",
             "1.14.1",
             "1.14.0",

airflow/providers/common/sql/hooks/sql.py

@@ -40,6 +40,7 @@ from urllib.parse import urlparse
 import sqlparse
 from more_itertools import chunked
 from sqlalchemy import create_engine
+from sqlalchemy.engine import Inspector
 
 from airflow.exceptions import (
     AirflowException,
@@ -242,7 +243,20 @@ class DbApiHook(BaseHook):
         """
         if engine_kwargs is None:
             engine_kwargs = {}
-        return create_engine(self.get_uri(), **engine_kwargs)
+        engine_kwargs["creator"] = self.get_conn
+
+        try:
+            url = self.sqlalchemy_url
+        except NotImplementedError:
+            url = self.get_uri()
+
+        self.log.debug("url: %s", url)
+        self.log.debug("engine_kwargs: %s", engine_kwargs)
+        return create_engine(url=url, **engine_kwargs)
+
+    @property
+    def inspector(self) -> Inspector:
+        return Inspector.from_engine(self.get_sqlalchemy_engine())
 
     def get_pandas_df(
         self,
@@ -571,6 +585,7 @@ class DbApiHook(BaseHook):
         replace=False,
         *,
         executemany=False,
+        autocommit=False,
         **kwargs,
     ):
         """
@@ -585,12 +600,14 @@ class DbApiHook(BaseHook):
         :param commit_every: The maximum number of rows to insert in one
             transaction. Set to 0 to insert all rows in one transaction.
         :param replace: Whether to replace instead of insert
-        :param executemany: (Deprecated) If True, all rows are inserted at once in
+        :param executemany: If True, all rows are inserted at once in
             chunks defined by the commit_every parameter. This only works if all rows
             have same number of column names, but leads to better performance.
+        :param autocommit: What to set the connection's autocommit setting to
+            before executing the query.
         """
         nb_rows = 0
-        with self._create_autocommit_connection() as conn:
+        with self._create_autocommit_connection(autocommit) as conn:
             conn.commit()
             with closing(conn.cursor()) as cur:
                 if self.supports_executemany or executemany:

airflow/providers/common/sql/hooks/sql.pyi

@@ -42,7 +42,7 @@ from airflow.providers.openlineage.extractors import OperatorLineage as Operator
 from airflow.providers.openlineage.sqlparser import DatabaseInfo as DatabaseInfo
 from functools import cached_property as cached_property
 from pandas import DataFrame as DataFrame
-from sqlalchemy.engine import URL as URL
+from sqlalchemy.engine import Inspector, URL as URL
 from typing import Any, Callable, Generator, Iterable, Mapping, Protocol, Sequence, TypeVar, overload
 
 T = TypeVar("T")
@@ -64,7 +64,6 @@ class DbApiHook(BaseHook):
     log_sql: Incomplete
     descriptions: Incomplete
     def __init__(self, *args, schema: str | None = None, log_sql: bool = True, **kwargs) -> None: ...
-
     def get_conn_id(self) -> str: ...
     @cached_property
     def placeholder(self): ...
@@ -73,6 +72,8 @@ class DbApiHook(BaseHook):
     @property
     def sqlalchemy_url(self) -> URL: ...
     def get_sqlalchemy_engine(self, engine_kwargs: Incomplete | None = None): ...
+    @property
+    def inspector(self) -> Inspector: ...
     def get_pandas_df(
         self, sql, parameters: list | tuple | Mapping[str, Any] | None = None, **kwargs
     ) -> DataFrame: ...
@@ -123,6 +124,7 @@ class DbApiHook(BaseHook):
         replace: bool = False,
         *,
         executemany: bool = False,
+        autocommit: bool = False,
         **kwargs,
     ): ...
     def bulk_dump(self, table, tmp_file) -> None: ...

{apache_airflow_providers_common_sql-1.14.2rc1.dist-info → apache_airflow_providers_common_sql-1.15.0rc1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: apache-airflow-providers-common-sql
-Version: 1.14.2rc1
+Version: 1.15.0rc1
 Summary: Provider package apache-airflow-providers-common-sql for Apache Airflow
 Keywords: airflow-provider,common.sql,airflow,integration
 Author-email: Apache Software Foundation <dev@airflow.apache.org>
@@ -28,8 +28,8 @@ Requires-Dist: apache-airflow-providers-openlineage ; extra == "openlineage"
 Requires-Dist: pandas>=2.1.2,<2.2 ; extra == "pandas" and (python_version>="3.9")
 Requires-Dist: pandas>=1.5.3,<2.2 ; extra == "pandas" and (python_version<"3.9")
 Project-URL: Bug Tracker, https://github.com/apache/airflow/issues
-Project-URL: Changelog, https://airflow.apache.org/docs/apache-airflow-providers-common-sql/1.14.2/changelog.html
-Project-URL: Documentation, https://airflow.apache.org/docs/apache-airflow-providers-common-sql/1.14.2
+Project-URL: Changelog, https://airflow.apache.org/docs/apache-airflow-providers-common-sql/1.15.0/changelog.html
+Project-URL: Documentation, https://airflow.apache.org/docs/apache-airflow-providers-common-sql/1.15.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
@@ -81,7 +81,7 @@ Provides-Extra: pandas
 
 Package ``apache-airflow-providers-common-sql``
 
-Release: ``1.14.2.rc1``
+Release: ``1.15.0.rc1``
 
 
 `Common SQL Provider <https://en.wikipedia.org/wiki/SQL>`__
@@ -94,7 +94,7 @@ This is a provider package for ``common.sql`` provider. All classes for this pro
 are in ``airflow.providers.common.sql`` python package.
 
 You can find package information and changelog for the provider
-in the `documentation <https://airflow.apache.org/docs/apache-airflow-providers-common-sql/1.14.2/>`_.
+in the `documentation <https://airflow.apache.org/docs/apache-airflow-providers-common-sql/1.15.0/>`_.
 
 Installation
 ------------
@@ -136,4 +136,4 @@ Dependent package
 ==============================================================================================================  ===============
 
 The changelog for the provider package can be found in the
-`changelog <https://airflow.apache.org/docs/apache-airflow-providers-common-sql/1.14.2/changelog.html>`_.
+`changelog <https://airflow.apache.org/docs/apache-airflow-providers-common-sql/1.15.0/changelog.html>`_.

{apache_airflow_providers_common_sql-1.14.2rc1.dist-info → apache_airflow_providers_common_sql-1.15.0rc1.dist-info}/RECORD

@@ -1,19 +1,19 @@
 airflow/providers/common/sql/LICENSE,sha256=FFb4jd2AXnOOf7XLP04pQW6jbdhG49TxlGY6fFpCV1Y,13609
 airflow/providers/common/sql/README_API.md,sha256=CxhaS8EedZ4dcbLUPC4-GLCMaY3OH96oHxXttUGU06E,5932
-airflow/providers/common/sql/__init__.py,sha256=_e9jiJfneIarxjGWFuYzbtLFQUoaqUUEfQ0tRZNr3Mc,1498
-airflow/providers/common/sql/get_provider_info.py,sha256=25Djfn7WkF5fNCpnIfjUmqxanaoxOQCGpS6Dw80Z4Uc,3255
+airflow/providers/common/sql/__init__.py,sha256=USPIhUqC6C_FYqv40mxCsNzwjkxUbIjwBXtp6M-AlNI,1498
+airflow/providers/common/sql/get_provider_info.py,sha256=EJiL5nJNV6RzbGj6-ZdsrYtCdeceKwvMO8HBo9B_f-0,3277
 airflow/providers/common/sql/doc/adr/0001-record-architecture-decisions.md,sha256=TfANqrzoFto9PMOMza3MitIkXHGLx2kY_BhhF-N0_ow,1675
 airflow/providers/common/sql/doc/adr/0002-return-common-data-structure-from-dbapihook-derived-hooks.md,sha256=ze5w9IVS-HkUwdZvPW8_JaJaVwel7-N6XdEVN4pTuCE,8457
 airflow/providers/common/sql/hooks/__init__.py,sha256=9hdXHABrVpkbpjZgUft39kOFL2xSGeG4GEua0Hmelus,785
-airflow/providers/common/sql/hooks/sql.py,sha256=Ax3Q1mUsTgIVUWglb6jrXGVONmwL1WJkSLsXRfa6Tqg,29728
-airflow/providers/common/sql/hooks/sql.pyi,sha256=5cERPx3bJVspqP_jDuhg0p4DUhge5W5q_HcJEStfztA,5673
+airflow/providers/common/sql/hooks/sql.py,sha256=KWJNSMCAcr2NX1C_hbXbD8m93HiKrOK3wbHSroqKBsY,30291
+airflow/providers/common/sql/hooks/sql.pyi,sha256=ADJ0aZ5EgFZfSZstNU8JbCdfFgk2f6PlRBvenjMltXA,5773
 airflow/providers/common/sql/operators/__init__.py,sha256=9hdXHABrVpkbpjZgUft39kOFL2xSGeG4GEua0Hmelus,785
 airflow/providers/common/sql/operators/sql.py,sha256=hX42WOWOLcVIORP8TIUTUVoA8g_nF4sucX4Cl2BMXVk,48553
 airflow/providers/common/sql/operators/sql.pyi,sha256=WrzYlWtRT8SGswxPsXMv8oqobrooOlEWeGOjtMnNEBI,8589
 airflow/providers/common/sql/sensors/__init__.py,sha256=9hdXHABrVpkbpjZgUft39kOFL2xSGeG4GEua0Hmelus,785
 airflow/providers/common/sql/sensors/sql.py,sha256=Yz_O4w5zA4HUZSH1f4aR2ry5P1fcTdyRnqIWZ69ggEs,5974
 airflow/providers/common/sql/sensors/sql.pyi,sha256=hDt9OeV5SHpz4s8yHhohrEUzNuKBgT4IHOU5b9rQnKo,2706
-apache_airflow_providers_common_sql-1.14.2rc1.dist-info/entry_points.txt,sha256=h8UXRp2crPuGmYVYRM5oe168qIh7g-4t2QQbVMizKjI,106
-apache_airflow_providers_common_sql-1.14.2rc1.dist-info/WHEEL,sha256=EZbGkh7Ie4PoZfRQ8I0ZuP9VklN_TvcZ6DSE5Uar4z4,81
-apache_airflow_providers_common_sql-1.14.2rc1.dist-info/METADATA,sha256=F55SR9cp9IDYheZD7Ea9noWxsB0qajZ2nxURa11HNaM,6227
-apache_airflow_providers_common_sql-1.14.2rc1.dist-info/RECORD,,
+apache_airflow_providers_common_sql-1.15.0rc1.dist-info/entry_points.txt,sha256=h8UXRp2crPuGmYVYRM5oe168qIh7g-4t2QQbVMizKjI,106
+apache_airflow_providers_common_sql-1.15.0rc1.dist-info/WHEEL,sha256=EZbGkh7Ie4PoZfRQ8I0ZuP9VklN_TvcZ6DSE5Uar4z4,81
+apache_airflow_providers_common_sql-1.15.0rc1.dist-info/METADATA,sha256=7aGK-Q0aTpFtQ2UfJ6j1PDrs1VqY9sDs-oIWLDMwY_U,6227
+apache_airflow_providers_common_sql-1.15.0rc1.dist-info/RECORD,,