Skip to content

Commit d1bb52c

Browse files
committed
issue #5: Start populating empty huc2 from substring of huc
1 parent b61bbf9 commit d1bb52c

File tree

1 file changed

+12
-0
lines changed

1 file changed

+12
-0
lines changed

get_AWDB_stations_3.py

Lines changed: 12 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -787,6 +787,7 @@ def update_forecast_point_ws():
787787
Append_management([os.path.join(settings.AWDB_FGDB_PATH, BOR_Active_Temp)], tmpForecastFc)
788788
Delete_management(os.path.join(settings.AWDB_FGDB_PATH, BOR_Active_Temp))
789789
HUC2 = "huc2"
790+
HUC = "huc"
790791
WINTER_START_MONTH = "winter_start_month"
791792
WINTER_END_MONTH = "winter_end_month"
792793
BAGIS_NOTE = "bagis_note"
@@ -813,6 +814,17 @@ def updateText(txtValue):
813814
# Update bagis_note
814815
expression = f"updateText(!{sourceLayer}.{BAGIS_NOTE}!)"
815816
CalculateField_management(joined_table, f"{FCST_Active_Temp}.{BAGIS_NOTE}", expression, "PYTHON3", codeblock)
817+
# Update huc2 from huc if still null
818+
expression = f"updateHuc2(!{sourceLayer}.{HUC2}!,!{sourceLayer}.{HUC}!)"
819+
codeblock = """
820+
def updateHuc2(huc2, huc):
821+
if (huc2 != None):
822+
return huc2
823+
elif (huc != None):
824+
return huc[0:2]
825+
else:
826+
return None"""
827+
CalculateField_management(joined_table, f"{FCST_Active_Temp}.{HUC2}", expression, "PYTHON3", codeblock)
816828
# Update winter_start_month
817829
expression = f"updateMonth(!{sourceLayer}.{WINTER_START_MONTH}!)"
818830
codeblock = """

0 commit comments

Comments
 (0)