Merge pull request #976 from MTES-MCT/feat-simplify-dbt-command

Simplification du dag pour lancer des commandes DBT arbitraires

Author: alexisig

airflow/dags/dbt_cmd.py

@@ -1,9 +1,3 @@
-"""
-
-Ce dag permet d'exécuter des commandes dbt.
-
-"""
-
 from datetime import datetime, timedelta
 
 from airflow.models.param import Param
@@ -22,32 +16,22 @@
 }
 
 with DAG(
-    "dbt_selector_dag",
+    "dbt_build",
     default_args=default_args,
-    description="Exécute des commandes DBT avec un sélecteur",
+    description="Exécute des commandes DBT",
     schedule_interval=None,
     start_date=datetime(2024, 1, 1),
     catchup=False,
     params={
-        "select": Param(default="default", type="string", description="Sélecteur DBT à utiliser pour l'exécution"),
         "command": Param(
-            default="build",
+            default="dbt build -s",
             type="string",
-            description="Commande DBT à exécuter (build, run, test, etc.)",
-            enum=["build", "run", "test", "compile", "debug", "deps", "clean"],
         ),
     },
 ) as dag:
-    # Validation du sélecteur
-    validate_selector = BashOperator(
-        task_id="validate_selector",
-        bash_command=get_dbt_command_from_directory("dbt ls --select {{ params.select }} || exit 1"),
-    )
-
-    # Exécution de dbt avec le sélecteur
     execute_dbt = BashOperator(
         task_id="execute_dbt",
-        bash_command=get_dbt_command_from_directory("dbt {{ params.command }} --select {{ params.select }}"),
+        bash_command=get_dbt_command_from_directory("{{ params.command }}"),
     )
 
-    validate_selector >> execute_dbt
+    execute_dbt