Source code for huracanpy._data._save

"""huracanpy module for saving tracks data"""

from . import _netcdf




def save(dataset, filename):
    """
    Save dataset as filename.
    The file type (NetCDF or csv supported) is detected based on filename extension.

    Parameters
    ----------
    dataset : xarray.Dataset
    filename : str
        Must end in ".nc" or ".csv"

    """
    if filename.split(".")[-1] == "nc":
        _netcdf.save(dataset, filename)
    elif filename.split(".")[-1] == "csv":
        dataset.to_dataframe().to_csv(filename, index=False)
    else:
        raise NotImplementedError(
            "File format not recognized. Please use one of {.nc, .csv}"
        )