a
    &ñ»j&  ã                   @   sÚ  d Z ddlZddlZddlZddlmZmZmZ ddlm	Z	 ddl
mZ ddlmZ ddlmZ eeeef\ZZZZejZedd	„ d
D ƒƒZedd	„ dD ƒƒZe d¡Zzer¸ddlZne‚W n eyÔ   dZY n00 zej dd� W n e!�y   e  ¡  Y n0 d2dd„Z"zddl"m"Z" W n e#�y2   Y n0 G dd„ dƒZ$G dd„ dƒZ%G dd„ dƒZ&G dd„ de&ƒZ'G dd„ de&ƒZ(G dd„ de&ƒZ)d d!„ Z*d"d#„ Z+d$d%„ Z,d&d'„ Z-d(d)„ Z.d*d+„ Z/d,d-„ Z0d.d/„ Z1d0d1„ Z2dS )3z*
General helpers required for `tqdm.std`.
é    N)ÚpartialÚpartialmethodÚwraps)Ú	signature©Úeast_asian_width)Úwarn©Úproxyc                 c   s   | ]}t  |¡V  qd S ©N©ÚCUR_OSÚ
startswith©Ú.0Úi© r   ú0E:\soccer\ml_env\Lib\site-packages\tqdm/utils.pyÚ	<genexpr>   ó    r   )Úwin32Úcygwinc                 c   s   | ]}t  |¡V  qd S r   r   r   r   r   r   r      r   )ÚaixÚlinuxÚdarwinZfreebsdz\x1b\[[;\d]*[A-Za-z]F)ÚstripÚ c                    s°   ˆdu ri ‰| d dkr2| dd… } t dtdd� |rH| | › d|› �fn| f}i ‰ |D ]8‰ˆ ¡ d ‰tˆƒ‰ˆ  ‡‡fdd„tj ¡ D ƒ¡ qV|r˜tnt	‰‡ ‡‡fd	d
„}|S )z€
    Basic (env-only) version of [envwrap](https://github.com/tqdm/envwrap).
    Install `envwrap` for config file support.
    NéÿÿÿÿÚ_z*Trailing underscore in `name` is automaticé   ©Ú
stacklevelc                 3   s0   | ](\}}|  ˆ¡r|ˆ d …  ¡ |fV  qd S r   )r   Úlower©r   ÚkÚv)r   Úprefixr   r   r   1   s   zenvwrap.<locals>.<genexpr>c              
      sâ   t | ƒj‰ ‡ fdd„ˆ ¡ D ƒ}|D ]¬}ˆ | }|j|jur‚t|jd|jfƒD ]0}z||| ƒ||< W n tyx   Y qN0  qÐqNq$|jd ur¤t|jƒ|| ƒ||< q$zˆ| || ƒ||< W q$ t	yÎ   Y q$0 q$ˆ| fi |¤ŽS )Nc                    s   i | ]\}}|ˆ v r||“qS r   r   r#   ©Úparamsr   r   Ú
<dictcomp>8   r   z)envwrap.<locals>.wrap.<locals>.<dictcomp>Ú__args__)
r   Ú
parametersÚitemsÚ
annotationÚemptyÚgetattrÚ	ExceptionÚdefaultÚtypeÚKeyError)ÚfuncZ	overridesr$   ÚparamÚtyp)Úenv_overridesÚpartÚtypesr'   r   Úwrap5   s$    

zenvwrap.<locals>.wrap)
r   ÚDeprecationWarningÚupperÚlenÚupdateÚosÚenvironr,   r   r   )ÚnameZappr9   Z	is_methodÚprefixesr:   r   )r7   r   r8   r&   r9   r   Úenvwrap"   s     ÿ
rC   )rC   c                   @   s"   e Zd ZdZddd„Zdd„ ZdS )	ÚFormatReplacezN
    >>> a = FormatReplace('something')
    >>> f"{a:5d}"
    'something'
    r   c                 C   s   || _ d| _d S ©Nr   )ÚreplaceÚformat_called)ÚselfrF   r   r   r   Ú__init__[   s    zFormatReplace.__init__c                 C   s   |  j d7  _ | jS )Né   )rG   rF   )rH   r   r   r   r   Ú
__format___   s    zFormatReplace.__format__N)r   )Ú__name__Ú
__module__Ú__qualname__Ú__doc__rI   rK   r   r   r   r   rD   U   s   
rD   c                   @   s@   e Zd ZdZdd„ Zdd„ Zdd„ Zdd	„ Zd
d„ Zdd„ Z	dS )Ú
Comparablez1Assumes child has self._comparable attr/@propertyc                 C   s   | j |j k S r   ©Z_comparable©rH   Úotherr   r   r   Ú__lt__f   s    zComparable.__lt__c                 C   s   | |k p| |kS r   r   rR   r   r   r   Ú__le__i   s    zComparable.__le__c                 C   s   | j |j kS r   rQ   rR   r   r   r   Ú__eq__l   s    zComparable.__eq__c                 C   s
   | |k S r   r   rR   r   r   r   Ú__ne__o   s    zComparable.__ne__c                 C   s
   | |k S r   r   rR   r   r   r   Ú__gt__r   s    zComparable.__gt__c                 C   s
   | |k  S r   r   rR   r   r   r   Ú__ge__u   s    zComparable.__ge__N)
