airflow-dbt-python 3.0.3__tar.gz → 3.1.0__tar.gz

{airflow_dbt_python-3.0.3 → airflow_dbt_python-3.1.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: airflow-dbt-python
-Version: 3.0.3
+Version: 3.1.0
 Summary: A collection of Airflow operators, hooks, and utilities to execute dbt commands
 Author-email: Tomás Farías Santana <tomas@tomasfarias.dev>
 License-Expression: MIT
@@ -51,7 +51,7 @@ Dynamic: license-file
 
 [![PyPI version](https://img.shields.io/pypi/v/airflow-dbt-python?style=plastic)](https://pypi.org/project/airflow-dbt-python/)
 [![CI testing](https://github.com/tomasfarias/airflow-dbt-python/actions/workflows/ci.yaml/badge.svg)](https://github.com/tomasfarias/airflow-dbt-python/actions)
-[![Code style: black](https://img.shields.io/badge/code%20style-black-000000.svg)](https://github.com/psf/black)
+[![Code style: Ruff](https://img.shields.io/endpoint?url=https://raw.githubusercontent.com/astral-sh/ruff/main/assets/badge/v2.json)](https://github.com/astral-sh/ruff)
 [![Test coverage](https://img.shields.io/endpoint?url=https://gist.githubusercontent.com/tomasfarias/81ef37701aa088d18db8a58ce07c79c7/raw/covbadge.json)](https://github.com/tomasfarias/airflow-dbt-python/actions)
 [![Documentation](https://readthedocs.org/projects/airflow-dbt-python/badge/?version=latest)](https://airflow-dbt-python.readthedocs.io/en/latest/?badge=latest)
 

{airflow_dbt_python-3.0.3 → airflow_dbt_python-3.1.0}/README.md

@@ -2,7 +2,7 @@
 
 [![PyPI version](https://img.shields.io/pypi/v/airflow-dbt-python?style=plastic)](https://pypi.org/project/airflow-dbt-python/)
 [![CI testing](https://github.com/tomasfarias/airflow-dbt-python/actions/workflows/ci.yaml/badge.svg)](https://github.com/tomasfarias/airflow-dbt-python/actions)
-[![Code style: black](https://img.shields.io/badge/code%20style-black-000000.svg)](https://github.com/psf/black)
+[![Code style: Ruff](https://img.shields.io/endpoint?url=https://raw.githubusercontent.com/astral-sh/ruff/main/assets/badge/v2.json)](https://github.com/astral-sh/ruff)
 [![Test coverage](https://img.shields.io/endpoint?url=https://gist.githubusercontent.com/tomasfarias/81ef37701aa088d18db8a58ce07c79c7/raw/covbadge.json)](https://github.com/tomasfarias/airflow-dbt-python/actions)
 [![Documentation](https://readthedocs.org/projects/airflow-dbt-python/badge/?version=latest)](https://airflow-dbt-python.readthedocs.io/en/latest/?badge=latest)
 

{airflow_dbt_python-3.0.3 → airflow_dbt_python-3.1.0}/airflow_dbt_python/__version__.py

@@ -3,4 +3,4 @@
 __author__ = "Tomás Farías Santana"
 __copyright__ = "Copyright 2021 Tomás Farías Santana"
 __title__ = "airflow-dbt-python"
-__version__ = "3.0.3"
+__version__ = "3.1.0"

{airflow_dbt_python-3.0.3 → airflow_dbt_python-3.1.0}/airflow_dbt_python/hooks/target.py

@@ -492,3 +492,38 @@ class DbtSparkHook(DbtConnectionHook):
         ),
     ]
     conn_extra_params = []
+
+
+class DbtTrinoHook(DbtConnectionHook):
+    """A hook to interact with dbt using a Trino connection."""
+
+    conn_type = "trino"
+    hook_name = "dbt Trino Hook"
+    airflow_conn_types = (conn_type,)
+
+    conn_params = [
+        "host",
+        DbtConnectionParam("schema", default="public"),
+        DbtConnectionParam("login", "user"),
+        "password",
+        DbtConnectionParam("port", default=443),
+    ]
+
+    conn_extra_params = [
+        DbtConnectionParam("type", default="trino"),
+        DbtConnectionParam("method", default="none"),
+        DbtConnectionParam("http_scheme", default="https"),
+        DbtConnectionParam("verify", default=True),
+        DbtConnectionParam("database", "catalog", "hive"),
+        "catalog",
+        "host",
+        "port",
+        DbtConnectionParam("user", "user"),
+        "password",
+        "schema",
+        "session_properties",
+        "http_headers",
+        "role",
+        "client_tags",
+        "source",
+    ]

{airflow_dbt_python-3.0.3 → airflow_dbt_python-3.1.0}/airflow_dbt_python/utils/configs.py

@@ -39,7 +39,10 @@ from dbt.task.test import TestTask
 from dbt.tracking import initialize_from_flags
 
 from airflow_dbt_python.utils.enums import FromStrEnum, LogFormat, Output
-from airflow_dbt_python.utils.version import DBT_INSTALLED_GTE_1_9
+from airflow_dbt_python.utils.version import (
+    DBT_INSTALLED_GTE_1_9,
+    DBT_INSTALLED_GTE_1_10_7,
+)
 
 
 def parse_yaml_args(args: Optional[Union[str, dict[str, Any]]]) -> dict[str, Any]:
@@ -360,7 +363,11 @@ class BaseConfig:
 
         runtime_config = self.create_runtime_config(project, profile)
 
-        if issubclass(self.dbt_task, ConfiguredTask) and runtime_config:
+        if (
+            issubclass(self.dbt_task, ConfiguredTask)
+            and runtime_config
+            and not DBT_INSTALLED_GTE_1_10_7
+        ):
             manifest = parse_manifest(
                 runtime_config,
                 write_perf_info=write_perf_info,
@@ -370,6 +377,22 @@ class BaseConfig:
             task: BaseTask = self.dbt_task(
                 args=local_flags, config=runtime_config, manifest=manifest
             )
+        elif (
+            issubclass(self.dbt_task, ConfiguredTask)
+            and runtime_config
+            and DBT_INSTALLED_GTE_1_10_7
+        ):
+            manifest = parse_manifest(
+                runtime_config,
+                write_perf_info=write_perf_info,
+                write=False,
+                write_json=False,
+                # TODO: Support for catalog integrations
+                active_integrations=[],  # type: ignore
+            )
+            task = self.dbt_task(
+                args=local_flags, config=runtime_config, manifest=manifest
+            )
         elif issubclass(self.dbt_task, DepsTask):
             task = self.dbt_task(args=local_flags, project=project)
         elif issubclass(self.dbt_task, DebugTask):

{airflow_dbt_python-3.0.3 → airflow_dbt_python-3.1.0}/airflow_dbt_python/utils/version.py

@@ -16,11 +16,15 @@ DBT_1_8 = VersionSpecifier(
 DBT_1_9 = VersionSpecifier(
     major="1", minor="9", patch="0", matcher=Matchers.GREATER_THAN_OR_EQUAL
 )
+DBT_1_10_7 = VersionSpecifier(
+    major="1", minor="10", patch="7", matcher=Matchers.GREATER_THAN_OR_EQUAL
+)
 DBT_2_0 = VersionSpecifier(
     major="2", minor="0", patch="0", matcher=Matchers.GREATER_THAN_OR_EQUAL
 )
 
 DBT_INSTALLED_GTE_1_9 = installed.compare(DBT_1_9) == 1
+DBT_INSTALLED_GTE_1_10_7 = installed.compare(DBT_1_10_7) == 1
 
 DBT_INSTALLED_1_8 = DBT_1_8 < installed < DBT_1_9
 DBT_INSTALLED_1_9 = DBT_1_9 < installed < DBT_2_0

{airflow_dbt_python-3.0.3 → airflow_dbt_python-3.1.0}/airflow_dbt_python.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: airflow-dbt-python
-Version: 3.0.3
+Version: 3.1.0
 Summary: A collection of Airflow operators, hooks, and utilities to execute dbt commands
 Author-email: Tomás Farías Santana <tomas@tomasfarias.dev>
 License-Expression: MIT
@@ -51,7 +51,7 @@ Dynamic: license-file
 
 [![PyPI version](https://img.shields.io/pypi/v/airflow-dbt-python?style=plastic)](https://pypi.org/project/airflow-dbt-python/)
 [![CI testing](https://github.com/tomasfarias/airflow-dbt-python/actions/workflows/ci.yaml/badge.svg)](https://github.com/tomasfarias/airflow-dbt-python/actions)
-[![Code style: black](https://img.shields.io/badge/code%20style-black-000000.svg)](https://github.com/psf/black)
+[![Code style: Ruff](https://img.shields.io/endpoint?url=https://raw.githubusercontent.com/astral-sh/ruff/main/assets/badge/v2.json)](https://github.com/astral-sh/ruff)
 [![Test coverage](https://img.shields.io/endpoint?url=https://gist.githubusercontent.com/tomasfarias/81ef37701aa088d18db8a58ce07c79c7/raw/covbadge.json)](https://github.com/tomasfarias/airflow-dbt-python/actions)
 [![Documentation](https://readthedocs.org/projects/airflow-dbt-python/badge/?version=latest)](https://airflow-dbt-python.readthedocs.io/en/latest/?badge=latest)
 

{airflow_dbt_python-3.0.3 → airflow_dbt_python-3.1.0}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "airflow-dbt-python"
-version = "3.0.3"
+version = "3.1.0"
 description = "A collection of Airflow operators, hooks, and utilities to execute dbt commands"
 authors = [{ name = "Tomás Farías Santana", email = "tomas@tomasfarias.dev" }]
 license = "MIT"