Shut off hardcoded huc: Note flow not workign yet

NOAA-OWP · Sep 26, 2024 · 43d3894 · 43d3894
1 parent d554508
commit 43d3894
Show file tree

Hide file tree

Showing 2 changed files with 33 additions and 1 deletion.
diff --git a/tools/generate_categorical_fim.py b/tools/generate_categorical_fim.py
@@ -172,7 +172,9 @@ def process_generate_categorical_fim(
     #     if os.path.isdir(os.path.join(fim_run_dir, x)) and x[0] in ['0', '1', '2'] and x[:2] != "19"
     # ]
 
-    valid_ahps_hucs = ["04130003"]
+
+    # valid_ahps_hucs = ["04130003"]
+
 
 
     # # Code variation for KEEPING Alaska HUCS:
@@ -509,6 +511,7 @@ def iterate_through_huc_stage_based(
                     "lid_alt_m": pd.Series(dtype='float'),
                     "mapped": pd.Series(dtype='str'),
                     "status": pd.Series(dtype='str'),
+                    "missing_stages_msg": pd.Series(dtype='str'),
                     }
 
 

diff --git a/tools/generate_categorical_fim_flows.py b/tools/generate_categorical_fim_flows.py
@@ -94,6 +94,35 @@ def generate_flows_for_huc(
             MP_LOG.lprint(f"huc {huc} has no applicable nws_lids")
             return
 
+        df_cols = {"nws_lid": pd.Series(dtype='str'),
+                "name": pd.Series(dtype='str'),
+                "WFO": pd.Series(dtype='str'),
+                "rfc": pd.Series(dtype='str'),
+                "huc": pd.Series(dtype='str'),
+                "state": pd.Series(dtype='str'),
+                "county": pd.Series(dtype='str'),
+                "magnitude": pd.Series(dtype='str'),
+                "q": pd.Series(dtype='str'),
+                "q_uni": pd.Series(dtype='str'),
+                "q_src": pd.Series(dtype='str'),
+                "stage": pd.Series(dtype='float'),
+                "stage_uni": pd.Series(dtype='str'),
+                "s_src": pd.Series(dtype='str'),
+                "wrds_time": pd.Series(dtype='str'),
+                "nrldb_time": pd.Series(dtype='str'),
+                "nwis_time": pd.Series(dtype='str'),
+                "lat": pd.Series(dtype='float'),
+                "lon": pd.Series(dtype='float'),
+                "dtm_adj_ft": pd.Series(dtype='str'),
+                "dadj_w_ft": pd.Series(dtype='float'),
+                "dadj_w_m": pd.Series(dtype='float'),
+                "lid_alt_ft": pd.Series(dtype='float'), 
+                "lid_alt_m": pd.Series(dtype='float'),
+                "mapped": pd.Series(dtype='str'),
+                "status": pd.Series(dtype='str'),
+            }
+
+
         # Loop through each lid in list to create flow file
         for lid in nws_lids:
             # Convert lid to lower case