Source code for

This file is part of CLIMADA.

Copyright (C) 2017 ETH Zurich, CLIMADA contributors listed in AUTHORS.

CLIMADA is free software: you can redistribute it and/or modify it under the
terms of the GNU General Public License as published by the Free
Software Foundation, version 3.

CLIMADA is distributed in the hope that it will be useful, but WITHOUT ANY
WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS FOR A
PARTICULAR PURPOSE.  See the GNU General Public License for more details.

You should have received a copy of the GNU General Public License along
with CLIMADA. If not, see <>.


define save functionalities

__all__ = ['save',

from pathlib import Path
import pickle
import logging

from climada.util.config import CONFIG

LOGGER = logging.getLogger(__name__)

[docs]def save(out_file_name, var): """Save variable with provided file name. Uses configuration save_dir folder if no absolute path provided. Parameters ---------- out_file_name : str file name (absolute path or relative to configured save_dir) var : object variable to save in pickle format """ out_file = Path(out_file_name) if Path(out_file_name).is_absolute() \ else CONFIG.local_data.save_dir.dir().joinpath(out_file_name) target_dir = out_file.parent try: # Generate folder if it doesn't exists if not target_dir.is_dir(): target_dir.mkdir()'Created folder %s.', target_dir) with'wb') as flh: pickle.dump(var, flh, pickle.HIGHEST_PROTOCOL)'Written file %s', out_file) except FileNotFoundError as err: raise FileNotFoundError(f'Folder {target_dir} not found: ' + str(err)) from err except OSError as ose: raise ValueError('Data is probably too big. Try splitting it: ' + str(ose)) from ose
[docs]def load(in_file_name): """Load variable contained in file. Uses configuration save_dir folder if no absolute path provided. Parameters ---------- in_file_name : str file name Returns ------- object """ in_file = Path(in_file_name) if Path(in_file_name).is_absolute() \ else CONFIG.local_data.save_dir.dir().joinpath(in_file_name) with'rb') as flh: data = pickle.load(flh) return data