diff --git a/config/access-om2.yaml b/config/access-om2.yaml index d6971f5..8c60cc0 100644 --- a/config/access-om2.yaml +++ b/config/access-om2.yaml @@ -261,6 +261,7 @@ sources: - metadata_yaml: /g/data/cj50/access-om2/raw-output/access-om2-01/01deg_jra55v140_iaf_cycle3/metadata.yaml path: - /g/data/cj50/access-om2/raw-output/access-om2-01/01deg_jra55v140_iaf_cycle3 + - /g/data/ik11/outputs/access-om2-01/01deg_jra55v140_iaf_cycle3 - metadata_yaml: /g/data/cj50/access-om2/raw-output/access-om2-01/01deg_jra55v140_iaf_cycle4/metadata.yaml path: diff --git a/src/access_nri_intake/source/utils.py b/src/access_nri_intake/source/utils.py index 37c53bb..df3f636 100644 --- a/src/access_nri_intake/source/utils.py +++ b/src/access_nri_intake/source/utils.py @@ -169,18 +169,19 @@ def parse_access_filename(filename): """ # ACCESS output file patterns - patterns = { - r"^iceh.*\.(\d{4}-\d{2}-\d{2})$", - r"^iceh.*\.(\d{4}-\d{2})$", - r"^iceh.*\.(\d{4}-\d{2})-.[^\d].*", - r"^iceh.*\.(\d{3})-.[^\d].*", - r"^ocean.*[^\d]_(\d{4}_\d{2}_\d{2})$", - r"^ocean.*[^\d]_(\d{4}_\d{2})$", - r"^ocean.*[^\d]_(\d{4})$", - r"^ocean.*[^\d]_(\d{2})$", - r"^.*\.p.(\d{6})_.*", - r"^.*\.p.-(\d{6})_.*", - } + not_multi_digit = "(?:\\d(?!\\d)|[^\\d](?=\\d)|[^\\d](?!\\d))" + ymd = "\\d{4}[_,-]\\d{2}[_,-]\\d{2}" + ym = "\\d{4}[_,-]\\d{2}" + y = "\\d{4}" + patterns = [ + rf"^iceh.*\.({ymd}|{ym})$", # ACCESS-ESM1.5/OM2 ice + rf"^iceh.*\.({ym})-{not_multi_digit}.*", # ACCESS-CM2 ice + rf"^iceh.*\.(\d{{3}})-{not_multi_digit}.*", # ACCESS-OM2 ice + rf"^ocean.*[_,-](?:ymd|ym|y)_({ymd}|{ym}|{y})(?:$|[_,-]{not_multi_digit}.*)", # ACCESS-OM2 ocean + r"^ocean.*[^\d]_(\d{2})$", # A few wierd files in ACCESS-OM2 01deg_jra55v13_ryf9091 + r"^.*\.p.(\d{6})_.*", # ACCESS-CM2 atmosphere + r"^.*\.p.-(\d{6})_.*", # ACCESS-ESM1.5 atmosphere + ] # Frequency translations frequencies = { "daily": (1, "day"), diff --git a/tests/test_source_utils.py b/tests/test_source_utils.py index 06e68da..b1afcce 100644 --- a/tests/test_source_utils.py +++ b/tests/test_source_utils.py @@ -22,6 +22,7 @@ ("bz687a.p7107912_dai", ("bz687a_p7XXXXXX_dai", "107912", (1, "day"))), ("iceh_m.2014-06", ("iceh_m_XXXX_XX", "2014-06", None)), ("iceh.1917-05-daily", ("iceh_XXXX_XX_daily", "1917-05", (1, "day"))), + ("iceh_03h.2016-01-3hourly", ("iceh_03h_XXXX_XX_3hourly", "2016-01", None)), ("ocean_bgc_ann", ("ocean_bgc_ann", None, (1, "yr"))), ("ocean_daily", ("ocean_daily", None, (1, "day"))), # Example ACCESS-ESM1.5 filenames @@ -40,19 +41,20 @@ ("iceh.1917-05", ("iceh_XXXX_XX", "1917-05", None)), # Example ACCESS-OM2 filenames ("iceh.057-daily", ("iceh_XXX_daily", "057", (1, "day"))), + ("iceh.1985-08-31", ("iceh_XXXX_XX_XX", "1985-08-31", None)), ("ocean", ("ocean", None, None)), ("ocean_month", ("ocean_month", None, (1, "mon"))), ( - "ocean_daily_3d_vhrho_nt_07", - ("ocean_daily_3d_vhrho_nt_XX", "07", (1, "day")), + "ocean-2d-area_t", + ("ocean_2d_area_t", None, None), ), ( - "oceanbgc-3d-caco3-1-yearly-mean-y_2015", - ("oceanbgc_3d_caco3_1_yearly_mean_y_XXXX", "2015", (1, "yr")), + "ocean_daily_3d_pot_rho_1", + ("ocean_daily_3d_pot_rho_1", None, (1, "day")), ), ( - "oceanbgc-2d-wdet100-1-daily-mean-y_2015", - ("oceanbgc_2d_wdet100_1_daily_mean_y_XXXX", "2015", (1, "day")), + "ocean_daily_3d_vhrho_nt_07", + ("ocean_daily_3d_vhrho_nt_XX", "07", (1, "day")), ), ( "ocean-3d-v-1-monthly-pow02-ym_1958_04", @@ -66,6 +68,30 @@ (1, "mon"), ), ), + ( + "ocean-2d-sea_level-540-seconds-snap-ym_2022_04_01", + ( + "ocean_2d_sea_level_540_seconds_snap_ym_XXXX_XX_XX", + "2022_04_01", + None, + ), + ), + ( + "ocean-3d-salt-1-daily-mean-ym_2018_10_jmax511_sigfig4", + ( + "ocean_3d_salt_1_daily_mean_ym_XXXX_XX_jmax511_sigfig4", + "2018_10", + (1, "day"), + ), + ), + ( + "oceanbgc-3d-caco3-1-yearly-mean-y_2015", + ("oceanbgc_3d_caco3_1_yearly_mean_y_XXXX", "2015", (1, "yr")), + ), + ( + "oceanbgc-2d-wdet100-1-daily-mean-y_2015", + ("oceanbgc_2d_wdet100_1_daily_mean_y_XXXX", "2015", (1, "day")), + ), ( "oceanbgc-3d-phy-1-daily-mean-3-sigfig-5-daily-ymd_2020_12_01", ( @@ -74,7 +100,10 @@ (1, "day"), ), ), - ("iceh.1985-08-31", ("iceh_XXXX_XX_XX", "1985-08-31", None)), + ( + "rregionPrydz_temp_xflux_adv", + ("rregionPrydz_temp_xflux_adv", None, None), + ), ], ) def test_parse_access_filename(filename, expected):