change dbt target resolution take 2 (#115)

dagster-io · Aug 12, 2024 · d0abda0 · d0abda0
1 parent 67bf118
commit d0abda0
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 11 deletions.
diff --git 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
@@ -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()
+dbt_project.prepare_if_dev()