U
    Zޫ^                     @   s   d dl mZmZ d dlZd dlZd dlZd dlZd dlZd dlm	Z	m
Z
 d dlmZ d dlmZ d dlmZ erd dlmZmZ eeZG dd	 d	eZG d
d deZG dd deZG dd deZejdd Zejdd ZdS )    )absolute_importdivisionN)HIDE_CURSORSHOW_CURSOR)WINDOWS)get_indentation)MYPY_CHECK_RUNNING)IteratorIOc                   @   s   e Zd Zdd Zdd ZdS )SpinnerInterfacec                 C   s
   t  d S NNotImplementedErrorself r   >/tmp/pip-unpacked-wheel-huh_82lg/pip/_internal/cli/spinners.pyspin   s    zSpinnerInterface.spinc                 C   s
   t  d S r   r   r   final_statusr   r   r   finish   s    zSpinnerInterface.finishN)__name__
__module____qualname__r   r   r   r   r   r   r      s   r   c                   @   s.   e Zd ZdddZdd Zdd	 Zd
d ZdS )InteractiveSpinnerN-\|/      ?c                 C   s\   || _ |d krtj}|| _t|| _d| _t|| _	| j
dt  | j  d  d| _d S )NF z ... r   )_messagesysstdout_fileRateLimiter_rate_limiter	_finished	itertoolscycle_spin_cyclewriter   _width)r   messagefile
spin_charsmin_update_interval_secondsr   r   r   __init__    s    
zInteractiveSpinner.__init__c                 C   s\   | j r
td| j }| j|d| j  |  | j| t|| _| j  | j  d S )Nr   )	r$   AssertionErrorr)   r!   r(   lenflushr#   reset)r   statusbackupr   r   r   _write0   s    



zInteractiveSpinner._writec                 C   s,   | j r
d S | j sd S | t| j d S r   )r$   r#   readyr6   nextr'   r   r   r   r   r   =   s
    
zInteractiveSpinner.spinc                 C   s4   | j r
d S | | | jd | j  d| _ d S )N
T)r$   r6   r!   r(   r2   r   r   r   r   r   E   s    

zInteractiveSpinner.finish)Nr   r   )r   r   r   r.   r6   r   r   r   r   r   r   r      s     
r   c                   @   s.   e Zd ZdddZdd Zdd Zdd	 Zd
S )NonInteractiveSpinner<   c                 C   s$   || _ d| _t|| _| d d S )NFstarted)r   r$   r"   r#   _update)r   r*   r-   r   r   r   r.   T   s    
zNonInteractiveSpinner.__init__c                 C   s(   | j r
t| j  td| j| d S )Nz%s: %s)r$   r0   r#   r3   loggerinfor   )r   r4   r   r   r   r=   [   s    

zNonInteractiveSpinner._updatec                 C   s&   | j r
d S | j sd S | d d S )Nzstill running...)r$   r#   r7   r=   r   r   r   r   r   a   s
    
zNonInteractiveSpinner.spinc                 C   s(   | j r
d S | djf t  d| _ d S )Nz%finished with status '{final_status}'T)r$   r=   formatlocalsr   r   r   r   r   i   s    zNonInteractiveSpinner.finishN)r;   )r   r   r   r.   r=   r   r   r   r   r   r   r:   S   s   
r:   c                   @   s$   e Zd Zdd Zdd Zdd ZdS )r"   c                 C   s   || _ d| _d S )Nr   )_min_update_interval_seconds_last_update)r   r-   r   r   r   r.   s   s    zRateLimiter.__init__c                 C   s   t   }|| j }|| jkS r   )timerC   rB   )r   nowdeltar   r   r   r7   x   s    
zRateLimiter.readyc                 C   s   t   | _d S r   )rD   rC   r   r   r   r   r3   ~   s    zRateLimiter.resetN)r   r   r   r.   r7   r3   r   r   r   r   r"   r   s   r"   c              	   c   s   t j r"t tjkr"t| }nt| }z t	t j |V  W 5 Q R X W n> t
k
rj   |d  Y n* tk
r   |d  Y nX |d d S )Ncancelederrordone)r   r    isattyr>   getEffectiveLevelloggingINFOr   r:   hidden_cursorKeyboardInterruptr   	Exception)r*   spinnerr   r   r   open_spinner   s    


rR   c              	   c   sP   t rd V  n@|  r"t tjkr*d V  n"| t z
d V  W 5 | t X d S r   )	r   rJ   r>   rK   rL   rM   r(   r   r   )r+   r   r   r   rN      s    

rN   )
__future__r   r   
contextlibr%   rL   r   rD   pip._vendor.progressr   r   pip._internal.utils.compatr   pip._internal.utils.loggingr   pip._internal.utils.typingr   typingr	   r
   	getLoggerr   r>   objectr   r   r:   r"   contextmanagerrR   rN   r   r   r   r   <module>   s(   

4
