3
ajM- @ 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!eeZ"G dd de#Z$G dd de#Z%G dd de#Z&G dd de&e$e%e"Z'G dd de&e$e%eeZ(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)signalSIGINTdefault_int_handler)WINDOWS) format_size)get_indentation)six)BarIncrementalBar) 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 )NencodingZ
empty_fillZfillphases)getattrfiler
Z text_typelistjoinencodeUnicodeEncodeError)Z preferredZfallbackr Z
characters r /usr/lib/python3.6/ui.py_select_progress_class s r c s4 e Zd ZdZ fddZ fddZdd 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_sigintoriginal_handlerr )selfargskwargs) __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" Zsignumframer 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 fddZedd 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 | jS )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" itnxr 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 fddZ ZS ) WindowsMixinc s\ t r jrd _tt j|| t rXtrXtj j _ fdd j_ fdd j_ d S )NFc s j jj S )N)r wrappedisattyr )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* sysstdoutr r0 suffixr r r r rB s rB c @ s& e Zd ZejZdZdd Zdd ZdS )DownloadProgressSpinnerz!%(downloaded)s %(download_speed)sc C s" t | dstj| j| _t| jS )N_phaser)hasattr itertoolscycler 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? loggergetEffectiveLevelloggingINFOwriter 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 ddZdd 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 sR d| j }| jj|d| j | | jj| t|| _ | jj | jj d S )Nr. )rc 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 rg 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 rg r_ rR rA )r" final_statusr r r r& s
zInteractiveSpinner.finish)Nr\ r] )r( r) r* r rg rh r& r r r r r[ s
r[ c @ s. e Zd Zd ddZdd 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 j tjd| j| d S )Nz%s: %s)r` rZ rN infor^ )r" rf r r r rm 0 s
z NonInteractiveSpinner._updatec C s& | j r
d S | jj sd S | jd d S )Nzstill running...)ra r` rY rm )r" r r r rh 5 s
zNonInteractiveSpinner.spinc C s$ | j r
d S | jd|f d| _ d S )Nzfinished with status '%s'T)ra rm )r" rj r r r r& < s z NonInteractiveSpinner.finishN)rl )r( r) r* r rm rh r& r r r r rk ) s
rk 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 )NZcancelederrordone)
rC rD r? rN rO rP rQ r[ rk rS KeyboardInterruptr& Exception)r0 Zspinnerr r r open_spinnerC s
rs )/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 rr Z getLoggerr( rN r Z_BaseBarobjectr r- r= rB rF contextmanagerrS rT r[ rk rs r r r r <module> sB
7
!0