o bP7@sddlZddlZddlZddlZddlZdddZdadadddZGdd d Z Gd d d Z d d Z ddZ Gddde Z GdddZdddZddZdS)NcCsDzd}t|tj|}td|d}|pdWSty!YdSw)z Get the real terminal width abcdefghhhhhrP)fcntlioctltermios TIOCGWINSZstructunpackIOError)fdbufretr1/usr/share/virt-manager/virtinst/_progresspriv.pyterminal_widths  r?cCs,t}tdus|t|krt|a|atS)z6 Get the real terminal width, but cache it for a bit. N)time_term_width_val_term_width_lastr)r cache_timeoutnowrrrterminal_width_cached$s rc@s>eZdZdZdddZdd Zdd d Zdd dZddZdS) TerminalLinezB Help create dynamic progress bars, uses terminal_width_cached(). rNrrcCs.|dur|}||_tt||||_d|_dS)NF)_min_lenmaxrllen_fin)selfmin_restbeg_lenr rrrr__init__3s  zTerminalLine.__init__cC |j|jS)z Usable length for elements. )rrrrrr__len__: zTerminalLine.__len__cCst|j|d|S)zw After a fixed length, split the rest of the line length among a number of different elements (default=2). r)rr)rfixedelementsrrr rest_split>szTerminalLine.rest_splitcCsB|durt|}t||krd|_|jrdS|jt|8_|S)z If there is room left in the line, above min_len, add element. Note that as soon as one add fails all the rest will fail too. NT)lenrr)relementfull_lenrrraddCs zTerminalLine.addcC|jS)zA Current rest of line, same as .rest_split(fixed=0, elements=1). )rr$rrrrestQzTerminalLine.rest)rNrr)r'N) __name__ __module__ __qualname____doc__r"r%r*r/r1rrrrr0s    rc@s<eZdZddZddZddZddZd d Zd d Zd S) BaseMetercCs0d|_d|_d|_d|_d|_d|_t|_dS)Ng333333?r) update_periodtextsize start_timelast_amount_readlast_update_time RateEstimatorrer$rrrr"Ws zBaseMeter.__init__cCs^||_||_t|ttdfvsJ|jdusJt}||_|j||d|_||_ dSNr) r:r;typeintrr<r@startr=r>)rr:r;rrrrrDas zBaseMeter.startcCsZt|tusJt}|jr||j|jkr+|j||||_||_||dSdSr3) rBrCrr>r9r@updater= _do_update)r amount_readrrrrrEmszBaseMeter.updatecCdSr3r)rrGrrrrFzzBaseMeter._do_updatecCs |dSr3)_do_endr$rrrend}s z BaseMeter.endcCrHr3rr$rrrrJrIzBaseMeter._do_endN) r4r5r6r"rDrErFrKrJrrrrr8Vs    r8cCs@||}t|}d|}||dkr|d7}|d|||fS)N=g?-z [%-*.*s])rCr/)tlbar_max_lengthpcbar_lenibar_len progressbarrrr _term_add_bars  rTcCs>|r||kr |ddfS||kr|ddfS|ddfS)Nz !!! Tz ...  F)r/)rNosizer;rrr _term_add_ends rWc@s*eZdZejfddZddZddZdS) TextMetercCst|||_dSr3)r8r"output)rrYrrrr"s  zTextMeter.__init__c CsL|j}t|}t|j}tdd}t|jdk}|d|}|jdurO|dt ||}|d} |d|} d| | |j | ||| f} nJ|j } t | |} |j }|d| }|d } |d |d }|d|} d |d }t|||}d| | |j ||| ||| f } |j| |jdS)Nr | %5sB %srUz %5sB/sz%-*.*s%s%s%s%s z ETA z %2i%%dz %-*.*s%s%s%s%s%s%s )r@ elapsed_time format_number average_raterboolrr/r; format_timer1r:remaining_time fraction_readr*rTrYwriteflush)rrGetimefreadave_dlrN use_hoursui_sizeui_timeui_endui_rateoutrtimefrtimefracui_pcblenui_barrrrrFs:          zTextMeter._do_updatec Cs|j}t|}td}t|jdk}|d|}|dt|j|}t ||j |\}}|} d| | |j |||f} |j | |j dS)NrZrr\r]z %-*.*s%s%s%s )r=rbrrdrr/rer@rarWr;r1r:rYrhri) rrG total_sizerNrmrnrorpnot_donedummyrrrrrrJs zTextMeter._do_endN)r4r5r6sysstderrr"rFrJrrrrrXs 'rXc@sVeZdZdddZddZddZdd Zd d Zd d ZddZ ddZ ddZ dS)r?@cCs(||_d|_d|_d|_d|_d|_dSrA) timescaletotalr<r>r=ave_rate)rrrrrr" s  zRateEstimator.__init__cCs"||_||_||_d|_d|_dSrA)rr<r>r=r)rrrrrrrDs  zRateEstimator.startcCsh|dks ||jkr||_||_d|_dS||j}||j}|jr/||_||||j|j|_||_dSrA)r=r>r_temporal_rolling_aver)rrGr time_diff read_diffrrrrEs    zRateEstimator.updatecCr0)z/get the average transfer rate (in bytes/second))rr$rrrrc+r2zRateEstimator.average_ratecCr#)zEthe time between the start of the transfer and the most recent update)r>r<r$rrrra/r&zRateEstimator.elapsed_timecCs"|jr|jsdS|j|j|jS)zestimated time remainingN)rrr=r$rrrrf3s zRateEstimator.remaining_timecCs,|jdurdS|jdkrdSt|j|jS)z[the fraction of the data that has been read (can be None for unknown transfer size)Nrr)rfloatr=r$rrrrg9s  zRateEstimator.fraction_readcCst||d}|||||S)aqa temporal rolling average performs smooth averaging even when updates come at irregular intervals. This is performed by scaling the "epsilon" according to the time since the last update. Specifically, epsilon = time_diff / timescale As a general rule, the average will take on a completely new value after 'timescale' seconds.r)min _rolling_ave)rrrlast_averepsilonrrrrDsz#RateEstimator._temporal_rolling_avecCsPz||}Wn tyd}Ynw|dur|S|dur|S||d||S)aKperform a "rolling average" iteration a rolling average "folds" new data into an existing average with some weight, epsilon. epsilon must be between 0.0 and 1.0 (inclusive) a value of 0.0 means only the old value (initial value) counts, and a value of 1.0 means only the newest value is considered.Nr)ZeroDivisionError)rrrrr recent_raterrrrOs  zRateEstimator._rolling_aveN)r~) r4r5r6r"rDrErcrarfrgrrrrrrr?s  r?cCsr|dus|dkr|r dSdS|tdkrdSt|}|d}|d}|r3|d}|d}d|||fSd||fS) Nrz--:--:--z--:--infInfinite<z%02i:%02i:%02iz %02i:%02i)rrC)secondsrmminuteshoursrrrrecs  recCsgd}d}d}d}t|d}|pd}||kr,||kr,|d}||}||kr,||kst|tr4d}n |dkr;d}nd }|t|pBdd ||fS) z4Turn numbers into human-readable metric-like numbers) r+kMGTPEZYg@irrz%i%s%sgfffff#@z%.1f%s%sz%.0f%s%s )r, isinstancerCr)numbersymbolsstepthreshdepth max_depthfmtrrrrbws    rb)r)rr)r)rr r|rrrrrrrr8rTrWrXr?rerbrrrrs"     &^  A [