apache-airflow-providers-dbt-cloud 3.11.2rc1__py3-none-any.whl → 4.0.0__py3-none-any.whl

airflow/providers/dbt/cloud/__init__.py

@@ -29,11 +29,11 @@ from airflow import __version__ as airflow_version
 
 __all__ = ["__version__"]
 
-__version__ = "3.11.2"
+__version__ = "4.0.0"
 
 if packaging.version.parse(packaging.version.parse(airflow_version).base_version) < packaging.version.parse(
-    "2.8.0"
+    "2.9.0"
 ):
     raise RuntimeError(
-        f"The package `apache-airflow-providers-dbt-cloud:{__version__}` needs Apache Airflow 2.8.0+"
+        f"The package `apache-airflow-providers-dbt-cloud:{__version__}` needs Apache Airflow 2.9.0+"
     )

airflow/providers/dbt/cloud/get_provider_info.py

@@ -28,8 +28,9 @@ def get_provider_info():
         "name": "dbt Cloud",
         "description": "`dbt Cloud <https://www.getdbt.com/product/dbt-cloud/>`__\n",
         "state": "ready",
-        "source-date-epoch": 1731569996,
+        "source-date-epoch": 1734533324,
         "versions": [
+            "4.0.0",
             "3.11.2",
             "3.11.1",
             "3.11.0",
@@ -64,7 +65,7 @@ def get_provider_info():
             "1.0.1",
         ],
         "dependencies": [
-            "apache-airflow>=2.8.0",
+            "apache-airflow>=2.9.0",
             "apache-airflow-providers-http",
             "asgiref>=2.3.0",
             "aiohttp>=3.9.2",

airflow/providers/dbt/cloud/hooks/dbt.py

@@ -336,7 +336,6 @@ class DbtCloudHook(HttpHook):
 
             raise ValueError("An endpoint is needed to paginate a response.")
 
-        # breakpoint()
         return self.run(endpoint=full_endpoint, data=payload, extra_options=extra_options)
 
     def list_accounts(self) -> list[Response]:

airflow/providers/dbt/cloud/sensors/dbt.py

@@ -17,14 +17,11 @@
 from __future__ import annotations
 
 import time
-import warnings
 from functools import cached_property
 from typing import TYPE_CHECKING, Any
 
-from deprecated import deprecated
-
 from airflow.configuration import conf
-from airflow.exceptions import AirflowException, AirflowProviderDeprecationWarning
+from airflow.exceptions import AirflowException
 from airflow.providers.dbt.cloud.hooks.dbt import DbtCloudHook, DbtCloudJobRunException, DbtCloudJobRunStatus
 from airflow.providers.dbt.cloud.triggers.dbt import DbtCloudRunJobTrigger
 from airflow.providers.dbt.cloud.utils.openlineage import generate_openlineage_events_from_dbt_cloud_run
@@ -62,17 +59,7 @@ class DbtCloudJobRunSensor(BaseSensorOperator):
     ) -> None:
         if deferrable:
             if "poke_interval" not in kwargs:
-                # TODO: Remove once deprecated
-                if "polling_interval" in kwargs:
-                    kwargs["poke_interval"] = kwargs["polling_interval"]
-                    warnings.warn(
-                        "Argument `poll_interval` is deprecated and will be removed "
-                        "in a future release.  Please use `poke_interval` instead.",
-                        AirflowProviderDeprecationWarning,
-                        stacklevel=2,
-                    )
-                else:
-                    kwargs["poke_interval"] = 5
+                kwargs["poke_interval"] = 5
 
                 if "timeout" not in kwargs:
                     kwargs["timeout"] = 60 * 60 * 24 * 7
@@ -142,22 +129,3 @@ class DbtCloudJobRunSensor(BaseSensorOperator):
     def get_openlineage_facets_on_complete(self, task_instance) -> OperatorLineage:
         """Implement _on_complete because job_run needs to be triggered first in execute method."""
         return generate_openlineage_events_from_dbt_cloud_run(operator=self, task_instance=task_instance)
