Limit zero padding for analysis_assim

RohanSunkarapalli · Feb 20, 2024 · bf2e22b · bf2e22b
1 parent 1542eca
commit bf2e22b
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 2 deletions.
diff --git a/nwmurl/.ipynb_checkpoints/urlgennwm-checkpoint.py b/nwmurl/.ipynb_checkpoints/urlgennwm-checkpoint.py
@@ -61,10 +61,17 @@ def makename(
 
     datetxt = f"nwm.{date.strftime('%Y%m%d')}"
     foldertxt = f"{run_type}{run_typesuffix}"
-    filetxt = f"nwm.t{fcst_cycle:02d}z.{run_name}{runsuffix}.{var_name}{varsuffix}.{fhprefix}{fcst_hour:03d}.{geography}.nc"
+
+    # Adjust zero padding for analysis_assim outputs
+    if "analysis_assim" in run_type:
+        filetxt = f"nwm.t{fcst_cycle:02d}z.{run_name}{runsuffix}.{var_name}{varsuffix}.{fhprefix}{fcst_hour:02d}.{geography}.nc"
+    else:
+        filetxt = f"nwm.t{fcst_cycle:02d}z.{run_name}{runsuffix}.{var_name}{varsuffix}.{fhprefix}{fcst_hour:03d}.{geography}.nc"
+
     return f"{urlbase_prefix}{datetxt}/{foldertxt}/{filetxt}"
 
 
+
 # setting run_type
 def run_type(runinput, varinput, geoinput, default=""):
     """This function takes the numeric command line input and converts to the text used in the url."""

diff --git a/nwmurl/urlgennwm.py b/nwmurl/urlgennwm.py
@@ -61,10 +61,17 @@ def makename(
 
     datetxt = f"nwm.{date.strftime('%Y%m%d')}"
     foldertxt = f"{run_type}{run_typesuffix}"
-    filetxt = f"nwm.t{fcst_cycle:02d}z.{run_name}{runsuffix}.{var_name}{varsuffix}.{fhprefix}{fcst_hour:03d}.{geography}.nc"
+
+    # Adjust zero padding for analysis_assim outputs
+    if "analysis_assim" in run_type:
+        filetxt = f"nwm.t{fcst_cycle:02d}z.{run_name}{runsuffix}.{var_name}{varsuffix}.{fhprefix}{fcst_hour:02d}.{geography}.nc"
+    else:
+        filetxt = f"nwm.t{fcst_cycle:02d}z.{run_name}{runsuffix}.{var_name}{varsuffix}.{fhprefix}{fcst_hour:03d}.{geography}.nc"
+
     return f"{urlbase_prefix}{datetxt}/{foldertxt}/{filetxt}"
 
 
+
 # setting run_type
 def run_type(runinput, varinput, geoinput, default=""):
     """This function takes the numeric command line input and converts to the text used in the url."""