pass args

stellar · Oct 2, 2024 · 430a539 · 430a539
1 parent f14ef22
commit 430a539
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 0 deletions.
diff --git a/dags/dbt_source_data_freshness_test_dag.py b/dags/dbt_source_data_freshness_test_dag.py
@@ -33,6 +33,7 @@
         dag,
         command_type="source",
         sub_command="freshness",
+        cmd_args=["source:crypto_stellar.*", "source:crypto_stellar_internal_2.*"],
         flag=None,
         tag=None,
         resource_cfg="dbt",

diff --git a/dags/stellar_etl_airflow/build_dbt_task.py b/dags/stellar_etl_airflow/build_dbt_task.py
@@ -65,6 +65,7 @@ def dbt_task(
     tag=None,
     flag="select",
     operator="",
+    cmd_args=[],
     command_type="build",
     excluded=None,
     resource_cfg="default",
@@ -95,6 +96,9 @@ def dbt_task(
     if flag:
         args.append(f"--{flag}")
 
+    if len(cmd_args):
+        args = [*args, *cmd_args]
+
     models = []
     if tag:
         task_name = tag