YM-@sFddlmZddlmZddlZddlZddlmZmZmZddlZddl Z ddl Z ddl m Z ddl mZddlmZddlmZdd lmZmZdd lmZmZmZdd lmZydd lmZWnek r(dZYnXe jeZ d dZ!e!eeZ"Gddde#Z$Gddde#Z%Gddde#Z&Gddde&e$e%e"Z'Gddde&e$e%eeZ(e j)ddZ*Gddde#Z+Gddde#Z,Gdd 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 Cst|jdd}|s|St|dtjt|dtjg}|tt|dg7}y tjj|j|Wntk r|SYnX|SdS)Nencoding empty_fillfillphases)getattrfiler text_typelistjoinencodeUnicodeEncodeError) preferredfallbackr charactersr +/tmp/pip-build-3puug3g5/pip/pip/utils/ui.py_select_progress_classs   r"csFeZdZdZfddZfddZddZS)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. csJtt|j||tt|j|_|jdkrFt|_dS)z= Save the original SIGINT handler for later. N)superr#__init__rr handle_sigintoriginal_handlerr)selfargskwargs) __class__r r!r%NszInterruptibleMixin.__init__cs'tt|jtt|jdS)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#finishrrr')r()r+r r!r,^szInterruptibleMixin.finishcCs|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(signumframer r r!r&hs z InterruptibleMixin.handle_sigint)__name__ __module__ __qualname____doc__r%r,r&r r )r+r!r#<s  r#csgeZdZfddZeddZeddZeddZd d d ZS) DownloadProgressMixincs8tt|j||dtd|j|_dS)N )r$r3r%r message)r(r)r*)r+r r!r%uszDownloadProgressMixin.__init__cCs t|jS)N)rindex)r(r r r! downloadedysz DownloadProgressMixin.downloadedcCs(|jdkrdStd|jdS)Ngz...z/s)avgr)r(r r r!download_speed}sz$DownloadProgressMixin.download_speedcCs|jrd|jSdS)Nzeta %s)etaeta_td)r(r r r! pretty_etas  z DownloadProgressMixin.pretty_etar9ccs1x |D]}|V|j|qW|jdS)N)nextr,)r(itnxr r r!iters zDownloadProgressMixin.iter) r/r0r1r%propertyr8r;r?rDr r )r+r!r3ss r3cs"eZdZfddZS) WindowsMixincstrjrd_ttj||trtrtjj_fddj_fddj_ dS)NFcsjjjS)N)rwrappedisattyr )r(r r!sz'WindowsMixin.__init__..csjjjS)N)rrGflushr )r(r r!rIs) r hide_cursorr$rFr%r AnsiToWin32rrHrJ)r(r)r*)r+)r(r!r%s  zWindowsMixin.__init__)r/r0r1r%r r )r+r!rFs rFc@s%eZdZejZdZdZdS)DownloadProgressBarz %(percent)d%%z0%(downloaded)s %(download_speed)s %(pretty_eta)sN)r/r0r1sysstdoutrr6suffixr r r r!rMs  rMc@s7eZdZejZdZddZddZdS)DownloadProgressSpinnerz!%(downloaded)s %(download_speed)scCs1t|ds$tj|j|_t|jS)N_phaser)hasattr itertoolscyclerrRr@)r(r r r! next_phasesz"DownloadProgressSpinner.next_phasecCsm|j|}|j}|j|}dj||r;dnd||rMdnd|g}|j|dS)Nr<r4)r6rVrPrwriteln)r(r6phaserPliner r r!updates    zDownloadProgressSpinner.updateN) r/r0r1rNrOrrPrVrZr r r r!rQs   rQc csctrdVnQ|j s0tjtjkr8dVn'|jtz dVWd|jtXdS)N) rrHloggergetEffectiveLevelloggingINFOwriterr)rr r r! hidden_cursors"  r`c@s4eZdZddZddZddZdS) RateLimitercCs||_d|_dS)Nr)_min_update_interval_seconds _last_update)r(min_update_interval_secondsr r r!r%s zRateLimiter.__init__cCs&tj}||j}||jkS)N)timercrb)r(nowdeltar r r!readys  zRateLimiter.readycCstj|_dS)N)rerc)r(r r r!resetszRateLimiter.resetN)r/r0r1r%rhrir r r r!ras   rac@sIeZdZdddddZddZdd Zd d ZdS) InteractiveSpinnerNz-\|/g?cCs||_|dkrtj}||_t||_d|_tj||_ |jj dt |jdd|_ dS)NFr4z ... r) _messagerNrO_filera _rate_limiter _finishedrTrU _spin_cycler_r _width)r(r6r spin_charsrdr r r!r%s     "zInteractiveSpinner.__init__cCsy|j std|j}|jj|d|j||jj|t||_|jj|jjdS)Nr4) rnAssertionErrorrprlr_lenrJrmri)r(statusbackupr r r!_write s  zInteractiveSpinner._writecCs:|jr dS|jjs dS|jt|jdS)N)rnrmrhrwr@ro)r(r r r!spins  zInteractiveSpinner.spincCsD|jr dS|j||jjd|jjd|_dS)N T)rnrwrlr_rJ)r( final_statusr r r!r,s    zInteractiveSpinner.finish)r/r0r1r%rwrxr,r r r r!rjs  rjc@sCeZdZdddZddZddZdd Zd S) NonInteractiveSpinner<cCs2||_d|_t||_|jddS)NFstarted)rkrnrarm_update)r(r6rdr r r!r%*s  zNonInteractiveSpinner.__init__cCs7|j st|jjtjd|j|dS)Nz%s: %s)rnrsrmrir[infork)r(rur r r!r~0s zNonInteractiveSpinner._updatecCs1|jr dS|jjs dS|jddS)Nzstill running...)rnrmrhr~)r(r r r!rx5s  zNonInteractiveSpinner.spincCs.|jr dS|jd|fd|_dS)Nzfinished with status '%s'T)rnr~)r(rzr r r!r,<s zNonInteractiveSpinner.finishN)r/r0r1r%r~rxr,r r r r!r{)s   r{ccstjjr3tjtjkr3t|}n t|}y t tj |VWdQRXWnCt k r|j dYn/t k r|j dYnX|j ddS)Ncancelederrordone) rNrOrHr[r\r]r^rjr{r`KeyboardInterruptr, Exception)r6spinnerr r r! open_spinnerCs$     r)/ __future__rrrTrNrrrre contextlibr] pip.compatr pip.utilsrpip.utils.loggingr pip._vendorr pip._vendor.progress.barr r Zpip._vendor.progress.helpersr rrpip._vendor.progress.spinnerrrr getLoggerr/r[r"_BaseBarobjectr#r3rFrMrQcontextmanagerr`rarjr{rr r r r!sB        7 !0