apache-airflow-providers-dbt-cloud 3.7.1rc1__py3-none-any.whl → 3.8.0rc1__py3-none-any.whl

airflow/providers/dbt/cloud/__init__.py

@@ -27,7 +27,7 @@ import packaging.version
 
 __all__ = ["__version__"]
 
-__version__ = "3.7.1"
+__version__ = "3.8.0"
 
 try:
     from airflow import __version__ as airflow_version
@@ -35,8 +35,8 @@ except ImportError:
     from airflow.version import version as airflow_version
 
 if packaging.version.parse(packaging.version.parse(airflow_version).base_version) < packaging.version.parse(
-    "2.6.0"
+    "2.7.0"
 ):
     raise RuntimeError(
-        f"The package `apache-airflow-providers-dbt-cloud:{__version__}` needs Apache Airflow 2.6.0+"
+        f"The package `apache-airflow-providers-dbt-cloud:{__version__}` needs Apache Airflow 2.7.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": 1712665602,
+        "source-date-epoch": 1714476223,
         "versions": [
+            "3.8.0",
             "3.7.1",
             "3.7.0",
             "3.6.1",
@@ -56,7 +57,7 @@ def get_provider_info():
             "1.0.1",
         ],
         "dependencies": [
-            "apache-airflow>=2.6.0",
+            "apache-airflow>=2.7.0",
             "apache-airflow-providers-http",
             "asgiref",
             "aiohttp>=3.9.2",

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

@@ -165,7 +165,7 @@ def provide_account_id(func: T) -> T:
 
 class DbtCloudHook(HttpHook):
     """
-    Interact with dbt Cloud using the V2 API.
+    Interact with dbt Cloud using the V2 (V3 if supported) API.
 
     :param dbt_cloud_conn_id: The ID of the :ref:`dbt Cloud connection <howto/connection:dbt-cloud>`.
     """
@@ -194,7 +194,7 @@ class DbtCloudHook(HttpHook):
 
     @staticmethod
     def get_request_url_params(
-        tenant: str, endpoint: str, include_related: list[str] | None = None
+        tenant: str, endpoint: str, include_related: list[str] | None = None, *, api_version: str = "v2"
     ) -> tuple[str, dict[str, Any]]:
         """
         Form URL from base url and endpoint url.
@@ -207,7 +207,7 @@ class DbtCloudHook(HttpHook):
         data: dict[str, Any] = {}
         if include_related:
             data = {"include_related": include_related}
-        url = f"https://{tenant}/api/v2/accounts/{endpoint or ''}"
+        url = f"https://{tenant}/api/{api_version}/accounts/{endpoint or ''}"
         return url, data
 
     async def get_headers_tenants_from_connection(self) -> tuple[dict[str, Any], str]:
@@ -270,7 +270,7 @@ class DbtCloudHook(HttpHook):
 
     def get_conn(self, *args, **kwargs) -> Session:
         tenant = self._get_tenant_domain(self.connection)
-        self.base_url = f"https://{tenant}/api/v2/accounts/"
+        self.base_url = f"https://{tenant}/"
 
         session = Session()
         session.auth = self.auth_type(self.connection.password)
@@ -298,23 +298,26 @@ class DbtCloudHook(HttpHook):
 
     def _run_and_get_response(
         self,
+        *,
         method: str = "GET",
         endpoint: str | None = None,
         payload: str | dict[str, Any] | None = None,
         paginate: bool = False,
+        api_version: str = "v2",
     ) -> Any:
         self.method = method
+        full_endpoint = f"api/{api_version}/accounts/{endpoint}" if endpoint else None
 
         if paginate:
             if isinstance(payload, str):
                 raise ValueError("Payload cannot be a string to paginate a response.")
 
-            if endpoint:
-                return self._paginate(endpoint=endpoint, payload=payload)
-            else:
-                raise ValueError("An endpoint is needed to paginate a response.")
+            if full_endpoint:
+                return self._paginate(endpoint=full_endpoint, payload=payload)
 
-        return self.run(endpoint=endpoint, data=payload)
+            raise ValueError("An endpoint is needed to paginate a response.")
+
+        return self.run(endpoint=full_endpoint, data=payload)
 
     def list_accounts(self) -> list[Response]:
         """
@@ -342,7 +345,7 @@ class DbtCloudHook(HttpHook):
         :param account_id: Optional. The ID of a dbt Cloud account.
         :return: List of request responses.
         """
-        return self._run_and_get_response(endpoint=f"{account_id}/projects/", paginate=True)
+        return self._run_and_get_response(endpoint=f"{account_id}/projects/", paginate=True, api_version="v3")
 
     @fallback_to_default_account
     def get_project(self, project_id: int, account_id: int | None = None) -> Response:
@@ -353,7 +356,7 @@ class DbtCloudHook(HttpHook):
         :param account_id: Optional. The ID of a dbt Cloud account.
         :return: The request response.
         """
-        return self._run_and_get_response(endpoint=f"{account_id}/projects/{project_id}/")
+        return self._run_and_get_response(endpoint=f"{account_id}/projects/{project_id}/", api_version="v3")
 
     @fallback_to_default_account
     def list_jobs(

airflow/providers/dbt/cloud/operators/dbt.py

@@ -135,7 +135,7 @@ class DbtCloudRunJobOperator(BaseOperator):
                 account_id=self.account_id,
                 payload={
                     "job_definition_id": self.job_id,
-                    "status": DbtCloudJobRunStatus.NON_TERMINAL_STATUSES,
+                    "status__in": DbtCloudJobRunStatus.NON_TERMINAL_STATUSES,
                     "order_by": "-created_at",
                 },
             ).json()["data"]

{apache_airflow_providers_dbt_cloud-3.7.1rc1.dist-info → apache_airflow_providers_dbt_cloud-3.8.0rc1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: apache-airflow-providers-dbt-cloud
-Version: 3.7.1rc1
+Version: 3.8.0rc1
 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>
@@ -23,13 +23,13 @@ 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.6.0rc0
+Requires-Dist: apache-airflow>=2.7.0rc0
 Requires-Dist: asgiref
 Requires-Dist: apache-airflow-providers-http ; extra == "http"
 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-dbt-cloud/3.7.1/changelog.html
-Project-URL: Documentation, https://airflow.apache.org/docs/apache-airflow-providers-dbt-cloud/3.7.1
+Project-URL: Changelog, https://airflow.apache.org/docs/apache-airflow-providers-dbt-cloud/3.8.0/changelog.html
+Project-URL: Documentation, https://airflow.apache.org/docs/apache-airflow-providers-dbt-cloud/3.8.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: openlineage
 
 Package ``apache-airflow-providers-dbt-cloud``
 
-Release: ``3.7.1.rc1``
+Release: ``3.8.0.rc1``
 
 
 `dbt Cloud <https://www.getdbt.com/product/dbt-cloud/>`__
@@ -94,7 +94,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.7.1/>`_.
+in the `documentation <https://airflow.apache.org/docs/apache-airflow-providers-dbt-cloud/3.8.0/>`_.
 
 Installation
 ------------
@@ -111,7 +111,7 @@ Requirements
 =================================  ==================
 PIP package                        Version required
 =================================  ==================
-``apache-airflow``                 ``>=2.6.0``
+``apache-airflow``                 ``>=2.7.0``
 ``apache-airflow-providers-http``
 ``asgiref``
 ``aiohttp``                        ``>=3.9.2``
@@ -138,4 +138,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.7.1/changelog.html>`_.
+`changelog <https://airflow.apache.org/docs/apache-airflow-providers-dbt-cloud/3.8.0/changelog.html>`_.

{apache_airflow_providers_dbt_cloud-3.7.1rc1.dist-info → apache_airflow_providers_dbt_cloud-3.8.0rc1.dist-info}/RECORD

@@ -1,17 +1,17 @@
 airflow/providers/dbt/cloud/LICENSE,sha256=ywUBpKZc7Jb96rVt5I3IDbg7dIJAbUSHkuoDcF3jbH4,13569
-airflow/providers/dbt/cloud/__init__.py,sha256=hUhn7pHUCF6aK-img5tLgPvG4I5FSxq_vAqYbWyA9IA,1584
-airflow/providers/dbt/cloud/get_provider_info.py,sha256=tZVUiIxNYsyDBnKudn9UQw2QV4E-tMIwLnUdcXm2At0,3308
+airflow/providers/dbt/cloud/__init__.py,sha256=DT9jyekBWhbypsfVRXU_S0MOw1olXQDGm97g2BmRucQ,1584
+airflow/providers/dbt/cloud/get_provider_info.py,sha256=RM3t64XWAmgADWrZKsRvq-RGwqnmfC1mCy8CODzO0xg,3329
 airflow/providers/dbt/cloud/hooks/__init__.py,sha256=9hdXHABrVpkbpjZgUft39kOFL2xSGeG4GEua0Hmelus,785
-airflow/providers/dbt/cloud/hooks/dbt.py,sha256=okv5AeLO-KI7kVUvTO98_6r-oytl8CqKGDg3uUCwPIo,26082
+airflow/providers/dbt/cloud/hooks/dbt.py,sha256=CleiIF1sfY7lY-r-5x3GzpWAciAHSNSy-A4ce4uMM2M,26284
 airflow/providers/dbt/cloud/operators/__init__.py,sha256=9hdXHABrVpkbpjZgUft39kOFL2xSGeG4GEua0Hmelus,785
-airflow/providers/dbt/cloud/operators/dbt.py,sha256=TW1tpuIWm4EgoEpv4wGtrXcR_Ve32vFQfUbsJzXMseA,15630
+airflow/providers/dbt/cloud/operators/dbt.py,sha256=7tNFqg4S9YYxvua8KE84keM3CM9UTcw-_m2lP7M_eYY,15634
 airflow/providers/dbt/cloud/sensors/__init__.py,sha256=mlJxuZLkd5x-iq2SBwD3mvRQpt3YR7wjz_nceyF1IaI,787
 airflow/providers/dbt/cloud/sensors/dbt.py,sha256=X8Q1YLj76qwj3nAy6a_aF4edID-oJWMfFLVSrrQMZ1k,6860
 airflow/providers/dbt/cloud/triggers/__init__.py,sha256=9hdXHABrVpkbpjZgUft39kOFL2xSGeG4GEua0Hmelus,785
 airflow/providers/dbt/cloud/triggers/dbt.py,sha256=pd7b6SF4IEQrZHb8xNIPHIsIbaKwom2h9eMUU-THMM4,4675
 airflow/providers/dbt/cloud/utils/__init__.py,sha256=9hdXHABrVpkbpjZgUft39kOFL2xSGeG4GEua0Hmelus,785
 airflow/providers/dbt/cloud/utils/openlineage.py,sha256=mLqqhJfH9W_dn_XME0T1JORCtUX7I7LaCgPUHf8-hsc,5675
-apache_airflow_providers_dbt_cloud-3.7.1rc1.dist-info/entry_points.txt,sha256=c18L1WEEK18WQeEGrm9kMVqutiYJHiWGH5jU1JqnToE,105
-apache_airflow_providers_dbt_cloud-3.7.1rc1.dist-info/WHEEL,sha256=EZbGkh7Ie4PoZfRQ8I0ZuP9VklN_TvcZ6DSE5Uar4z4,81
-apache_airflow_providers_dbt_cloud-3.7.1rc1.dist-info/METADATA,sha256=lH5NnqkWoe2IgxfE_znT9nQgqdZ1qJhUBCIQfP4G2W8,6353
-apache_airflow_providers_dbt_cloud-3.7.1rc1.dist-info/RECORD,,
+apache_airflow_providers_dbt_cloud-3.8.0rc1.dist-info/entry_points.txt,sha256=c18L1WEEK18WQeEGrm9kMVqutiYJHiWGH5jU1JqnToE,105
+apache_airflow_providers_dbt_cloud-3.8.0rc1.dist-info/WHEEL,sha256=EZbGkh7Ie4PoZfRQ8I0ZuP9VklN_TvcZ6DSE5Uar4z4,81
+apache_airflow_providers_dbt_cloud-3.8.0rc1.dist-info/METADATA,sha256=_BPtwiMa5C4DCPRnYcALFg6fkktSjks6pkL_Iy5HB_g,6353
+apache_airflow_providers_dbt_cloud-3.8.0rc1.dist-info/RECORD,,