rL   rM   rN   rO   rT   rU   rV   rW   rX   rY   r   r   r   r   rP   d   s   rP   c                   @   s4   e Zd Zdd„ Zdd„ Zdd„ Zdd„ Zd	d
„ ZdS )ÚObjectWrapperc                 C   s   t | j|ƒS r   )r/   Ú_wrapped©rH   rA   r   r   r   Ú__getattr__z   s    zObjectWrapper.__getattr__c                 C   s   t | j||ƒS r   )Úsetattrr[   ©rH   rA   Úvaluer   r   r   Ú__setattr__}   s    zObjectWrapper.__setattr__c                 C   s0   zt  | |¡W S  ty*   t| |ƒ Y S 0 dS )z7Actual `self.getattr` rather than self._wrapped.getattrN)Úobjectr]   ÚAttributeErrorr/   r\   r   r   r   Úwrapper_getattr€   s    zObjectWrapper.wrapper_getattrc                 C   s   t  | ||¡S )z7Actual `self.setattr` rather than self._wrapped.setattr)rb   ra   r_   r   r   r   Úwrapper_setattr‡   s    zObjectWrapper.wrapper_setattrc                 C   s   |   d|¡ dS )z4
        Thin wrapper around a given object
        r[   N)re   )rH   Úwrappedr   r   r   rI   ‹   s    zObjectWrapper.__init__N)rL   rM   rN   r]   ra   rd   re   rI   r   r   r   r   rZ   y   s
   rZ   c                       s0   e Zd ZdZ‡ fdd„Zdd„ Zdd„ Z‡  ZS )ÚSimpleTextIOWrapperz™
    Change only `.write()` of the wrapped object by encoding the passed
    value and passing the result to the wrapped object's `.write()` method.
    c                    s   t ƒ  |¡ |  d|¡ d S )NÚencoding)ÚsuperrI   re   )rH   rf   rh   ©Ú	__class__r   r   rI   ˜   s    zSimpleTextIOWrapper.__init__c                 C   s   | j  | |  d¡¡¡S )zP
        Encode `s` and pass to the wrapped object's `.write()` method.
        rh   )r[   ÚwriteÚencoderd   )rH   Úsr   r   r   rl   œ   s    zSimpleTextIOWrapper.writec                 C   s   | j t|d|ƒkS ©Nr[   ©r[   r/   rR   r   r   r   rV   ¢   s    zSimpleTextIOWrapper.__eq__)rL   rM   rN   rO   rI   rl   rV   Ú__classcell__r   r   rj   r   rg   ’   s   rg   c                       s4   e Zd ZdZedd„ ƒZ‡ fdd„Zdd„ Z‡  ZS )ÚDisableOnWriteErrorzO
    Disable the given `tqdm_instance` upon `write()` or `flush()` errors.
    c                    s   t ˆƒ‰‡ ‡fdd„}|S )zV
        Quietly set `tqdm_instance.miniters=inf` if `func` raises `errno=5`.
        c                     sº   zˆ | i |¤ŽW S  t yb } z:|jdkr,‚ ztdƒˆ_W n tyL   Y n0 W Y d }~n\d }~0  ty´ } z<dt|ƒvr~‚ ztdƒˆ_W n tyž   Y n0 W Y d }~n
d }~0 0 d S )Né   ÚinfÚclosed)ÚOSErrorÚerrnoÚfloatZminitersÚReferenceErrorÚ
ValueErrorÚstr)ÚargsÚkwargsÚe©r4   Útqdm_instancer   r   Úinner±   s     
z7DisableOnWriteError.disable_on_exception.<locals>.innerr	   )r€   r4   r�   r   r   r   Údisable_on_exceptionª   s    z(DisableOnWriteError.disable_on_exceptionc                    sP   t ƒ  |¡ t|dƒr,|  d|  ||j¡¡ t|dƒrL|  d|  ||j¡¡ d S )Nrl   Úflush)ri   rI   Úhasattrre   r‚   rl   rƒ   )rH   rf   r€   rj   r   r   rI   Ä   s    
ÿ
ÿzDisableOnWriteError.__init__c                 C   s   | j t|d|ƒkS ro   rp   rR   r   r   r   rV   Í   s    zDisableOnWriteError.__eq__)	rL   rM   rN   rO   Ústaticmethodr‚   rI   rV   rq   r   r   rj   r   rr   ¦   s
   
	rr   c                       s   e Zd Zd‡ fdd„	Z‡  ZS )ÚCallbackIOWrapperÚreadc                    sz   t ƒ  |¡ t||ƒ‰|dkrBtˆƒ‡ ‡fdd„ƒ}|  d|¡ n4|dkrntˆƒ‡ ‡fdd„ƒ}|  d|¡ ntdƒ‚dS )	z{
        Wrap a given `file`-like object's `read()` or `write()` to report
        lengths to the given `callback`
        rl   c                    s&   ˆ| g|¢R i |¤Ž}ˆ t | ƒƒ |S r   ©r=   )Údatar|   r}   Úres©Úcallbackr4   r   r   rl   Ú   s    z)CallbackIOWrapper.__init__.<locals>.writer‡   c                     s   ˆ| i |¤Ž}ˆ t |ƒƒ |S r   rˆ   )r|   r}   r‰   r‹   r   r   r‡   á   s    z(CallbackIOWrapper.__init__.<locals>.readz Can only wrap read/write methodsN)ri   rI   r/   r   re   r3   )rH   rŒ   ÚstreamÚmethodrl   r‡   rj   r‹   r   rI   Ò   s    
