File: //usr/local/lib/python3.6/site-packages/tqdm/__pycache__/utils.cpython-36.pyc
3
LBjK& � ; @ sP d Z ddlZddlZddlZddlmZ ddlmZ ddlm Z ye
Z W n e k
r` e
Z Y nX yeZW n e k
r� eZY nX yeZW n e k
r� eZY nX yeZW n e k
r� eZY nX ejZedd� d6D ��Zed
d� d7D ��Zejd�Z ye�rddl Z ne �W n e k
�r. dZ Y n2X ye j dd� W n e k
�r^ e j � Y nX G dd� de!�Z"G dd� de!�Z#G dd� de!�Z$G dd� de$�Z%G dd� de$�Z&G dd � d e$�Z'd d � Z(d d � Z)d!d"� Z*d#d$� Z+d%d&� Z,d'd(� Z-d)d*� Z.d+d,� Z/d-d.� Z0ydd/l1m2Z2 W n e k
�r2 e3Z4Y n
X d0d1� Z4d2d3� Z5d4d5� Z6dS )8z*
General helpers required for `tqdm.std`.
� N)�wraps)�warn)�proxyc c s | ]}t j|�V qd S )N)�CUR_OS�
startswith)�.0�i� r �*/tmp/pip-build-44150kxb/tqdm/tqdm/utils.py� <genexpr>! s r �win32�cygwinc c s | ]}t j|�V qd S )N)r r )r r r r r
r " s �aix�linux�darwinz\x1b\[[;\d]*[A-Za-z]F)�stripc @ s" e Zd ZdZddd�Zdd� ZdS ) �
FormatReplacezV
>>> a = FormatReplace('something')
>>> "{:5d}".format(a)
'something'
� c C s || _ d| _d S )Nr )�replace�
format_called)�selfr r r r
�__init__9 s zFormatReplace.__init__c C s | j d7 _ | jS )N� )r r )r �_r r r
�
__format__= s zFormatReplace.__format__N)r )�__name__�
__module__� __qualname__�__doc__r r r r r r
r 3 s
r c @ s@ e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� Z dS )�
Comparablez1Assumes child has self._comparable attr/@propertyc C s | j |j k S )N)� _comparable)r �otherr r r
�__lt__D s zComparable.__lt__c C s | |k p| |kS )Nr )r r! r r r
�__le__G s zComparable.__le__c C s | j |j kS )N)r )r r! r r r
�__eq__J s zComparable.__eq__c C s
| |k S )Nr )r r! r r r
�__ne__M s zComparable.__ne__c C s
| |k S )Nr )r r! r r r
�__gt__P s zComparable.__gt__c C s
| |k S )Nr )r r! r r r
�__ge__S s zComparable.__ge__N)
r r r r r" r# r$ r% r& r' r r r r
r B s r c @ s4 e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd S ) �
ObjectWrapperc C s t | j|�S )N)�getattr�_wrapped)r �namer r r
� __getattr__X s zObjectWrapper.__getattr__c C s t | j||�S )N)�setattrr* )r r+ �valuer r r
� __setattr__[ s zObjectWrapper.__setattr__c C s, y t j| |�S tk
r& t| |�S X dS )z7Actual `self.getattr` rather than self._wrapped.getattrN)�objectr, �AttributeErrorr) )r r+ r r r
�wrapper_getattr^ s z ObjectWrapper.wrapper_getattrc C s t j| ||�S )z7Actual `self.setattr` rather than self._wrapped.setattr)r0 r/ )r r+ r. r r r
�wrapper_setattre s z ObjectWrapper.wrapper_setattrc C s | j d|� dS )z4
Thin wrapper around a given object
r* N)r3 )r �wrappedr r r
r i s zObjectWrapper.__init__N)r r r r, r/ r2 r3 r r r r r
r( W s
r( c s0 e Zd ZdZ� fdd�Zdd� Zdd� Z� ZS )�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 t| �j|� | jd|� d S )N�encoding)�superr5 r r3 )r r4 r6 )� __class__r r
r v s z SimpleTextIOWrapper.__init__c C s | j j|j| jd���S )zP
Encode `s` and pass to the wrapped object's `.write()` method.
r6 )r* �write�encoder2 )r �sr r r
r9 z s zSimpleTextIOWrapper.writec C s | j t|d|�kS )Nr* )r* r) )r r! r r r
r$ � s zSimpleTextIOWrapper.__eq__)r r r r r r9 r$ �
__classcell__r r )r8 r
r5 p s r5 c s4 e Zd ZdZedd� �Z� fdd�Zdd� Z� ZS )�DisableOnWriteErrorzO
Disable the given `tqdm_instance` upon `write()` or `flush()` errors.
c s t ���� �fdd�}|S )zV
Quietly set `tqdm_instance.miniters=inf` if `func` raises `errno=5`.
c s� y
� | |�S t k
r\ } z6|jdkr(� ytd��_W n tk
rJ Y nX W Y d d }~X nT tk
r� } z8dt|�krz� ytd��_W n tk
r� Y nX W Y d d }~X nX d S )N� �inf�closed)�OSError�errno�floatZminiters�ReferenceError�
ValueError�str)�args�kwargs�e)�func�
tqdm_instancer r
�inner� s
z7DisableOnWriteError.disable_on_exception.<locals>.inner)r )rK rJ rL r )rJ rK r
�disable_on_exception� s z(DisableOnWriteError.disable_on_exceptionc sT t t| �j|� t|d�r0| jd| j||j�� t|d�rP| jd| j||j�� d S )Nr9 �flush)r7 r= r �hasattrr3 rM r9 rN )r r4 rK )r8 r r
r � s
z DisableOnWriteError.__init__c C s | j t|d|�kS )Nr* )r* r) )r r! r r r
r$ � s zDisableOnWriteError.__eq__) r r r r � staticmethodrM r r$ r< r r )r8 r
r= � s r= c s e Zd Zd� fdd� Z� ZS )�CallbackIOWrapper�readc s~ t t| �j|� t||��|dkrFt��� �fdd��}| jd|� n4|dkrrt��� �fdd��}| jd|� ntd��dS ) z{
Wrap a given `file`-like object's `read()` or `write()` to report
lengths to the given `callback`
r9 c s �| f|�|�}� t | �� |S )N)�len)�datarG rH �res)�callbackrJ r r
r9 � s z)CallbackIOWrapper.__init__.<locals>.writerR c s �| |�}� t |�� |S )N)rS )rG rH rT )rV rJ r r
rR � s
z(CallbackIOWrapper.__init__.<locals>.readz Can only wrap read/write methodsN)r7 rQ r r) r r3 �KeyError)r rV �stream�methodr9 rR )r8 )rV rJ r
r � s
zCallbackIOWrapper.__init__)rR )r r r r r< r r )r8 r
rQ � s rQ c C sj ydj | � W nR tk
r" dS tk
r` y| j� jd�pFd| kS tk
rZ dS X Y nX dS d S )Nu █▉Fzutf-ZU8T)r: �UnicodeEncodeError� Exception�lowerr )r6 r r r
�_is_utf� s r] c C s$ y
t | j�S tk
r dS X d S )NF)r] r6 r1 )�fpr r r
�_supports_unicode� s
r_ c C s4 t | t�r,x | D ]}t|�dkrdS qW dS t| �S )N� FT)�
isinstancerF �ordr_ )r; �cr r r
� _is_ascii� s
rd c C s$ d} t rt} | dkrt} tr t} | S )zy
Return a function which returns console dimensions (width, height).
Supported: linux, osx, windows, cygwin.
N)�IS_WIN�_screen_shape_windows�_screen_shape_tput�IS_NIX�_screen_shape_linux)Z
_screen_shaper r r
�_screen_shape_wrapper� s rj c C s� y�dd l }ddlm}m} ddlm}m} d }| |kr<d
}n | |krHd }|jj|�}|d�}|jj ||�} | r�|j
d|j �\ }
} } }
}}}}}}}|| || fS W n t k
r� Y nX d S )
Nr )�create_string_buffer�windll)�stdin�stdout� �
� � Z hhhhHhhhhhhi�i����i����)NN)
�struct�ctypesrk rl �sysrm rn �kernel32Z GetStdHandle�GetConsoleScreenBufferInfo�unpack�rawr[ )r^ rs rk rl rm rn Z io_handle�h�csbirU Z_bufxZ_bufyZ_curxZ_curyZ_wattr�left�top�rightZbottomZ_maxxZ_maxyr r r
rf � s$ $rf c sB y(ddl �ddlm� � �fdd�dD �S tk
r< Y nX d S )
zcygwin xterm (windows)r N)�
check_callc s& g | ] }t � �jd | ���d �qS )ztput r )�int�split)r r )r �shlexr r
�
<listcomp> s z&_screen_shape_tput.<locals>.<listcomp>�cols�lines)r� r� )NN)r� �
subprocessr r[ )r r )r r� r
rg s rg c C s� y(ddl m } ddlm} ddlm} W n tk
r< d
S X y&|d|| |d��d d� \}}||fS tk
r� yd d
� dD �S ttfk
r� dS X Y nX d S )Nr )�array)�ioctl)�
TIOCGWINSZrz � � � c S s g | ]}t tj| �d �qS )r )r� �os�environ)r r r r r
r� ( s z'_screen_shape_linux.<locals>.<listcomp>�COLUMNS�LINES)NNz )r� r� )NN) r� �fcntlr� �termiosr� � ImportErrorr[ rW rE )r^ r� r� r� Zrowsr� r r r
ri s ri c s4 t dtdd� t� � � s dS t� �� fdd��} | 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� )�
stacklevelNc s � | �d S )Nr r )r^ )�shaper r
rL 8 s z$_environ_cols_wrapper.<locals>.inner)r �DeprecationWarningrj r )rL r )r� r
�_environ_cols_wrapper- s
r� c C s t jdkrtd krdS dS )N�ntr z[A)r� r+ �coloramar r r r
�
_term_move_up? s r� )�east_asian_widthc C s t dd� t| �D ��S )Nc s s" | ]}t |�d krdndV qdS )ZFWr� r N)r� )r �chr r r
r J s z _text_width.<locals>.<genexpr>)�sum�_unicode)r; r r r
� _text_widthI s r� c C s t tjd| ��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)rT r r r
�disp_lenM s r� c C st t | �t| �kr | d|� S ttj| ��}x t| �|krF| dd� } q,W |rpttj| ��rp| jd�rh| S | d S | S )zB
Trim a string which may contain ANSI control characters.
Nr z[0m���)rS r� �boolr� �search�endswith)rT �lengthZ ansi_presentr r r
� disp_trimU s r� )r r
)r r r )7r r� �reru � functoolsr �warningsr �weakrefr �xrangeZ_range� NameError�range�unichrZ_unich�chr�unicoder� rF �
basestringZ _basestring�platformr �anyre rh �compiler� r� r� �init� TypeErrorr0 r r r( r5 r= rQ r] r_ rd rj rf rg ri r� r� � unicodedatar� rS r� r� r� r r r r
�<module> st
+