B
    `d                 @   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-hu9qqap1/pip/_internal/cli/spinners.pyspin   s    zSpinnerInterface.spinc             C   s
   t  d S )N)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 )N)r    r   readyr2   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    r2   r   r$   r.   )r   r   r   r   r   r   E   s    

zInteractiveSpinner.finish)Nr   r   )r   r   r   r*   r2   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    r,   r   r/   loggerinfor   )r   r0   r   r   r   r9   [   s    

zNonInteractiveSpinner._updatec             C   s&   | j r
d S | j sd S | d d S )Nzstill running...)r    r   r3   r9   )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    r9   formatlocals)r   r   r   r   r   r   i   s
    zNonInteractiveSpinner.finishN)r7   )r   r   r   r*   r9   r   r   r   r   r   r   r6   S   s   
r6   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 )N)timer?   r>   )r   nowdeltar   r   r   r3   x   s    
zRateLimiter.readyc             C   s   t   | _d S )N)r@   r?   )r   r   r   r   r/   ~   s    zRateLimiter.resetN)r   r   r   r*   r3   r/   r   r   r   r   r   r   s   r   c          	   c   s   t j r"t tjkr"t| }nt| }y t	t j |V  W d 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   r6   hidden_cursorKeyboardInterruptr   	Exception)r&   spinnerr   r   r   open_spinner   s    


rN   c          	   c   sP   t rd V  n@|  r"t tjkr*d V  n"| t z
d V  W d | t X d S )N)	r   rF   r:   rG   rH   rI   r$   r   r   )r'   r   r   r   rJ      s    

rJ   )
__future__r   r   
contextlibr!   rH   r   r@   Zpip._vendor.progressr   r   pip._internal.utils.compatr   pip._internal.utils.loggingr   pip._internal.utils.typingr   typingr	   r
   	getLoggerr   r:   objectr   r   r6   r   contextmanagerrN   rJ   r   r   r   r   <module>   s$   

4