Extend dbt-coves generate docs with PROFILES_DIR (#419)

* Extend dbt-coves generate docs with PROFILES_DIR * Fix generate airflow-dags
datacoves · Nov 24, 2023 · 8164a7f · 8164a7f
1 parent ee93b33
commit 8164a7f
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 2 deletions.
diff --git a/dbt_coves/tasks/generate/airflow_dags.py b/dbt_coves/tasks/generate/airflow_dags.py
@@ -303,7 +303,7 @@ def generate_task_group(self, tg_name: str, tg_conf: Dict[str, Any]):
                 elif isinstance(task_call, dict):
                     trigger = task_call.pop("trigger")
                     sensor = task_call.pop("sensor")
-                    task_group_output.append(f"{' ' *8}{trigger['name']}\n")
+                    task_group_output.append(f"{' ' *8}{trigger['call']}\n")
                     task_group_output.append(f"{' ' *8}{sensor['call']}\n")
                     task_group_output.append(f"{' ' *8}{trigger['name']} >> {sensor['name']}\n")
 

diff --git a/dbt_coves/tasks/generate/docs.py b/dbt_coves/tasks/generate/docs.py
@@ -52,7 +52,10 @@ def get_config_value(self, key):
         return self.coves_config.integrated["generate"]["docs"][key]
 
     def _generate_dbt_docs(self):
-        output = run_and_capture_cwd(["dbt", "docs", "generate"], self.config.project_root)
+        command = ["dbt", "docs", "generate"]
+        if self.config.args.PROFILES_DIR:
+            command.extend(["--profiles-dir", self.config.args.PROFILES_DIR])
+        output = run_and_capture_cwd(command, self.config.project_root)
 
         if output.returncode == 0:
             deps_status = "[green]SUCCESS :heavy_check_mark:[/green]"