3
�PfM- � @ s� d dl mZ d dl mZ d dlZd dlZd dlmZmZmZ d dlZd dl Z d dl
Z
d dl m Z d dl
mZ d dlmZ d dlmZ d d lmZmZ d d
lmZmZmZ d d lmZ yd d lm Z W n e k
r� dZ Y nX e
j e �Z d
d� Z!e!ee�Z"G dd� de#�Z$G dd� de#�Z%G dd� de#�Z&G dd� de&e$e%e"�Z'G dd� de&e$e%ee�Z(e j)dd� �Z*G dd � d e#�Z+G d d � d e#�Z,G d d � d e#�Z-e j)d!d"� �Z.dS )#� )�absolute_import)�divisionN)�signal�SIGINT�default_int_handler)�WINDOWS)� format_size)�get_indentation)�six)�Bar�IncrementalBar)� WritelnMixin� HIDE_CURSOR� SHOW_CURSOR)�Spinner)�coloramac C s� t | jdd �}|s|S t | dtj� �t | dtj� �g}|tt | dg ��7 }ytj� j|�j|� W n tk
rv |S X | S d S )N�encodingZ
empty_fillZfill�phases)�getattr�filer
Z text_type�list�join�encode�UnicodeEncodeError)Z preferredZfallbackr Z
characters� r �/usr/lib/python3.6/ui.py�_select_progress_class s r c s4 e Zd ZdZ� fdd�Z� fdd�Zdd� Z� ZS )�InterruptibleMixina�
Helper to ensure that self.finish() gets called on keyboard interrupt.
This allows downloads to be interrupted without leaving temporary state
(like hidden cursors) behind.
This class is similar to the progress library's existing SigIntMixin
helper, but as of version 1.2, that helper has the following problems:
1. It calls sys.exit().
2. It discards the existing SIGINT handler completely.
3. It leaves its own handler in place even after an uninterrupted finish,
which will have unexpected delayed effects if the user triggers an
unrelated keyboard interrupt some time after a progress-displaying
download has already completed, for example.
c s4 t t| �j||� tt| j�| _| jdkr0t| _dS )z=
Save the original SIGINT handler for later.
N)�superr �__init__r r �
handle_sigint�original_handlerr )�self�args�kwargs)� __class__r r r N s
zInterruptibleMixin.__init__c s t t| �j� tt| j� dS )z�
Restore the original SIGINT handler after finishing.
This should happen regardless of whether the progress display finishes
normally, or gets interrupted.
N)r r �finishr r r! )r" )r% r r r& ^ s zInterruptibleMixin.finishc C s | j � | j||� dS )z�
Call self.finish() before delegating to the original SIGINT handler.
This handler should only be in place while the progress display is
active.
N)r& r! )r" Zsignum�framer r r r h s z InterruptibleMixin.handle_sigint)�__name__�
__module__� __qualname__�__doc__r r& r �
__classcell__r r )r% r r < s
r c sJ e Zd Z� fdd�Zedd� �Zedd� �Zedd� �Zd d
d �Z� Z S )
�DownloadProgressMixinc s, t t| �j||� dt� d | j | _d S )N� � )r r- r r �message)r" r# r$ )r% r r r u s z DownloadProgressMixin.__init__c C s
t | j�S )N)r �index)r" r r r �
downloadedy s z DownloadProgressMixin.downloadedc C s | j dkrdS td| j �d S )Ng z...� z/s)Zavgr )r" r r r �download_speed} s
z$DownloadProgressMixin.download_speedc C s | j rd| j S dS )Nzeta %s� )ZetaZeta_td)r" r r r �
pretty_eta� s
z DownloadProgressMixin.pretty_etar3 c c s* x |D ]}|V | j |� qW | j� d S )N)�nextr& )r" �it�n�xr r r �iter� s
zDownloadProgressMixin.iter)r3 )
r( r) r* r �propertyr2 r4 r6 r; r, r r )r% r r- s s
r- c s e Zd Z� fdd�Z� ZS )� WindowsMixinc s\ t r� jrd� _tt� �j||� t rXtrXtj� j�� _� fdd�� j_� fdd�� j_ d S )NFc s � j jj� S )N)r �wrapped�isattyr )r" r r �<lambda>� s z'WindowsMixin.__init__.<locals>.<lambda>c s � j jj� S )N)r r> �flushr )r" r r r@ � s )
r Z hide_cursorr r= r r Z AnsiToWin32r r? rA )r" r# r$ )r% )r" r r � s
zWindowsMixin.__init__)r( r) r* r r, r r )r% r r= � s r= c @ s e Zd ZejZdZdZdS )�DownloadProgressBarz
%(percent)d%%z0%(downloaded)s %(download_speed)s %(pretty_eta)sN)r( r) r* �sys�stdoutr r0 �suffixr r r r rB � s rB c @ s& e Zd ZejZdZdd� Zdd� ZdS )�DownloadProgressSpinnerz!%(downloaded)s %(download_speed)sc C s" t | d�stj| j�| _t| j�S )N�_phaser)�hasattr� itertools�cycler rG r7 )r" r r r �
next_phase� s
z"DownloadProgressSpinner.next_phasec C sN | j | }| j� }| j| }dj||r*dnd||r6dnd|g�}| j|� d S )Nr5 r. )r0 rK rE r Zwriteln)r" r0 ZphaserE �liner r r �update� s
z DownloadProgressSpinner.updateN) r( r) r* rC rD r rE rK rM r r r r rF � s rF c c sR t r d V nB| j� s$tj� tjkr,d V n"| jt� z
d V W d | jt� X d S )N) r r? �logger�getEffectiveLevel�logging�INFO�writer r )r r r r �
hidden_cursor� s
rS c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )� RateLimiterc C s || _ d| _d S )Nr )� _min_update_interval_seconds� _last_update)r" �min_update_interval_secondsr r r r � s zRateLimiter.__init__c C s t j � }|| j }|| jkS )N)�timerV rU )r" ZnowZdeltar r r �ready� s
zRateLimiter.readyc C s t j � | _d S )N)rX rV )r" r r r �reset� s zRateLimiter.resetN)r( r) r* r rY rZ r r r r rT � s rT c @ s. e Zd Zd dd�Zdd� Zdd � Zd
d � ZdS )
�InteractiveSpinnerN�-\|/� �?c C s\ || _ |d krtj}|| _t|�| _d| _tj|�| _ | jj
dt � | j d � d| _ d S )NFr. z ... r )
�_messagerC rD �_filerT �
_rate_limiter� _finishedrI rJ � _spin_cyclerR r �_width)r" r0 r Z
spin_charsrW r r r r � s
zInteractiveSpinner.__init__c C s^ | j s t�d| j }| jj|d| j | � | jj|� t|�| _| jj� | jj� d S )N�r. ) ra �AssertionErrorrc r_ rR �lenrA r` rZ )r" �statusZbackupr r r �_write s
zInteractiveSpinner._writec C s, | j r
d S | jj� sd S | jt| j�� d S )N)ra r` rY rh r7 rb )r" r r r �spin s
zInteractiveSpinner.spinc C s4 | j r
d S | j|� | jjd� | jj� d| _ d S )N�
T)ra rh r_ rR rA )r" � final_statusr r r r& s
zInteractiveSpinner.finish)Nr\ r] )r( r) r* r rh ri r& r r r r r[ � s
r[ c @ s. e Zd Zd dd�Zdd� Zdd� Zdd � Zd
S ) �NonInteractiveSpinner�< c C s$ || _ d| _t|�| _| jd� d S )NFZstarted)r^ ra rT r` �_update)r" r0 rW r r r r * s
z NonInteractiveSpinner.__init__c C s* | j s t�| jj� tjd| j|� d S )Nz%s: %s)ra re r` rZ rN �infor^ )r" rg r r r rn 0 s
z NonInteractiveSpinner._updatec C s& | j r
d S | jj� sd S | jd� d S )Nzstill running...)ra r` rY rn )r" r r r ri 5 s
zNonInteractiveSpinner.spinc C s$ | j r
d S | jd|f � d| _ d S )Nzfinished with status '%s'T)ra rn )r" rk r r r r& < s z NonInteractiveSpinner.finishN)rm )r( r) r* r rn ri r& r r r r rl ) s
rl c c s� t jj� r"tj� tjkr"t| �}nt| �}y t t j�� |V W d Q R X W n> t
k
rj |j d� � Y n* t k
r� |j d� � Y n X |j d� d S )NZcanceled�error�done)
rC rD r? rN rO rP rQ r[ rl rS �KeyboardInterruptr& � Exception)r0 Zspinnerr r r � open_spinnerC s
rt )/Z
__future__r r rI rC r r r rX �
contextlibrP Z
pip.compatr Z pip.utilsr Zpip.utils.loggingr Z pip._vendorr
Zpip._vendor.progress.barr r Z pip._vendor.progress.helpersr
r r Z pip._vendor.progress.spinnerr r rs Z getLoggerr( rN r Z_BaseBar�objectr r- r= rB rF �contextmanagerrS rT r[ rl rt r r r r �<module> sB
7
!0