add TRIGGERED_BY env variable support

vvkh · vvkh · commit 9d4d747d35a5 · 2023-07-24T18:33:53.000+02:00
diff --git a/data_diff/__main__.py b/data_diff/__main__.py
@@ -33,7 +33,7 @@
     "-": "red",
 }
 
-set_entrypoint_name("CLI")
+set_entrypoint_name(os.getenv("TRIGGERED_BY", "CLI"))
 
 
 def _get_log_handlers(is_dbt: Optional[bool] = False) -> Dict[str, logging.Handler]:
diff --git a/data_diff/dbt.py b/data_diff/dbt.py
@@ -74,7 +74,7 @@ def dbt_diff(
 ) -> None:
     print_version_info()
     diff_threads = []
-    set_entrypoint_name("CLI-dbt")
+    set_entrypoint_name(os.getenv("TRIGGERED_BY", "CLI-dbt"))
     dbt_parser = DbtParser(profiles_dir_override, project_dir_override, state)
     models = dbt_parser.get_models(dbt_selection)
     config = dbt_parser.get_datadiff_config()

Original file line number	Diff line number	Diff line change
`@@ -33,7 +33,7 @@`
`33`	`33`	`"-": "red",`
`34`	`34`	`}`
`35`	`35`
`36`		`-set_entrypoint_name("CLI")`
	`36`	`+set_entrypoint_name(os.getenv("TRIGGERED_BY", "CLI"))`
`37`	`37`
`38`	`38`
`39`	`39`	`def _get_log_handlers(is_dbt: Optional[bool] = False) -> Dict[str, logging.Handler]:`