lodel2_mirror/lodel/editorial_model/translator/picklefile.py

#-*- coding: utf-8 -*-

import pickle

## @brief Save a model in a file
# @param model EditorialModel : the model to save
# @param filename str|None : if None return the model as pickle bytes
# @return None if filename is a string, else returns bytes representation of model
def save(self, model, filename = None):
    with open(filename, 'w+') as ffd:
        pickle.dump(model, ffd)
    return filename

## @brief Load a model from a file
# @param filename str : the filename to use to load the model
def load(self, filename):
    with open(filename, 'r') as ffd:
        edmod = pickle.load(ffd)
    return edmod