A function to cache the result of a function call. For long calculations/data loading.

functionCache.py

import os
from numpy import savez_compressed, load


def functionCache(fn, fct, *args, **kwargs):
    """
    force_recalc = True -> self-descriptive
    """
    force_recalc = kwargs.pop('force_recalc', False)
    if os.path.isfile(fn) and not force_recalc:
        return load(fn)['arr_0']
    else:
        tmp = fct(*args, **kwargs)
        savez_compressed(fn, tmp)
        return tmp