U
.eR6 @ sx d dl mZmZ d dlZd dlZd dlZd dlZd dlZd dlm Z m
Z
mZ d dl m Z d dl
mZmZ d dlmZmZmZ d dlmZ d dlmZ d d lmZ d d
lmZ d d l m Z e rd d l m Z m Z m!Z! zd d
l m"Z" W n e#k
r dZ"Y nX e$e%Z&dd Z'e'eeZ(G dd de)Z*G dd deZ+G dd deZ,G dd de)Z-G dd de)Z.G dd de.e*e-Z/G d d d e/e(Z0G d d d e/e+Z1G d d! d!e/eZ2G d"d# d#e/eZ3G d$d% d%e/e,Z4G d&d' d'e.e*e-eZ5e1e1fe0e5fe2e5fe3e5fe4e5fd(Z6d7d)d*Z7ej8d+d, Z9G d-d. d.e)Z:G d/d0 d0e)Z;G d1d2 d2e;ZdS )8 )absolute_importdivisionN)SIGINTdefault_int_handlersignal)six) HIDE_CURSOR SHOW_CURSOR)BarFillingCirclesBarIncrementalBar)Spinner)WINDOWS)get_indentation) format_size)MYPY_CHECK_RUNNING)AnyIteratorIO)coloramac C s t | jdd }|s|S t | dt t | dt g}|tt | dg 7 }zt || W n tk
rz | Y S X | S d S )NencodingZ
empty_fillZfillphases)getattrfiler Z text_typelistjoinencodeUnicodeEncodeError)Z preferredZfallbackr Z
characters r :/usr/lib/python3.8/site-packages/pip/_internal/utils/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# U s
zInterruptibleMixin.__init__c s t t| 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, e s zInterruptibleMixin.finishc C s | | || 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$ o s z InterruptibleMixin.handle_sigint)__name__
__module__ __qualname____doc__r# r, r$
__classcell__r r r* r r! C s
r! c @ s e Zd Zdd ZdS ) SilentBarc C s d S Nr r- r r r update| s zSilentBar.updateN)r/ r0 r1 r6 r r r r r4 z s r4 c @ s e Zd ZdZdZdZdZdS ) BlueEmojiBar
%(percent)d%% )u 🔹u 🔷u 🔵N)r/ r0 r1 suffixZ
bar_prefixZ
bar_suffixr r r r r r7 s r7 c sJ e Zd Z fddZedd Zedd Zedd Zd d
d Z Z S )
DownloadProgressMixinc s, t t| j|| dt d | j | _d S )Nr9 )r" r; r# r messager& r* r r r# s z DownloadProgressMixin.__init__c C s
t | jS r5 )r indexr- r r r
downloaded 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_tdr- r r r
pretty_eta s
z DownloadProgressMixin.pretty_etar@ c c s& |D ]}|V | | q| d S r5 )nextr, )r' itnxr r r iter s zDownloadProgressMixin.iter)r@ )
r/ r0 r1 r# propertyr? rA rC rH r3 r r r* r r; s
r; c s e Zd Z fddZ ZS ) WindowsMixinc s\ t r jrd _tt j|| t rXtrXt j _ fdd j_ fdd j_ d S )NFc s j j S r5 )r wrappedisattyr r- r r z'WindowsMixin.__init__..c s j j S r5 )r rK flushr r- r r rM rN )
r Z hide_cursorr" rJ r# r Z AnsiToWin32r rL rO r& r* r- r r# s
zWindowsMixin.__init__)r/ r0 r1 r# r3 r r r* r rJ s rJ c @ s e Zd ZejZdZdZdS )BaseDownloadProgressBarr8 z0%(downloaded)s %(download_speed)s %(pretty_eta)sN)r/ r0 r1 sysstdoutr r= r: r r r r rP s rP c @ s e Zd ZdS )DefaultDownloadProgressBarNr/ r0 r1 r r r r rS s rS c @ s e Zd ZdS )DownloadSilentBarNrT r r r r rU s rU c @ s e Zd ZdS ) DownloadBarNrT r r r r rV s rV c @ s e Zd ZdS )DownloadFillingCirclesBarNrT r r r r rW s rW c @ s e Zd ZdS ) DownloadBlueEmojiProgressBarNrT r r r r rX s rX c @ s& e Zd ZejZdZdd Zdd ZdS )DownloadProgressSpinnerz!%(downloaded)s %(download_speed)sc C s" t | dst| j| _t| jS )N_phaser)hasattr itertoolscycler rZ rD r- r r r
next_phase s
z"DownloadProgressSpinner.next_phasec C sN | j | }| }| j| }d||r*dnd||r6dnd|g}| | d S )NrB r9 )r= r^ r: r Zwriteln)r' r= Zphaser: liner r r r6 s
z DownloadProgressSpinner.updateN) r/ r0 r1 rQ rR r r: r^ r6 r r r r rY s rY )ZoffZonasciiZprettyZemojic C s8 |d ks|dkr t | d jS t | d |djS d S )Nr r@ )max) BAR_TYPESrH )Z progress_barra r r r DownloadProgressProvider s rc c c sP t r d V n@| r"t tjkr*d V n"| t z
d V W 5 | t X d S r5 ) r rL loggergetEffectiveLevelloggingINFOwriter r )r r r r
hidden_cursor s
ri 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 }|| jkS r5 )timerl rk )r' ZnowZdeltar r r ready2 s
zRateLimiter.readyc C s t | _d S r5 )rn rl r- r r r reset8 s zRateLimiter.resetN)r/ r0 r1 r# ro rp r r r r rj , s rj c @ s e Zd Zdd Zdd ZdS )SpinnerInterfacec C s
t d S r5 NotImplementedErrorr- r r r spin> s zSpinnerInterface.spinc C s
t d S r5 rr r' Z final_statusr r r r, B s zSpinnerInterface.finishN)r/ r0 r1 rt r, r r r r rq = s rq c @ s. e Zd Zd ddZdd Zdd Zd
d ZdS )
InteractiveSpinnerN-\|/ ?c C s\ || _ |d krtj}|| _t|| _d| _t|| _ | j
dt | j d d| _ d S )NFr9 z ... r )
_messagerQ rR _filerj
_rate_limiter _finishedr\ r] _spin_cyclerh r _width)r' r= r Z
spin_charsrm r r r r# H s
zInteractiveSpinner.__init__c C sR d| j }| j|d| j | | j| t|| _ | j | j d S )Nr9 )r~ rz rh lenrO r{ rp )r' statusZbackupr r r _writeW s
zInteractiveSpinner._writec C s, | j r
d S | j sd S | t| j d S r5 )r| r{ ro r rD r} r- r r r rt c s
zInteractiveSpinner.spinc C s4 | j r
d S | | | jd | j d| _ d S )N
T)r| r rz rh rO ru r r r r, k s
zInteractiveSpinner.finish)Nrw rx )r/ r0 r1 r# r rt r, r r r r rv G s
rv c @ s. e Zd Zd ddZdd Zdd Zdd Zd
S ) NonInteractiveSpinner< c C s$ || _ d| _t|| _| d d S )NFZstarted)ry r| rj r{ _update)r' r= rm r r r r# z s
z NonInteractiveSpinner.__init__c C s | j td| j| d S )Nz%s: %s)r{ rp rd infory )r' r r r r r s
z NonInteractiveSpinner._updatec C s& | j r
d S | j sd S | d d S )Nzstill running...)r| r{ ro r r- r r r rt s
zNonInteractiveSpinner.spinc C s$ | j r
d S | d|f d| _ d S )Nzfinished with status '%s'T)r| r ru r r r r, s z NonInteractiveSpinner.finishN)r )r/ r0 r1 r# r rt r, r r r r r y s
r c c s t j r"t tjkr"t| }nt| }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 )NZcancelederrorZdone)
rQ rR rL rd re rf rg rv r ri KeyboardInterruptr, Exception)r= Zspinnerr r r open_spinner s
r )N)?Z
__future__r r
contextlibr\ rf rQ rn r r r Z pip._vendorr Zpip._vendor.progressr r Zpip._vendor.progress.barr
r r Z pip._vendor.progress.spinnerr
Zpip._internal.utils.compatr Zpip._internal.utils.loggingr Zpip._internal.utils.miscr Zpip._internal.utils.typingr typingr r r r r Z getLoggerr/ rd r Z_BaseBarobjectr! r4 r7 r; rJ rP rS rU rV rW rX rY rb rc contextmanagerri rj rq rv r r r r r r s~
7
2