o _cK&@s<dZddlZddlZddlZddlmZddlmZddlm Z ze Z Wn e y/e Z YnwzeZWn e y?eZYnwzeZWn e yOeZYnwzeZWn e y_eZYnwejZedddDZed dd DZed Zz erddlZneWn eydZYnwzejd d Wn e yeYnwGddde!Z"Gddde!Z#Gddde!Z$Gddde$Z%Gddde$Z&Gddde$Z'ddZ(ddZ)ddZ*d d!Z+d"d#Z,d$d%Z-d&d'Z.d(d)Z/d*d+Z0zdd,l1m2Z2Wn eye3Z4Ynwd-d.Z4d/d0Z5d1d2Z6dS)3z* General helpers required for `tqdm.std`. N)wraps)warnproxycc|]}t|VqdSNCUR_OS startswith.0ir5/usr/local/lib/python3.10/dist-packages/tqdm/utils.py !r)win32cygwinccrrrr rrrr"r)aixlinuxdarwinz\x1b\[[;\d]*[A-Za-z]F)stripc@s"eZdZdZdddZddZdS) FormatReplacezV >>> a = FormatReplace('something') >>> "{:5d}".format(a) 'something' cCs||_d|_dSNr)replace format_called)selfrrrr__init__9s zFormatReplace.__init__cCs|jd7_|jS)N)rr)r_rrr __format__=szFormatReplace.__format__N)r)__name__ __module__ __qualname____doc__rr!rrrrr3s  rc@s@eZdZdZddZddZddZdd Zd d Zd d Z dS) Comparablez1Assumes child has self._comparable attr/@propertycCs |j|jkSrZ _comparablerotherrrr__lt__D zComparable.__lt__cCs||kp||kSrrr(rrr__le__GszComparable.__le__cCs |j|jkSrr'r(rrr__eq__Jr+zComparable.__eq__cCs ||k Srrr(rrr__ne__M zComparable.__ne__cCs ||k Srrr(rrr__gt__Pr/zComparable.__gt__cCs ||k Srrr(rrr__ge__Sr/zComparable.__ge__N) r"r#r$r%r*r,r-r.r0r1rrrrr&Bs r&c@s4eZdZddZddZddZddZd d Zd S) ObjectWrappercCs t|j|Sr)getattr_wrappedrnamerrr __getattr__Xr+zObjectWrapper.__getattr__cCst|j||Sr)setattrr4rr6valuerrr __setattr__[szObjectWrapper.__setattr__cCs,zt||WStyt||YSw)z7Actual `self.getattr` rather than self._wrapped.getattr)objectr7AttributeErrorr3r5rrrwrapper_getattr^s  zObjectWrapper.wrapper_getattrcCst|||S)z7Actual `self.setattr` rather than self._wrapped.setattr)r<r;r9rrrwrapper_setattreszObjectWrapper.wrapper_setattrcCs|d|dS)z4 Thin wrapper around a given object r4N)r?)rwrappedrrrriszObjectWrapper.__init__N)r"r#r$r7r;r>r?rrrrrr2Ws  r2cs0eZdZdZfddZddZddZZS)SimpleTextIOWrapperz Change only `.write()` of the wrapped object by encoding the passed value and passing the result to the wrapped object's `.write()` method. cs tt|||d|dS)Nencoding)superrArr?)rr@rB __class__rrrvszSimpleTextIOWrapper.__init__cCs|j||dS)zP Encode `s` and pass to the wrapped object's `.write()` method. rB)r4writeencoder>)rsrrrrFzszSimpleTextIOWrapper.writecC|jt|d|kSNr4r4r3r(rrrr-zSimpleTextIOWrapper.__eq__)r"r#r$r%rrFr- __classcell__rrrDrrAps  rAcs4eZdZdZeddZfddZddZZS)DisableOnWriteErrorzO Disable the given `tqdm_instance` upon `write()` or `flush()` errors. cstfdd}|S)zV Quietly set `tqdm_instance.miniters=inf` if `func` raises `errno=5`. c sz|i|WSty9}z%|jdkrztd_Wn ty&YnwWYd}~dSWYd}~dSd}~wtyj}z&dt|vrGztd_Wn tyWYnwWYd}~dSWYd}~dSd}~ww)Ninfclosed)OSErrorerrnofloatZminitersReferenceError ValueErrorstr)argskwargsefunc tqdm_instancerrinners2    z7DisableOnWriteError.disable_on_exception..innerr)r]r\r^rr[rdisable_on_exceptionsz(DisableOnWriteError.disable_on_exceptioncsXtt||t|dr|d|||jt|dr*|d|||jdSdS)NrFflush)rCrNrhasattrr?r_rFr`)rr@r]rDrrrs  zDisableOnWriteError.__init__cCrIrJrKr(rrrr-rLzDisableOnWriteError.__eq__) r"r#r$r% staticmethodr_rr-rMrrrDrrNs    rNcseZdZdfdd ZZS)CallbackIOWrapperreadcs~tt||t|||dkr$tfdd}|d|dS|dkr;tfdd}|d|dStd) z{ Wrap a given `file`-like object's `read()` or `write()` to report lengths to the given `callback` rFcs&|g|Ri|}t||Srlen)datarXrYrescallbackr\rrrFs z)CallbackIOWrapper.__init__..writerdcs|i|}t||Srre)rXrYrgrirrrds z(CallbackIOWrapper.__init__..readz Can only wrap read/write methodsN)rCrcrr3rr?KeyError)rrjstreammethodrFrdrDrirrs zCallbackIOWrapper.__init__)rd)r"r#r$rrMrrrDrrcsrcc Csfzd|WdStyYdSty2z|dp#d|kWYSty1YYdSww)Nu█▉Fzutf-ZU8T)rGUnicodeEncodeError Exceptionlowerr )rBrrr_is_utfs    rqcCs"zt|jWStyYdSw)NF)rqrBr=fprrr_supports_unicodes   rtcCs2t|tr|D] }t|dkrdSqdSt|S)NFT) isinstancerWordrt)rHcrrr _is_asciis  rycCs$d}tr t}|dur t}trt}|S)zy Return a function which returns console dimensions (width, height). Supported: linux, osx, windows, cygwin. N)IS_WIN_screen_shape_windows_screen_shape_tputIS_NIX_screen_shape_linux)Z _screen_shaperrr_screen_shape_wrappersrc CszTddl}ddlm}m}ddlm}m}d}||krd}n||kr$d}|j|}|d}|j ||} | rR| d|j \ } } } } }}}}}}}||||fWSWd St y^Yd Sw) Nr)create_string_bufferwindll)stdinstdoutiiiZ hhhhHhhhhhhNN) structctypesrrsysrrZkernel32Z GetStdHandleGetConsoleScreenBufferInfounpackrawro)rsrrrrrZ io_handlehcsbirhZ_bufxZ_bufyZ_curxZ_curyZ_wattrlefttoprightbottomZ_maxxZ_maxyrrrr{s.   r{cs@zddlddlmfdddDWStyYdSw)zcygwin xterm (windows)rN) check_callcs&g|]}td|dqS)ztput r)intsplitr rshlexrr sz&_screen_shape_tput..)colslinesr)r subprocessrro)r rrrr|s   r|c Cszddlm}ddlm}ddlm}Wn tyYdSwz|d|||ddd\}}||fWStyRz d d d DWYSttfyQYYdSww) Nr)array)ioctl) TIOCGWINSZrrzcSsg|] }ttj|dqS)r)rosenvironr rrrr(sz'_screen_shape_linux..)COLUMNSLINES) rfcntlrtermiosr ImportErrorrorkrV)rsrrrrowsrrrrr~s"     r~cs4tdtddtsdStfdd}|S)zd Return a function which returns console width. Supported: linux, osx, windows, cygwin. zQUse `_screen_shape_wrapper()(file)[0]` instead of `_environ_cols_wrapper()(file)`r) stacklevelNcs |dSrrrrshaperrr^8s z$_environ_cols_wrapper..inner)rDeprecationWarningrr)r^rrr_environ_cols_wrapper-srcCstjdkr tdur dSdS)Nntrz)rr6coloramarrrr _term_move_up?sreast_asian_widthcCstddt|DS)Ncss$|] }t|dvr dndVqdS)ZFWrrNr)r chrrrrJs"z_text_width..)sum_unicode)rHrrr _text_widthIsrcCsttd|S)zp Returns the real on-screen length of a string which may contain ANSI control codes and wide chars. r)rRE_ANSIsub)rgrrrdisp_lenMsrcCszt|t|kr|d|Stt|}t||kr'|dd}t||ks|r;tt|r;|dr7|S|dS|S)zB Trim a string which may contain ANSI control characters. Nz)rfrboolrsearchendswith)rglengthZ ansi_presentrrr disp_trimUs    r)7r%rrer functoolsrwarningsrweakrefrxrange_range NameErrorrangeunichrZ_unichchrunicoderrW basestringZ _basestringplatformr anyrzr}compilerrrinit TypeErrorr<rr&r2rArNrcrqrtryrr{r|r~rr unicodedatarrfrrrrrrrs            +