xTc@ sSdZddlmZdZdZdZedZdZdZ d S( s Utility functions for automatically opening and serializing a python object into a file using the latest pickling protocol. Can optionally compress the output if file_name ends with .gz. Also provides functions to read that object back. i(tdivisioncC sddl}|jj|}ddlm}ddlm}y|jd|d|SWn5tk r}td||t |fnXdS(st Wraps libtbx.smart_open. Parameters ---------- file_name : str mode : str Returns ------- file iN(t show_string(t smart_opent file_nametmodesCannot open pickle file %s (%s)( tostpatht expandusertlibtbx.str_utilsRtlibtbxRtfiletIOErrortstr(RRRRRte((s@/public2/ftp/pub/ccp4/6.5/ccp4-6.5/lib/py2/libtbx/easy_pickle.pyt_open s cC s+ddl}|j|t|d|jS(s Wraps cPickle.dump. Parameters ---------- file_name : str obj : object Examples -------- >>> from libtbx.easy_pickle import dump, load >>> dump("output.pkl.gz", [1, 2, 3]) >>> print load("output.pkl.gz") [1, 2, 3] iNtwb(tcPickletdumpRtHIGHEST_PROTOCOL(RtobjR((s@/public2/ftp/pub/ccp4/6.5/ccp4-6.5/lib/py2/libtbx/easy_pickle.pyRs cC sddl}|j||jS(s_ Wraps cPickle.dumps. Parameters ---------- obj : object Returns ------- str iN(RtdumpsR(RR((s@/public2/ftp/pub/ccp4/6.5/ccp4-6.5/lib/py2/libtbx/easy_pickle.pyR2s cC sDddl}|r.|jt|djS|jt|dS(s Wraps cPickle.load. Parameters ---------- file_name : str faster_but_using_more_memory : bool, optional Optionally read the entirety of a file into memory before converting it into a python object. Returns ------- object iNtrb(RtloadsRtreadtload(Rtfaster_but_using_more_memoryR((s@/public2/ftp/pub/ccp4/6.5/ccp4-6.5/lib/py2/libtbx/easy_pickle.pyRAs cC sddl}|j|S(s Wraps cPickle.loads. Parameters ---------- string : str Returns ------- object Examples -------- >>> from libtbx.easy_pickle import dumps, loads >>> print loads(dumps([1, 2, 3]) [1, 2, 3] iN(RR(tstringR((s@/public2/ftp/pub/ccp4/6.5/ccp4-6.5/lib/py2/libtbx/easy_pickle.pyRUs cO std||fdS(s1 Dumps args and keyword_args to args.pickle. s args.pickleN(R(targst keyword_args((s@/public2/ftp/pub/ccp4/6.5/ccp4-6.5/lib/py2/libtbx/easy_pickle.pyt dump_argsjsN( t__doc__t __future__RRRRtTrueRRR(((s@/public2/ftp/pub/ccp4/6.5/ccp4-6.5/lib/py2/libtbx/easy_pickle.pyts