3 ]M-@sddlmZddlmZddlZddlZddlmZmZmZddlZddl Z ddl Z ddl m Z ddl mZddlmZddlmZdd lmZmZdd lmZmZmZdd lmZydd lmZWnek rdZYnXe 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}ytjj|j|Wntk rv|SX|SdS)Nencoding empty_fillfillphases)getattrfiler text_typelistjoinencodeUnicodeEncodeError) preferredfallbackr charactersr +/tmp/pip-build-8jg9gmni/pip/pip/utils/ui.py_select_progress_classsr"cs4eZdZdZfddZfddZddZZS)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. cs4tt|j||tt|j|_|jdkr0t|_dS)z= Save the original SIGINT handler for later. N)superr#__init__rr handle_sigintoriginal_handlerr)selfargskwargs) __class__r r!r%Ns zInterruptibleMixin.__init__cstt|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&hsz InterruptibleMixin.handle_sigint)__name__ __module__ __qualname____doc__r%r,r& __classcell__r r )r+r!r#<s  r#csJeZdZfddZeddZeddZeddZd d d ZZ S) DownloadProgressMixincs,tt|j||dtd|j|_dS)N )r$r4r%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}s z$DownloadProgressMixin.download_speedcCs|jrd|jSdS)Nzeta %s)etaeta_td)r(r r r! pretty_etas z DownloadProgressMixin.pretty_etar:ccs*x|D]}|V|j|qW|jdS)N)nextr,)r(itnxr r r!iters zDownloadProgressMixin.iter)r:) r/r0r1r%propertyr9r<r@rEr3r r )r+r!r4ss     r4cseZdZfddZZS) WindowsMixincs\trjrd_ttj||trXtrXtjj_fddj_fddj_ dS)NFcs jjjS)N)rwrappedisattyr )r(r r!sz'WindowsMixin.__init__..cs jjjS)N)rrHflushr )r(r r!rJs) r hide_cursorr$rGr%r AnsiToWin32rrIrK)r(r)r*)r+)r(r!r%s zWindowsMixin.__init__)r/r0r1r%r3r r )r+r!rGsrGc@seZdZejZdZdZdS)DownloadProgressBarz %(percent)d%%z0%(downloaded)s %(download_speed)s %(pretty_eta)sN)r/r0r1sysstdoutrr7suffixr r r r!rNsrNc@s&eZdZejZdZddZddZdS)DownloadProgressSpinnerz!%(downloaded)s %(download_speed)scCs"t|dstj|j|_t|jS)N_phaser)hasattr itertoolscyclerrSrA)r(r r r! next_phases z"DownloadProgressSpinner.next_phasecCsN|j|}|j}|j|}dj||r*dnd||r6dnd|g}|j|dS)Nr=r5)r7rWrQrwriteln)r(r7phaserQliner r r!updates    zDownloadProgressSpinner.updateN) r/r0r1rOrPrrQrWr[r r r r!rRsrRc csRtr dVnB|j s$tjtjkr,dVn"|jtz dVWd|jtXdS)N) rrIloggergetEffectiveLevelloggingINFOwriterr)rr r r! hidden_cursors  rac@s$eZdZddZddZddZdS) RateLimitercCs||_d|_dS)Nr)_min_update_interval_seconds _last_update)r(min_update_interval_secondsr r r!r%szRateLimiter.__init__cCstj}||j}||jkS)N)timerdrc)r(nowdeltar r r!readys zRateLimiter.readycCstj|_dS)N)rfrd)r(r r r!resetszRateLimiter.resetN)r/r0r1r%rirjr r r r!rbsrbc@s.eZdZd ddZddZdd Zd d ZdS) InteractiveSpinnerN-\|/?cCs\||_|dkrtj}||_t||_d|_tj||_ |jj dt |jdd|_ dS)NFr5z ... r) _messagerOrP_filerb _rate_limiter _finishedrUrV _spin_cycler`r _width)r(r7r spin_charsrer r r!r%s  zInteractiveSpinner.__init__cCs^|j s td|j}|jj|d|j||jj|t||_|jj|jjdS)Nr5) rqAssertionErrorrsror`lenrKrprj)r(statusbackupr r r!_write s     zInteractiveSpinner._writecCs,|jr dS|jjsdS|jt|jdS)N)rqrprirzrArr)r(r r r!spins  zInteractiveSpinner.spincCs4|jr dS|j||jjd|jjd|_dS)N T)rqrzror`rK)r( final_statusr r r!r,s    zInteractiveSpinner.finish)Nrlrm)r/r0r1r%rzr{r,r r r r!rks   rkc@s.eZdZd ddZddZddZdd Zd S) NonInteractiveSpinner<cCs$||_d|_t||_|jddS)NFstarted)rnrqrbrp_update)r(r7rer r r!r%*s zNonInteractiveSpinner.__init__cCs*|j s t|jjtjd|j|dS)Nz%s: %s)rqrvrprjr\inforn)r(rxr r r!r0s  zNonInteractiveSpinner._updatecCs&|jr dS|jjsdS|jddS)Nzstill running...)rqrprir)r(r r r!r{5s  zNonInteractiveSpinner.spincCs$|jr dS|jd|fd|_dS)Nzfinished with status '%s'T)rqr)r(r}r r r!r,<szNonInteractiveSpinner.finishN)r)r/r0r1r%rr{r,r r r r!r~)s r~ccstjjr"tjtjkr"t|}nt|}y t tj |VWdQRXWn>t k rj|j dYn*t k r|j dYn X|j ddS)Ncancelederrordone) rOrPrIr\r]r^r_rkr~raKeyboardInterruptr, Exception)r7spinnerr r r! open_spinnerCs    r)/ __future__rrrUrOrrrrf 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#r4rGrNrRcontextmanagerrarbrkr~rr r r r!sB          7 !0