fix(ceda_global): Use constants for folders

openclimatefix · Sep 10, 2024 · 6cd0670 · 6cd0670
1 parent 2f4b9ff
commit 6cd0670
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/local_archives/nwp/ceda/ceda_global.py b/local_archives/nwp/ceda/ceda_global.py
@@ -4,12 +4,16 @@
 import datetime as dt
 
 from dagster_docker import PipesDockerClient
+from constants import LOCATIONS_BY_ENVIRONMENT
+
+env = os.getenv("ENVIRONMENT", "local")
+ZARR_FOLDER = LOCATIONS_BY_ENVIRONMENT[env].NWP_ZARR_FOLDER
 
 @dg.asset(
         name="zarr_archive",
         key_prefix=["nwp", "ceda", "global"],
         metadata={
-            "archive_folder": dg.MetadataValue.text("/mnt/storage_b/nwp/ceda/global"),
+            "archive_folder": dg.MetadataValue.text(f"{ZARR_FOLDER}/nwp/ceda/global"),
             "area": dg.MetadataValue.text("global"),
             "source": dg.MetadataValue.text("ceda"),
         },
@@ -42,7 +46,7 @@ def ceda_global(
             "CEDA_FTP_PASSWORD": os.environ["CEDA_FTP_PASSWORD"],
         },
         container_kwargs={
-            "volumes": ["/mnt/storage_b/nwp/ceda/global:/work"],
+            "volumes": [f"{ZARR_FOLDER}/nwp/ceda/global:/work"],
         },
         context=context,
     ).get_results()