Add dbt build operators

cosmos/__init__.py

@@ -20,6 +20,7 @@
 from cosmos.log import get_logger
 from cosmos.operators.lazy_load import MissingPackage
 from cosmos.operators.local import (
+    DbtBuildLocalOperator,
     DbtDepsLocalOperator,
     DbtLSLocalOperator,
     DbtRunLocalOperator,
@@ -97,6 +98,7 @@
     "DbtRunLocalOperator",
     "DbtSeedLocalOperator",
     "DbtTestLocalOperator",
+    "DbtBuildLocalOperator",
     "DbtDepsLocalOperator",
     "DbtSnapshotLocalOperator",
     "DbtDag",
@@ -106,12 +108,14 @@
     "DbtRunDockerOperator",
     "DbtSeedDockerOperator",
     "DbtTestDockerOperator",
+    "DbtBuildDockerOperator",
     "DbtSnapshotDockerOperator",
     "DbtLSKubernetesOperator",
     "DbtRunOperationKubernetesOperator",
     "DbtRunKubernetesOperator",
     "DbtSeedKubernetesOperator",
     "DbtTestKubernetesOperator",
+    "DbtBuildKubernetesOperator",
     "DbtSnapshotKubernetesOperator",
     "ExecutionMode",
     "LoadMode",

cosmos/operators/__init__.py

@@ -1,3 +1,4 @@
+from .local import DbtBuildLocalOperator as DbtBuildOperator
 from .local import DbtDepsLocalOperator as DbtDepsOperator
 from .local import DbtDocsAzureStorageLocalOperator as DbtDocsAzureStorageOperator
 from .local import DbtDocsLocalOperator as DbtDocsOperator
@@ -16,6 +17,7 @@
     "DbtSnapshotOperator",
     "DbtRunOperator",
     "DbtTestOperator",
+    "DbtBuildOperator",
     "DbtRunOperationOperator",
     "DbtDepsOperator",
     "DbtDocsOperator",

cosmos/operators/base.py

@@ -239,6 +239,13 @@ def build_cmd(
         return dbt_cmd, env
 
 
+class DbtBuildMixin:
+    """Mixin for dbt build command."""
+
+    base_cmd = ["build"]
+    ui_color = "#8194E0"
+
+
 class DbtLSMixin:
     """
     Executes a dbt core ls command.

cosmos/operators/docker.py

@@ -7,6 +7,7 @@
 from cosmos.log import get_logger
 from cosmos.operators.base import (
     AbstractDbtBaseOperator,
+    DbtBuildMixin,
     DbtRunMixin,
     DbtSeedMixin,
     DbtSnapshotMixin,
@@ -66,6 +67,12 @@ def execute(self, context: Context) -> None:
         self.build_and_run_cmd(context=context)
 
 
+class DbtBuildDockerOperator(DbtBuildMixin, DbtDockerBaseOperator):
+    """
+    Executes a dbt core build command.
+    """
+
+
 class DbtLSDockerOperator(DbtLSMixin, DbtDockerBaseOperator):
     """
     Executes a dbt core ls command.

cosmos/operators/kubernetes.py

@@ -9,6 +9,7 @@
 from cosmos.config import ProfileConfig
 from cosmos.operators.base import (
     AbstractDbtBaseOperator,
+    DbtBuildMixin,
     DbtRunMixin,
     DbtSeedMixin,
     DbtSnapshotMixin,
@@ -100,6 +101,12 @@ def execute(self, context: Context) -> None:
         self.build_and_run_cmd(context=context)
 
 
+class DbtBuildKubernetesOperator(DbtBuildMixin, DbtKubernetesBaseOperator):
+    """
+    Executes a dbt core build command.
+    """
+
+
 class DbtLSKubernetesOperator(DbtLSMixin, DbtKubernetesBaseOperator):
     """
     Executes a dbt core ls command.

cosmos/operators/local.py

@@ -39,6 +39,7 @@
 from cosmos.log import get_logger
 from cosmos.operators.base import (
     AbstractDbtBaseOperator,
+    DbtBuildMixin,
     DbtRunMixin,
     DbtSeedMixin,
     DbtSnapshotMixin,
@@ -384,6 +385,12 @@ def on_kill(self) -> None:
             self.subprocess_hook.send_sigterm()
 
 
+class DbtBuildLocalOperator(DbtBuildMixin, DbtLocalBaseOperator):
+    """
+    Executes a dbt core build command.
+    """
+
+
 class DbtLSLocalOperator(DbtLSMixin, DbtLocalBaseOperator):
     """
     Executes a dbt core ls command.

cosmos/operators/virtualenv.py

@@ -10,6 +10,7 @@
 
 from cosmos.log import get_logger
 from cosmos.operators.local import (
+    DbtBuildLocalOperator,
     DbtDocsLocalOperator,
     DbtLocalBaseOperator,
     DbtLSLocalOperator,
@@ -98,6 +99,13 @@ def execute(self, context: Context) -> None:
         logger.info(output)
 
 
+class DbtBuildVirtualenvOperator(DbtVirtualenvBaseOperator, DbtBuildLocalOperator):
+    """
+    Executes a dbt core build command within a Python Virtual Environment, that is created before running the dbt command
+    and deleted just after.
+    """
+
+
 class DbtLSVirtualenvOperator(DbtVirtualenvBaseOperator, DbtLSLocalOperator):
     """
     Executes a dbt core ls command within a Python Virtual Environment, that is created before running the dbt command

tests/operators/test_base.py

@@ -2,6 +2,7 @@
 
 from cosmos.operators.base import (
     AbstractDbtBaseOperator,
+    DbtBuildMixin,
     DbtLSMixin,
     DbtSeedMixin,
     DbtRunOperationMixin,
@@ -26,6 +27,7 @@ def test_dbt_base_operator_is_abstract():
         ("ls", DbtLSMixin),
         ("seed", DbtSeedMixin),
         ("run", DbtRunMixin),
+        ("build", DbtBuildMixin),
     ],
 )
 def test_dbt_mixin_base_cmd(dbt_command, dbt_operator_class):

tests/operators/test_docker.py

@@ -5,6 +5,7 @@
 from pendulum import datetime
 
 from cosmos.operators.docker import (
+    DbtBuildDockerOperator,
     DbtDockerBaseOperator,
     DbtLSDockerOperator,
     DbtRunDockerOperator,
@@ -84,6 +85,7 @@ def test_dbt_docker_operator_get_env(p_context_to_airflow_vars: MagicMock) -> No
     "ls": DbtLSDockerOperator(**base_kwargs),
     "run": DbtRunDockerOperator(**base_kwargs),
     "test": DbtTestDockerOperator(**base_kwargs),
+    "build": DbtBuildDockerOperator(**base_kwargs),
     "seed": DbtSeedDockerOperator(**base_kwargs),
 }
 

tests/operators/test_kubernetes.py

@@ -5,6 +5,7 @@
 from pendulum import datetime
 
 from cosmos.operators.kubernetes import (
+    DbtBuildKubernetesOperator,
     DbtKubernetesBaseOperator,
     DbtLSKubernetesOperator,
     DbtRunKubernetesOperator,
@@ -94,6 +95,7 @@ def test_dbt_kubernetes_operator_get_env(p_context_to_airflow_vars: MagicMock) -
     "ls": DbtLSKubernetesOperator(**base_kwargs),
     "run": DbtRunKubernetesOperator(**base_kwargs),
     "test": DbtTestKubernetesOperator(**base_kwargs),
+    "build": DbtBuildKubernetesOperator(**base_kwargs),
     "seed": DbtSeedKubernetesOperator(**base_kwargs),
 }
 

tests/operators/test_local.py

@@ -22,6 +22,7 @@
     DbtSnapshotLocalOperator,
     DbtRunLocalOperator,
     DbtTestLocalOperator,
+    DbtBuildLocalOperator,
     DbtDocsLocalOperator,
     DbtDocsS3LocalOperator,
     DbtDocsAzureStorageLocalOperator,
@@ -380,6 +381,7 @@ def test_operator_execute_with_flags(mock_build_and_run_cmd, operator_class, kwa
         DbtLSLocalOperator,
         DbtSnapshotLocalOperator,
         DbtTestLocalOperator,
+        DbtBuildLocalOperator,
         DbtDocsLocalOperator,
         DbtDocsS3LocalOperator,
         DbtDocsAzureStorageLocalOperator,