-
-
-@deprecated(
-    reason=(
-        "Class `DbtCloudJobRunAsyncSensor` is deprecated and will be removed in a future release. "
-        "Please use `DbtCloudJobRunSensor` and set `deferrable` attribute to `True` instead"
-    ),
-    category=AirflowProviderDeprecationWarning,
-)
-class DbtCloudJobRunAsyncSensor(DbtCloudJobRunSensor):
-    """
-    This class is deprecated.
-
-    Please use :class:`airflow.providers.dbt.cloud.sensor.dbt.DbtCloudJobRunSensor`
-    with ``deferrable=True``.
-    """
-
-    def __init__(self, **kwargs: Any) -> None:
-        super().__init__(deferrable=True, **kwargs)

airflow/providers/dbt/cloud/triggers/dbt.py

@@ -18,7 +18,8 @@ from __future__ import annotations
 
 import asyncio
 import time
-from typing import Any, AsyncIterator
+from collections.abc import AsyncIterator
+from typing import Any
 
 from airflow.providers.dbt.cloud.hooks.dbt import DbtCloudHook, DbtCloudJobRunStatus
 from airflow.triggers.base import BaseTrigger, TriggerEvent

airflow/providers/dbt/cloud/utils/openlineage.py

@@ -21,20 +21,32 @@ import re
 from contextlib import suppress
 from typing import TYPE_CHECKING
 
+from packaging.version import parse
+
 from airflow import __version__ as airflow_version
 
 if TYPE_CHECKING:
+    from packaging.version import Version
+
     from airflow.models.taskinstance import TaskInstance
     from airflow.providers.dbt.cloud.operators.dbt import DbtCloudRunJobOperator
     from airflow.providers.dbt.cloud.sensors.dbt import DbtCloudJobRunSensor
     from airflow.providers.openlineage.extractors.base import OperatorLineage
 
 
+_AIRFLOW_VERSION: Version = parse(parse(airflow_version).base_version)
+
+
+def _get_logical_date(task_instance):
+    # todo: remove when min airflow version >= 3.0
+    if parse("3") > _AIRFLOW_VERSION:
+        return task_instance.execution_date
+    return task_instance.logical_date
+
+
 def _get_try_number(val):
     # todo: remove when min airflow version >= 2.10.0
-    from packaging.version import parse
-
-    if parse(parse(airflow_version).base_version) < parse("2.10.0"):
+    if parse("2.10.0") > _AIRFLOW_VERSION:
         return val.try_number - 1
     else:
         return val.try_number
