def __init__(self, *args, **kw):
self.__f = None
if len(args) == 1 and isinstance(args[0], int):
self.__f = StringIO()
PythonPickler.__init__(self, self.__f, args[0], **kw)
else:
PythonPickler.__init__(self, *args, **kw)
def dumps(obj, protocol=None):
if protocol > HIGHEST_PROTOCOL:
# use cPickle error message, not pickle.py one
raise ValueError("pickle protocol %d asked for; "
"the highest available protocol is %d" % (
protocol, HIGHEST_PROTOCOL))
file = StringIO()
Pickler(file, protocol).dump(obj)
return file.getvalue()
def loads(str):
f = StringIO(str)
return Unpickler(f).load()
def dumps(obj, protocol=None):
file = StringIO()
Pickler(file, protocol).dump(obj)
return file.getvalue()
