pickle_get.py

def pickle_load(filename):
    return pickle.load(open(filename, "rb"))

def pickle_dump(data, filename):
    return pickle.dump(data, open(filename, "wb"))

def pickle_get(filename, fn, *args, fresh=False, verbose=False):
    """Try to load data from a pickle file. Compute it with `fn` and save it if it doesn't exist."""
    if not fresh and file_exists(filename):
        if verbose:
            print("using pickled save from %s." % filename)
        return pickle_load(filename)
    else:
        if verbose:
            print("computing data...")
        data = fn(*args)
        pickle_dump(data, filename)
        if verbose:
            print("wrote to %s." % filename)
        return data