From d0abda02b2cdc56012c3e8f520a839addebc4c69 Mon Sep 17 00:00:00 2001 From: Alex Langenfeld Date: Mon, 12 Aug 2024 17:35:12 -0500 Subject: [PATCH] change dbt target resolution take 2 (#115) --- .../hooli_demo_assets/resources/__init__.py | 12 +++++------- hooli_data_eng/project.py | 9 +++++---- 2 files changed, 10 insertions(+), 11 deletions(-) diff --git a/hooli-demo-assets/hooli_demo_assets/resources/__init__.py b/hooli-demo-assets/hooli_demo_assets/resources/__init__.py index fa47459..bd4ab98 100644 --- a/hooli-demo-assets/hooli_demo_assets/resources/__init__.py +++ b/hooli-demo-assets/hooli_demo_assets/resources/__init__.py @@ -9,13 +9,11 @@ def get_env(): - if os.getenv("DAGSTER_CLOUD_DEPLOYMENT_NAME", "") == "data-eng-prod": - return "PROD" - if os.getenv("DAGSTER_IS_DEV_CLI"): - return "LOCAL" - # default to BRANCH so we use that in github CI - return "BRANCH" - + if os.getenv("DAGSTER_CLOUD_IS_BRANCH_DEPLOYMENT", "") == "1": + return "BRANCH" + if os.getenv("DAGSTER_CLOUD_DEPLOYMENT_NAME", "") == "data-eng-prod": + return "PROD" + return "LOCAL" # Path for duckdb connection - needed for local dev current_file_path = Path(__file__) diff --git a/hooli_data_eng/project.py b/hooli_data_eng/project.py index e3732bc..27bcade 100644 --- a/hooli_data_eng/project.py +++ b/hooli_data_eng/project.py @@ -4,11 +4,12 @@ def get_env(): - if os.getenv("DAGSTER_CLOUD_IS_BRANCH_DEPLOYMENT", "") == "1": - return "BRANCH" if os.getenv("DAGSTER_CLOUD_DEPLOYMENT_NAME", "") == "data-eng-prod": return "PROD" - return "LOCAL" + if os.getenv("DAGSTER_IS_DEV_CLI"): + return "LOCAL" + # default to BRANCH so we use that in github CI + return "BRANCH" dbt_project_path = Path(__file__).parent.parent.joinpath("dbt_project") @@ -19,4 +20,4 @@ def get_env(): state_path="target/slim_ci", target=get_env(), ) -dbt_project.prepare_if_dev() \ No newline at end of file +dbt_project.prepare_if_dev()