@@ -142,8 +154,9 @@ def generate_openlineage_events_from_dbt_cloud_run(
         parent_run_id = OpenLineageAdapter.build_task_instance_run_id(
             dag_id=task_instance.dag_id,
             task_id=operator.task_id,
-            execution_date=task_instance.execution_date,
+            logical_date=_get_logical_date(task_instance),
             try_number=_get_try_number(task_instance),
+            map_index=task_instance.map_index,
         )
 
         parent_job = ParentRunMetadata(

{apache_airflow_providers_dbt_cloud-3.11.2rc1.dist-info → apache_airflow_providers_dbt_cloud-4.0.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: apache-airflow-providers-dbt-cloud
-Version: 3.11.2rc1
+Version: 4.0.0
 Summary: Provider package apache-airflow-providers-dbt-cloud for Apache Airflow
 Keywords: airflow-provider,dbt.cloud,airflow,integration
 Author-email: Apache Software Foundation <dev@airflow.apache.org>
@@ -22,18 +22,16 @@ Classifier: Programming Language :: Python :: 3.12
 Classifier: Topic :: System :: Monitoring
 Requires-Dist: aiohttp>=3.9.2
 Requires-Dist: apache-airflow-providers-http
-Requires-Dist: apache-airflow>=2.8.0rc0
+Requires-Dist: apache-airflow>=2.9.0
 Requires-Dist: asgiref>=2.3.0
-Requires-Dist: apache-airflow-providers-http ; extra == "http"
-Requires-Dist: apache-airflow-providers-openlineage>=1.7.0rc0 ; extra == "openlineage"
+Requires-Dist: apache-airflow-providers-openlineage>=1.7.0 ; extra == "openlineage"
 Project-URL: Bug Tracker, https://github.com/apache/airflow/issues
-Project-URL: Changelog, https://airflow.apache.org/docs/apache-airflow-providers-dbt-cloud/3.11.2/changelog.html
-Project-URL: Documentation, https://airflow.apache.org/docs/apache-airflow-providers-dbt-cloud/3.11.2
+Project-URL: Changelog, https://airflow.apache.org/docs/apache-airflow-providers-dbt-cloud/4.0.0/changelog.html
+Project-URL: Documentation, https://airflow.apache.org/docs/apache-airflow-providers-dbt-cloud/4.0.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
+Project-URL: Twitter, https://x.com/ApacheAirflow
 Project-URL: YouTube, https://www.youtube.com/channel/UCSXwxpWZQ7XZ1WL3wqevChA/
-Provides-Extra: http
 Provides-Extra: openlineage
 
 
@@ -80,7 +78,7 @@ Provides-Extra: openlineage
 
 Package ``apache-airflow-providers-dbt-cloud``
 
-Release: ``3.11.2.rc1``
+Release: ``4.0.0``
 
 
 `dbt Cloud <https://www.getdbt.com/product/dbt-cloud/>`__
@@ -93,7 +91,7 @@ This is a provider package for ``dbt.cloud`` provider. All classes for this prov
 are in ``airflow.providers.dbt.cloud`` python package.
 
 You can find package information and changelog for the provider
-in the `documentation <https://airflow.apache.org/docs/apache-airflow-providers-dbt-cloud/3.11.2/>`_.
+in the `documentation <https://airflow.apache.org/docs/apache-airflow-providers-dbt-cloud/4.0.0/>`_.
 
 Installation
 ------------
@@ -110,7 +108,7 @@ Requirements
 =================================  ==================
 PIP package                        Version required
 =================================  ==================
-``apache-airflow``                 ``>=2.8.0``
+``apache-airflow``                 ``>=2.9.0``
 ``apache-airflow-providers-http``
 ``asgiref``                        ``>=2.3.0``
 ``aiohttp``                        ``>=3.9.2``
@@ -137,4 +135,4 @@ Dependent package
 ==============================================================================================================  ===============
 
 The changelog for the provider package can be found in the
-`changelog <https://airflow.apache.org/docs/apache-airflow-providers-dbt-cloud/3.11.2/changelog.html>`_.
+`changelog <https://airflow.apache.org/docs/apache-airflow-providers-dbt-cloud/4.0.0/changelog.html>`_.

apache_airflow_providers_dbt_cloud-4.0.0.dist-info/RECORD

@@ -0,0 +1,17 @@
+airflow/providers/dbt/cloud/LICENSE,sha256=FFb4jd2AXnOOf7XLP04pQW6jbdhG49TxlGY6fFpCV1Y,13609
+airflow/providers/dbt/cloud/__init__.py,sha256=aFLIA3l_kzPW3lUS3Wtr65D9ZDCRtahs78rMwWtOgDM,1496
+airflow/providers/dbt/cloud/get_provider_info.py,sha256=WrJzEFLJMZk9LoIsrVTvM6rcGz1T2xJSy4wuCzTPF8Y,3777
+airflow/providers/dbt/cloud/hooks/__init__.py,sha256=9hdXHABrVpkbpjZgUft39kOFL2xSGeG4GEua0Hmelus,785
+airflow/providers/dbt/cloud/hooks/dbt.py,sha256=Zkrpnis7OVPjFGly7dK8dvSFpEfbSFrjl8CdnOO6u8M,29192
+airflow/providers/dbt/cloud/operators/__init__.py,sha256=9hdXHABrVpkbpjZgUft39kOFL2xSGeG4GEua0Hmelus,785
+airflow/providers/dbt/cloud/operators/dbt.py,sha256=xilyondmB5yDWxh0gfaJKKWZYN52diFXsfrdBsn_1UE,16222
+airflow/providers/dbt/cloud/sensors/__init__.py,sha256=mlJxuZLkd5x-iq2SBwD3mvRQpt3YR7wjz_nceyF1IaI,787
+airflow/providers/dbt/cloud/sensors/dbt.py,sha256=OGf-VNKcCNCrImrdE8PudmtDak_MKJfPhmaf7_ccOLg,5157
+airflow/providers/dbt/cloud/triggers/__init__.py,sha256=9hdXHABrVpkbpjZgUft39kOFL2xSGeG4GEua0Hmelus,785
+airflow/providers/dbt/cloud/triggers/dbt.py,sha256=Oabdc7FcNhCQxkjDC5SqAiYEw4hSZ9mQGZgSt36a1E0,4707
+airflow/providers/dbt/cloud/utils/__init__.py,sha256=9hdXHABrVpkbpjZgUft39kOFL2xSGeG4GEua0Hmelus,785
+airflow/providers/dbt/cloud/utils/openlineage.py,sha256=6hGaRpfPE0HJSyEQNo4xt9Pfvdi9GI-mysIb4dBACpk,6611
+apache_airflow_providers_dbt_cloud-4.0.0.dist-info/entry_points.txt,sha256=c18L1WEEK18WQeEGrm9kMVqutiYJHiWGH5jU1JqnToE,105
+apache_airflow_providers_dbt_cloud-4.0.0.dist-info/WHEEL,sha256=CpUCUxeHQbRN5UGRQHYRJorO5Af-Qy_fHMctcQ8DSGI,82
+apache_airflow_providers_dbt_cloud-4.0.0.dist-info/METADATA,sha256=GiaeZfIhEZyVPl5FJMixWY2uzpGB6iaU4pNw16PfVO4,6248
+apache_airflow_providers_dbt_cloud-4.0.0.dist-info/RECORD,,

apache_airflow_providers_dbt_cloud-3.11.2rc1.dist-info/RECORD

@@ -1,17 +0,0 @@
-airflow/providers/dbt/cloud/LICENSE,sha256=FFb4jd2AXnOOf7XLP04pQW6jbdhG49TxlGY6fFpCV1Y,13609
-airflow/providers/dbt/cloud/__init__.py,sha256=ulO4-15mf9wqC44nUFtSoVfli1A7pehMc_WWeMxqb98,1497
-airflow/providers/dbt/cloud/get_provider_info.py,sha256=1_319hwhM2UgCND23fYFJzGTFOhFS5i6yMo1rBz6Nq4,3756
-airflow/providers/dbt/cloud/hooks/__init__.py,sha256=9hdXHABrVpkbpjZgUft39kOFL2xSGeG4GEua0Hmelus,785
-airflow/providers/dbt/cloud/hooks/dbt.py,sha256=ExKsYoJOcPgCKf_QcB7XPYcXVe69czXDZT_XdmoJyMk,29215
-airflow/providers/dbt/cloud/operators/__init__.py,sha256=9hdXHABrVpkbpjZgUft39kOFL2xSGeG4GEua0Hmelus,785
-airflow/providers/dbt/cloud/operators/dbt.py,sha256=xilyondmB5yDWxh0gfaJKKWZYN52diFXsfrdBsn_1UE,16222
-airflow/providers/dbt/cloud/sensors/__init__.py,sha256=mlJxuZLkd5x-iq2SBwD3mvRQpt3YR7wjz_nceyF1IaI,787
-airflow/providers/dbt/cloud/sensors/dbt.py,sha256=23kRgLxST-WJKpF1NGk_CRtQGn91kNw7Z_2ZTjzcQh0,6357
-airflow/providers/dbt/cloud/triggers/__init__.py,sha256=9hdXHABrVpkbpjZgUft39kOFL2xSGeG4GEua0Hmelus,785
-airflow/providers/dbt/cloud/triggers/dbt.py,sha256=30fntKeUi1bUJtv50Rv09KtJNsN8gBtjLPsf3Xoccks,4680
-airflow/providers/dbt/cloud/utils/__init__.py,sha256=9hdXHABrVpkbpjZgUft39kOFL2xSGeG4GEua0Hmelus,785
-airflow/providers/dbt/cloud/utils/openlineage.py,sha256=tLHfM6L9cOy4ZvpnJFjJAI3qxAztn073GbLHsXpjvFQ,6265
-apache_airflow_providers_dbt_cloud-3.11.2rc1.dist-info/entry_points.txt,sha256=c18L1WEEK18WQeEGrm9kMVqutiYJHiWGH5jU1JqnToE,105
-apache_airflow_providers_dbt_cloud-3.11.2rc1.dist-info/WHEEL,sha256=CpUCUxeHQbRN5UGRQHYRJorO5Af-Qy_fHMctcQ8DSGI,82
-apache_airflow_providers_dbt_cloud-3.11.2rc1.dist-info/METADATA,sha256=sF3QXnPMmU-ZjUJ7z4lGSfKw9F-FsMjOaHcLjCc5rng,6357
-apache_airflow_providers_dbt_cloud-3.11.2rc1.dist-info/RECORD,,