ypickle: a simple script to achieve matlab type save and load, with the file compressed with zlib

ypickle.py

"""ypickle.py implements an interface similar to pickle, but allows one to
perform a matlab-like store and load with zlib compression
"""
import cPickle as pickle
import zlib

class YPickle(object):
    """The abstract class that holds the data as attributes
    """
    pass


def load(fid):
    if type(fid) is str:
        fid = open(fid, 'rb')
    pickled = pickle.loads(zlib.decompress(fid.read()))
    data = YPickle()
    for key, val in pickled.iteritems():
        setattr(data, key, val)
    return data


def dump(fid, **kwargs):
    if type(fid) is str:
        fid = open(fid, 'wb')
    pickled = pickle.dumps(kwargs)
    fid.write(zlib.compress(pickled))
    fid.close()