From 9d4d747d35a513b0925c9e282330bd11e7d68030 Mon Sep 17 00:00:00 2001 From: Valentin Khomutenko Date: Mon, 24 Jul 2023 18:33:53 +0200 Subject: [PATCH 1/3] add TRIGGERED_BY env variable support --- data_diff/__main__.py | 2 +- data_diff/dbt.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/data_diff/__main__.py b/data_diff/__main__.py index dfd12577..68896a9f 100644 --- 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 index fd3223de..9bb5c2ea 100644 --- 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() From 4a848eb82c8c474feb07b4c50663734bccd9bb4b Mon Sep 17 00:00:00 2001 From: Valentin Khomutenko Date: Wed, 26 Jul 2023 13:29:48 +0200 Subject: [PATCH 2/3] Update __main__.py --- data_diff/__main__.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/data_diff/__main__.py b/data_diff/__main__.py index 68896a9f..82334932 100644 --- a/data_diff/__main__.py +++ b/data_diff/__main__.py @@ -33,7 +33,7 @@ "-": "red", } -set_entrypoint_name(os.getenv("TRIGGERED_BY", "CLI")) +set_entrypoint_name(os.getenv("DATAFOLD_TRIGGERED_BY", "CLI")) def _get_log_handlers(is_dbt: Optional[bool] = False) -> Dict[str, logging.Handler]: From 7686db8c08b3980a5ef6f151bc72cbf4ead37d8c Mon Sep 17 00:00:00 2001 From: Valentin Khomutenko Date: Wed, 26 Jul 2023 13:30:09 +0200 Subject: [PATCH 3/3] Update dbt.py --- data_diff/dbt.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/data_diff/dbt.py b/data_diff/dbt.py index 9bb5c2ea..fdebb532 100644 --- 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(os.getenv("TRIGGERED_BY", "CLI-dbt")) + set_entrypoint_name(os.getenv("DATAFOLD_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()