U a+d @sdZddlmZddlmZddlmZmZddlm Z ddgiZ d d d gZ Gd d d e Z ddZ ddefdd Zdd Zdd ZdS)z^ Thin wrappers around common functions. Subpackages contain potentially unstable extensions. )warn)tqdm)TqdmDeprecationWarningr) ObjectWrapperz github.com/Z casperdcl tenumeratetziptmapcs2eZdZdZfddZd ddZddZZS) DummyTqdmFilez'Dummy file-like that will write to tqdmcstt||g|_dS)N)superr __init___buf)selfwrapped __class__9/tmp/pip-unpacked-wheel-tf4cttdx/tqdm/contrib/__init__.pyr szDummyTqdmFile.__init__FcCsnt|trdnd}||\}}}|r^t|}tj||j||g||j|d|g|_n |j |dS)N  )endfilenolock) isinstancebytes rpartitiontyperwritejoinr _wrappedappend)rxrnlpreseppostblankrrrrs  zDummyTqdmFile.writec CsR|jrNt|jd}ztj||j||jdWnttfk rLYnXdS)Nr)rr)r rrrrrOSError ValueError)rr&rrr__del__"s zDummyTqdmFile.__del__)F)__name__ __module__ __qualname____doc__r rr) __classcell__rrrrr s  r cCstdtdd|S)zReturns `func`z?This function has no effect, and will be removed in tqdm==5.0.0r) stacklevel)rr)funcrrrbuiltin_iterable+s r1NcKshz ddl}Wntk r Yn.Xt||jrN|||fd|pD|ji|St||fd|i||S)z Equivalent of `numpy.ndenumerate` or builtin `enumerate`. Parameters ---------- tqdm_class : [default: tqdm.auto.tqdm]. rNtotal)Znumpy ImportErrorrZndarrayZ ndenumeratesize enumerate)iterablestartr2 tqdm_class tqdm_kwargsnprrrr2s  cos:|}|dt}t||f|f|D] }|Vq*dS)zr Equivalent of builtin `zip`. Parameters ---------- tqdm_class : [default: tqdm.auto.tqdm]. r8N)copypop tqdm_autozip)Ziter1Z iter2plusr9kwargsr8irrrrEs cost||D]}||Vq dS)zr Equivalent of builtin `map`. Parameters ---------- tqdm_class : [default: tqdm.auto.tqdm]. N)r)function sequencesr9r@rrrr Ss)r-warningsrautorr=Zstdrutilsr __author____all__r r1rrr rrrrs