ó œ]c@s¿ddlmZddlZddlZddlZddlZd„Zedd„ƒZd„Zd„Z d„Z ed „ƒZ ed „ƒZ e d „Zd „Zd „Zd„ZdS(iÿÿÿÿ(tcontextmanagerNcCs(y|ƒtSWn|k r#tSXdS(N(tFalsetTrue(texctlamda((s*lib/python2.7/site-packages/partd/utils.pytraisess  tccs—d|jdƒ}tj|ƒ\}}tj|ƒtj|ƒz |VWdtjj|ƒr’tjj|ƒr‚t j |ƒq’tj|ƒnXdS(Nt.( tlstripttempfiletmkstemptostclosetremovetpathtexiststisdirtshutiltrmtree(t extensionthandletfilename((s*lib/python2.7/site-packages/partd/utils.pyttmpfiles   cCstjdt|ƒƒ|S(sÔ Pack the length of the bytes in front of the bytes TODO: This does a full copy. This should maybe be inlined somehow wherever this gets used instead. My laptop shows a data bandwidth of 2GB/s tQ(tstructtpacktlen(tbytes((s*lib/python2.7/site-packages/partd/utils.pytframe!sccswd}t|ƒ}tƒ}xU||krrtjd|||d!ƒd}|d7}||||!V||7}qWdS(s± Split buffer into frames of concatenated chunks >>> data = frame(b'Hello') + frame(b'World') >>> list(framesplit(data)) # doctest: +SKIP [b'Hello', b'World'] iRiN(RtlistRtunpack(Rtitntchunkstnbytes((s*lib/python2.7/site-packages/partd/utils.pyt framesplit+s  ! ccsQt|ƒ|kr|Vn3x0tdt|ƒ|ƒD]}||||!Vq3WdS(sâ Partition bytes into evenly sized blocks The final block holds the remainder and so may not be of equal size >>> list(partition_all(2, b'Hello')) ['He', 'll', 'o'] See Also: toolz.partition_all iN(Rtrange(R RR((s*lib/python2.7/site-packages/partd/utils.pyt partition_all;s cgs!y dVWn|k rnXdS(N((R((s*lib/python2.7/site-packages/partd/utils.pytignoringMs  cos dVdS(N((targstkwargs((s*lib/python2.7/site-packages/partd/utils.pyt do_nothingUscsat|tƒrUˆr,‡‡fd†|DƒSg|D]}t|ˆdˆƒ^q3Snˆ|SdS(sá Get nested index from collection Examples -------- >>> nested_get(1, 'abc') 'b' >>> nested_get([1, 0], 'abc') ['b', 'a'] >>> nested_get([[1, 0], [0, 1]], 'abc') [['b', 'a'], ['a', 'b']] c3s$|]}t|ˆdˆƒVqdS(tlazyN(t nested_get(t.0R(tcollR*(s*lib/python2.7/site-packages/partd/utils.pys isR*N(t isinstanceRR+(tindR-R*R((R-R*s*lib/python2.7/site-packages/partd/utils.pyR+Zs )ccsHxA|D]9}t|tƒr;x!t|ƒD] }|Vq)Wq|VqWdS(sJ >>> list(flatten([1])) [1] >>> list(flatten([[1, 2], [1, 2]])) [1, 2, 1, 2] >>> list(flatten([[[1], [2]], [[1], [2]]])) [1, 2, 1, 2] >>> list(flatten(((1, 2), (1, 2)))) # Don't flatten tuples [(1, 2), (1, 2)] >>> list(flatten((1, 2, [3, 4]))) # support heterogeneous [1, 2, 3, 4] N(R.Rtflatten(tseqtitemtitem2((s*lib/python2.7/site-packages/partd/utils.pyR0ps   cCsYt|tƒr||St|tƒrB|d t|d|ƒfStt|ƒ|ƒSdS(s¿ suffix a key with a suffix Works if they key is a string or a tuple >>> suffix('x', '.dtype') 'x.dtype' >>> suffix(('a', 'b', 'c'), '.dtype') ('a', 'b', 'c.dtype') iÿÿÿÿN(R.tstrttupletsuffix(tkeytterm((s*lib/python2.7/site-packages/partd/utils.pyR6Šs cCs_t|tƒrn*t|tƒr-|f}nt|ƒf}t|tƒsW|f}n||S(sÝ extend a key with a another element in a tuple Works if they key is a string or a tuple >>> extend('x', '.dtype') ('x', '.dtype') >>> extend(('a', 'b', 'c'), '.dtype') ('a', 'b', 'c', '.dtype') (R.R5R4(R7R8((s*lib/python2.7/site-packages/partd/utils.pytextendœs   (t contextlibRR RR RRRRR#R%R&R)RR+R0R6R9(((s*lib/python2.7/site-packages/partd/utils.pyts