o ߒOj{@sNdZddlmZddlmZddlZddlZddlZddl Z ddl Z ddl Z ddl Z ddl ZddlmZddlmZddlmZddlmZdd lmZdd lmZdd lmZdd lmZdd lmZddlmZddlmZddlmZddl m!Z!ddl m"Z"e#dZ$Gdddej%Z&dPddZ'e j(dkrdZ)dZ*ndZ)d Z*Gd!d"d"ej+e$Z,Gd#d$d$ej-Z. dQdRd(d)Z/ej0dQdSd+d,Z1ej0 dQdTd0d1Z2ej0 dQdTd2d3Z3ej0 dQdUd6d7Z4Gd8d9d9Z5dVdWdAdBZ6dXdEdFZ7e j8dGkrddl9Z9ej0dYdIdJZ:dZdLdMZ;dSddlZ e d_d?d@Z d`dAdBZ d`dCdDZd`dEdFZd`dGdHZd`dIdJZd^dKdLZdadNdOZdbdcdRdSZd^dTdUZd\dVdWZdS)d ProgressBarN# %(bar)s FTriterablecabc.Iterable[V] | Nonelength int | None fill_charstr empty_char bar_templateinfo_sephiddenboolshow_eta show_percent bool | Noneshow_positem_show_func)t.Callable[[V | None], str | None] | Nonelabel str | Nonefilet.TextIO | Nonecolorupdate_min_stepsintwidthrNonecCs>||_||_||_||_||_||_| |_| |_| |_| pd|_ | dur.t } | dur.t } | |_ ||_ ||_d|_||_|dk|_|durWddlm}||d}|dkrWd}|durk|durctdtdt|}t||_||_d|_g|_||t|_|_d|_d|_ d|_!d|_"d|_#t$|j |_%d|_&dS)Nr) length_hintziterable or length is requiredzcabc.Iterable[V]F)'r.r0r1r2r3r5r6r8r9r;rrr=r?r@_completed_intervalsrB autowidthoperatorrE TypeErrortcastrangeiterr,posavgtimestartlast_eta eta_knownfinished max_widthentered current_itemr _is_atty _last_line)selfr*r,r.r0r1r2r3r5r6r8r9r;r=r?r@rBrErrr__init__:sV       zProgressBar.__init__ProgressBar[V]cCsd|_||S)NT)rWrender_progressr[rrr __enter__szProgressBar.__enter__exc_typetype[BaseException] | None exc_valueBaseException | NonetbTracebackType | NonecCs |dSN) render_finish)r[rarcrerrr__exit__ zProgressBar.__exit__cabc.Iterator[V]cCs|jstd||S)N.You need to use progress bars in a with block.)rW RuntimeErrorr^ generatorr_rrr__iter__szProgressBar.__iter__rcCs tt|Srg)nextrNr_rrr__next__rjzProgressBar.__next__cCs*|js|jsdS|jt|jdSrg)r3rYr=write AFTER_BARflushr_rrrrhs  zProgressBar.render_finishfloatcCs(|jrdSt|jt|jp dpddS)N?r)rUminrOrur,r_rrrpctszProgressBar.pctcCs"|jsdSt|jtt|jSNg)rPsumrulenr_rrrtime_per_iterationszProgressBar.time_per_iterationcCs&|jdur|js|j|j|jSdSry)r,rUr|rOr_rrretaszProgressBar.etacCs~|jr=t|j}|d}|d}|d}|d}|d}|d}|dkr0dj|td|||dS|dd|dd|dSd S) N<rz#{d}{day_label} {h:02}:{m:02}:{s:02}d)r day_labelhms02:rD)rTrAr}format_)r[rKsecondsminuteshoursrrr format_etas$ zProgressBar.format_etacCs(t|j}|jdur|d|j7}|S)N/)r/rOr,)r[rOrrr format_poss  zProgressBar.format_poscCst|jdddddS)Ndz 4%r)rArxr_rrr format_pctszProgressBar.format_pctcCs|jdurt|j|j}|j|}||j|j|7}|S|jr)|j|j}|St|j|jp0d}|jdkrM|j|tt |j |jdd|j<d |}|S)Nrrg@g?rD) r,rArxrBr.r0rUlistr|mathcosrOjoin)r[ bar_lengthbarcharsrrr format_bars(       zProgressBar.format_barcCs|j}g}|jdur|dur|j }|jr|||r%|||jr5|jr5|js5|| |j durI| |j }|durI|||j |j ||j|dS)N)r;rinfo)r6r,r8appendrrr5rTrUrr9rXr1r;rr2rrstrip)r[r6 info_bits item_inforrrformat_progress_lines,    z ProgressBar.format_progress_linec CsX|jrdS|js|j|jkr|j|_t|j|j|jddSg}|jrYddl}|j }d|_ t | }t d| j|}||krV|jdurV|t|d|j||_||_ |j }|jdurd|j}|t| }t |}|jdus{|j|kr~||_|||d||d|}||jkr||_t||j|jdd|jdSdS)N)r=r?rr&rDF)r=r?nl)r3rYrZr;rr=r?rHshutilrBr rmaxget_terminal_sizecolumnsrVr BEFORE_BARrrt) r[bufr old_widthclutter_length new_width clear_widthlineline_lenrrrr^sF        zProgressBar.render_progressn_stepscCs|j|7_|jdur|j|jkrd|_t|jdkr dSt|_|jr3t|j|j}nt|j}|jdd|g|_|jdu|_dS)NTrvi)rOr,rUrQrSrRrPrT)r[rsteprrr make_step.s zProgressBar.make_steprXV | NonecCsJ|dur||_|j|7_|j|jkr#||j|d|_dSdS)aUpdate the progress bar by advancing a specified number of steps, and optionally set the ``current_item`` for this new position. :param n_steps: Number of steps to advance. :param current_item: Optional item to set as ``current_item`` for the updated position. .. versionchanged:: 8.0 Added the ``current_item`` optional parameter. .. versionchanged:: 8.0 Only render when the number of steps meets the ``update_min_steps`` threshold. Nr)rXrGr@rr^)r[rrXrrrupdateDs   zProgressBar.updatecCsd|_d|_d|_dS)NFT)rTrXrUr_rrrfinish^s zProgressBar.finishccsn|jstd|js|jEdHdS|jD]}||_|jdkr$||V|dq||dS)zReturn a generator which yields the items added to the bar during construction, and updates the progress bar *after* the yielded block returns. rlNrr) rWrmrYrNrXrGr^rr)r[rvrrrrncs     zProgressBar.generator)Nr%r&r'r(FTNFNNNNrr))"r*r+r,r-r.r/r0r/r1r/r2r/r3r4r5r4r6r7r8r4r9r:r;r<r=r>r?r7r@rArBrArrC)rr])rarbrcrdrerfrrC)rrk)rr)rrC)rrurr/)rrArrCrg)rrArXrrrC)rrrr\r`rirorqrhpropertyrxr|r}rrrrrr^rrrrnrrrrr$9sL G              .  r$cs,eZdZdfdd Zdfd d ZZS)MaybeStripAnsir t.IO[bytes]r?r4kwargst.Anyc stj|fi|||_dSrg)superr\r?)r[rr?r __class__rrr\s zMaybeStripAnsi.__init__textr/rrAcs|jst|}t|Srg)r?r rrrr[rrrrrrs zMaybeStripAnsi.write)rrr?r4rr)rr/rrA)rrrr\rr __classcell__rrrrrsrr?r79t.ContextManager[tuple[t.BinaryIO | t.TextIO, str, bool]]cCst}|dur t}ttjrt|st||Sttj dd}|r0t r+t ||St ||Stj ddvr=t||St sEtjdrKt dg|St dg|S) z2Decide what method to use for paging through text.NPAGERrDTERM)dumbemacsos2moreless)rrr sysstdin _nullpagershlexsplitosenvirongetr_tempfilepager _pipepagerplatform startswith)r?stdoutpager_cmd_partsrrr_pager_contextmanagers      r!t.Generator[t.TextIO, None, None]ccst|dY\}}}d}t|rt|j||d}|}z"ttj|VWz|W|dur2|n$|dur<|wwz|W|durL|ww|durV|wwWddS1sbwYdS)zContext manager. Yields a writable file-like object which can be used as an output pager. .. versionadded:: 8.4.0 :param color: controls if the pager supports ANSI colors or not. The default is autodetection. )r?N)r?encoding) rr rrrKrLTextIOrtdetach)r?rrwrapperrrrget_pager_files2      "r cmd_parts list[str]3t.Iterator[tuple[t.BinaryIO | t.TextIO, str, bool]]ccsL|s$tpt}t|| }|VWddS1swYdSddl}|d}|dd}||}|sZtp>t}t|| }|VWddS1sSwYdSt|}|j} ddl} t t j } |dur| dkrt j ddd |} | sd| d<d }n d | vsd | vrd }|durd }| jt|g|d | j| d d d} ttj| j}t|}zGz|||fVWntyYnty}z| |d}~wwWz|Wn tyYnw z| WdStyYnwqz|Wn tyYnw z| W wty#Ynwq)aPage through text by feeding it to another program. Invokes the pager via :class:`subprocess.Popen` with an ``argv`` list produced by :func:`shlex.split`. The command is resolved to an absolute path with :func:`shutil.which` as recommended by the :mod:`subprocess` docs for Windows compatibility. Invoking a pager through this might support colors: if piping to ``less`` and the user hasn't decided on colors, ``LESS=-R`` is set automatically. NrrrLESSrDr&-RTrRFreplace)shellrenverrorsr)rrrrwhichrabsolutename subprocessdictrrrrPopenr/PIPErKrLBinaryIOrr BrokenPipeError Exception terminateclosewaitKeyboardInterrupt)rr?rrrcmd cmd_params cmd_filepathcmd_pathcmd_namerr less_flagscrrerrrrs                    rc csD|s$tpt}t|| }|VWddS1swYdSddl}ddl}|d}||}|sXtp._unquote_filedarwinr4z-Wrz /dev/nullw)stderrexplorerz/select,cygstartz-w.zxdg-open)zhttp://zhttps://r)r;r/rr/)rrrrr4rrrrrrr startfiler r1dirnamer webbrowser) r;rr<rr@rnullrrJrrropen_urlsp                 rLchrCcCs:|dkrt|dkrtst|dkrtrtdSdS)N)rrEOFError)rMrrr_translate_ch_to_excJs  rRwin32cabc.Iterator[int]ccs dVdS)NrFrrrrr raw_terminalXs rUrcCsZ|rttjgtftj}n ttjgtftj}|}|dvr'||7}t||S)N)à)rKrLCallabler/msvcrtgetwchegetwchrR)rfuncrrrrgetchar\s r]c csttjstd}|}ntj}d}zAt|}z$t||VWt |tj |tj |dur>| WdSWdSt |tj |tj |durW| wwtjybYdSw)Nz/dev/tty)r rrr4filenotermios tcgetattrttysetraw tcsetattr TCSADRAINrrtrerror)rr8 old_settingsrrrrUs0        cCsjt(}t|dttjd}|rttjrtj |t ||WdS1s.wYdS)N r) rUrr5r6r rrr rrrrR)rr8rMrrrr]s $)rrrrrg)r?r7rr)r?r7rr)rrr?r7rr)rrr?r7rr)FF)r;r/rr4r<r4rrA)rMr/rrC)rrT)rr4rr/)>__doc__ __future__rcollections.abcabccabc contextlibiorrrrrQtypingrKrrrpathlibrtypesr_compatrr r r r r r exceptionsrutilsrrTypeVarrProtocolrr rrrsGenericr$ TextIOWrapperrrcontextmanagerrrrrrrLrRrrYrUr]r_rarrrrs~                   N %m8 t F  /