Calendar conversion using xclim/xarray

xclim_convert_calendars.py

from pathlib import Path
from typing import List

import xarray as xr
from xclim.core import calendar as xcal

root_dir = Path(Path().cwd().root)
source_dir = root_dir.joinpath("my/source/files/folder/")
processed_files = [x for x in source_dir.glob("*.nc")]

def adjust_calendars(files: List[Path], sources: Path):
    for file in files:
        ds = xr.open_dataset(file)

        try:
            calendar = ds.time.calendar
        except AttributeError:
            calendar = xcal.get_calendar(ds, dim="time")

        if calendar != "default":
            dsOut = xcal.convert_calendar(ds, "default", dim="time")  # noqa
        else:
            ds.close()
            continue

        dsOut.to_netcdf(sources.joinpath(f"{file.stem}_calendar-fixed.nc"))
        dsOut.close()
        ds.close()


if __name__ == "__main__":
    adjust_calendars(processed_files, source_dir)