Source code for lcc.utils.output_process_modules
import os
import pickle
from lcc.entities.exceptions import InvalidFilesPath, InvalidFile
[docs]def saveIntoFile(obj, path=".", file_name="saved_object.pickle",
folder_name=None):
'''
This method serialize object (save it into file)
obj : object
Object to serialize
path : str
Path to the folder
file_name : str
Name of result file
folder_name : str
Name of folder
Returns
-------
None
'''
path_with_name = "%s/%s" % (path, file_name)
if folder_name:
os.makedirs(path_with_name + folder_name)
path_with_name = "%s/%s/%s" % (path, folder_name, file_name)
try:
with open(path_with_name, 'wb') as output:
pickle.dump(obj, output, pickle.HIGHEST_PROTOCOL)
print "Object has been saved into %s/%s" % (path, file_name)
except IOError:
raise InvalidFilesPath("Path: %s\tfile name: %s" % (path, file_name))
[docs]def loadFromFile(file_name="saved_object.pickle"):
'''
Open object from file
Parameters
----------
file_name : str
Name of the serialized file
Returns
-------
object
Loaded object
'''
try:
with open(file_name, 'rb') as inputToLoad:
loaded_object = pickle.load(inputToLoad)
return loaded_object
except IOError:
raise InvalidFilesPath
except ImportError:
raise InvalidFile(
"Structure of project has been changed since saving this object")
except TypeError:
return pickle.load(file_name)