zCallbackIOWrapper.__init__)r‡   )rL   rM   rN   rI   rq   r   r   rj   r   r†   Ñ   s   r†   c                 C   sp   zd  | ¡ W nX ty"   Y dS  tyf   z|  ¡  d¡pDd| kW  Y S  ty`   Y Y dS 0 Y n0 dS d S )Nu   â–ˆâ–‰Fzutf-ZU8T)rm   ÚUnicodeEncodeErrorr0   r"   r   )rh   r   r   r   Ú_is_utfë   s    r�   c                 C   s&   zt | jƒW S  ty    Y dS 0 d S )NF)r�   rh   rc   ©Úfpr   r   r   Ú_supports_unicodeù   s    r“   c                 C   s2   t | tƒr*| D ]}t|ƒdkr dS qdS t| ƒS )Néÿ   FT)Ú
isinstancer{   Úordr“   )rn   Úcr   r   r   Ú	_is_ascii   s    
r˜   c                  C   s   dd„ } | S )zy
    Return a function which returns console dimensions (width, height).
    Supported: linux, osx, windows, cygwin.
    c                 S   sR   z8ddl m} |t| ddd„ ƒƒ ƒ\}}|d |d fW S  tyL   Y dS 0 d S )Nr   )Úget_terminal_sizeÚfilenoc                   S   s   d S r   r   r   r   r   r   Ú<lambda>  r   z6_screen_shape_wrapper.<locals>.inner.<locals>.<lambda>rJ   )NN)r?   r™   r/   r0   )r’   r™   ÚcolsÚlinesr   r   r   r�     s    z$_screen_shape_wrapper.<locals>.innerr   ©r�   r   r   r   Ú_screen_shape_wrapper	  s    rŸ   c                     s4   t dtdd� tƒ ‰ ˆ sdS tˆ ƒ‡ fdd„ƒ} | 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   r    Nc                    s   ˆ | ƒd S rE   r   r‘   ©Úshaper   r   r�   $  s    z$_environ_cols_wrapper.<locals>.inner)r   r;   rŸ   r   rž   r   r    r   Ú_environ_cols_wrapper  s    ÿr¢   c                   C   s   t jdkrtd u rdS dS )NÚntr   z[A)r?   rA   Úcoloramar   r   r   r   Ú_term_move_up+  s    r¥   c                 C   s   t dd„ t| ƒD ƒƒS )Nc                 s   s"   | ]}t |ƒd v rdndV  qdS )ZFWr   rJ   Nr   )r   Úchr   r   r   r   0  r   z_text_width.<locals>.<genexpr>)Úsumr{   )rn   r   r   r   Ú_text_width/  s    r¨   c                 C   s   t t d| ¡ƒS )zp
    Returns the real on-screen length of a string which may contain
    ANSI control codes and wide chars.
    r   )r¨   ÚRE_ANSIÚsub)r‰   r   r   r   Údisp_len3  s    r«   c                 C   sp   t | ƒt| ƒkr| d|… S tt | ¡ƒ}t| ƒ|krD| dd… } q*|rltt | ¡ƒrl|  d¡rd| S | d S | S )zB
    Trim a string which may contain ANSI control characters.
    Nr   z[0m)r=   r«   Úboolr©   ÚsearchÚendswith)r‰   ÚlengthZansi_presentr   r   r   Ú	disp_trim;  s    r°   )r   NF)3rO   r?   ÚreÚsysÚ	functoolsr   r   r   Úinspectr   Úunicodedatar   Úwarningsr   Úweakrefr
   ÚrangeÚchrr{   Ú_rangeZ_unichZ_unicodeZ_basestringÚplatformr   ÚanyZIS_WINZIS_NIXÚcompiler©   r¤   ÚImportErrorÚinitÚ	TypeErrorrC   ÚModuleNotFoundErrorrD   rP   rZ   rg   rr   r†   r�   r“   r˜   rŸ   r¢   r¥   r¨   r«   r°   r   r   r   r   Ú<module>   sV   



-+	