gtddlmZddlZddlZddlZddlZdZed dZdZdZ dZ d dZ d Z d Z d Zy))contextmanagerNc* |y#|$rYywxYw)NFT)exclamdas +lib/python3.12/site-packages/partd/utils.pyraisesr s!  s  c#tKd|jdz}tj|\}}tj|tj | |tj j|rKtj j|rtj|ytj |yy#tj j|rKtj j|rtj|wtj |wwxYww)N.) lstriptempfilemkstemposcloseremovepathexistsisdirshutilrmtree) extensionhandlefilenames rtmpfilersi&&s++I'' 2FHHHVIIh$ 77>>( #ww}}X& h' (# $277>>( #ww}}X& h' (# $s AD8C A+D8 A,D55D8cFtjdt||zS)z 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 Q)structpacklen)bytess rframer!!s ;;sCJ '% //c#Kd}t|}t}||kr:tjd|||dzd}|dz }||||z||z }||kr9yyw)z Split buffer into frames of concatenated chunks >>> data = frame(b'Hello') + frame(b'World') >>> list(framesplit(data)) # doctest: +SKIP [b'Hello', b'World'] rrN)rlistrunpack)r inchunksnbytess r framesplitr++sn As5z1 VF a%sE!AaCL1!4 QAq6z"" V  a%s AAAc#|Kt||kr|ytdt||D] }||||zyw)z 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')) [b'He', b'll', b'o'] See Also: toolz.partition_all rN)rrange)r(r r's r partition_allr.;s? 5zA~ q#e*a(A1Q3- )s:<ct|tr*r fd|DS|Dcgc]}t|c}S|Scc}w)z 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']] c3:K|]}t|yw)lazyN) nested_get).0r'collr2s r znested_get..\s@CqJq$T22Csr1) isinstancer%r3)indr5r2r's `` rr3r3MsK#t @C@ @>> 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)r7r%flatten)seqitems rr:r:cs1$ dD !t} $ $J  $s $53 5ct|tr||zSt|tr|ddt|d|fzStt||S)z 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') N)r7strtuplesuffixkeyterms rrArA|sR#sTz C 3Bx6#b'40222c#h%%r"ct|trn t|tr|f}n t|f}t|ts|f}||zS)z 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') )r7r@r?rBs rextendrFsH$ D# wD | c5 !f :r"))F) contextlibrrrr rr rr!r+r.r3r:rArFrr"rrIsP%   $ $ 0   $,2&$r"