o Nrf@sddlmZddlZddlZddlZddlZddZedddZdd Zd d Z d d Z dddZ ddZ ddZ ddZdS))contextmanagerNcCs"z|WdS|yYdSw)NFT)excZlamdarrT/var/www/html/software/conda/envs/catlas/lib/python3.10/site-packages/partd/utils.pyraisess  rc csd|d}t|\}}t|t|z |VWtj|r8tj|r1t |dSt|dSdStj|rRtj|rLt |wt|ww)N.) lstriptempfilemkstemposcloseremovepathexistsisdirshutilrmtree) extensionhandlefilenamerrrtmpfiles        rcCstdt||S)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)bytesrrrframe!srccsjd}t|}t}||kr3td|||dd}|d7}||||V||7}||ksdSdS)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)rinchunksnbytesrrr framesplit+s r%ccsDt||kr |VdStdt||D] }||||VqdS)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"rr!rrr partition_all;s  r'Fcs>t|trrfdd|DSfdd|DS|S)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']] c3s|] }t|dVqdS)lazyN nested_get.0r!collr)rr \sznested_get..csg|] }t|dqS)r(r*r,r.rr ^sznested_get..) isinstancer)indr/r)rr.rr+Ms r+ccs0|D]}t|trt|EdHq|VqdS)aJ >>> 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)r2rflatten)seqitemrrrr4cs  r4cCsFt|tr ||St|tr|ddt|d|fStt||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)r2strtuplesuffixkeytermrrrr:|s  r:cCs@t|trnt|tr|f}nt|f}t|ts|f}||S)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') )r2r9r8r;rrrextends    r>)r)F) contextlibrr rr rrrrr%r'r+r4r:r>rrrrs