o  jt@sBdZddlmZddlmZddlZddlZddlZddl Z ddl Z ddl Z ddl Z ddl ZddlmZddlmZddlmZddlmZdd lmZdd lmZdd lmZdd lmZdd lmZddlmZddlmZddlmZddl m!Z!e"dZ#Gdddej$Z%dOddZ&e j'dkrdZ(dZ)ndZ(dZ)Gd d!d!ej*e#Z+Gd"d#d#ej,Z- dPdQd'd(Z.ej/dPdRd*d+Z0ej/ dPdSd/d0Z1ej/ dPdSd1d2Z2ej/ dPdTd5d6Z3Gd7d8d8Z4dUdVd@dAZ5dWdDdEZ6e j7dFkr ddl8Z8ej/dXdHdIZ9dYdKdLZ:dSddl;Z;ddlZ 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__9sV       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?rHshutilrBrrmaxget_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[rrXrrrupdateCs   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[rvrrrrnbs     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$8sL 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]c cst|d5\}}}t|rt|j||d}n t|tjr$t|||d}z |VW|n|wWddS1s>wYdS)zContext manager. Yields a writable file-like object which can be used as an output pager. .. versionadded:: 8.4 :param color: controls if the pager supports ANSI colors or not. The default is autodetection. )r?)r?encodingN)rr rr isinstancerKBinaryIOrt)r?rrrrrget_pager_files "r cmd_parts list[str]3t.Iterator[tuple[t.BinaryIO | t.TextIO, str, bool]]ccs|stpt}|ddfVdSddl}|d}|dd}||}|s4tp+t}|ddfVdSt|}|j}ddl} tt j } |duro|dkrot j ddd |} | sed | d<d }n d | vsmd | vrod }|durud}| j t|g|d| j| dd d} ttj| j} t| }zGz| ||fVWntyYnty}z| |d}~wwWz| Wn tyYnw z| WdStyYnwqz| Wn tyYnw z| W wtyYnwq)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. utf-8FNrrrLESSrDr&-RTrRreplace)shellrenverrorsr)rrrwhichrabsolutename subprocessdictrrrrPopenr/PIPErKrLrrr BrokenPipeError Exception terminateclosewaitKeyboardInterrupt)rr?rrcmd cmd_params cmd_filepathcmd_pathcmd_namerr less_flagscrrerrrrs                    rc cs|stpt}|ddfVdSddl}ddl}|d}||}|s2tp)t}|ddfVdSt|}ddl}tt j } |durGd}|j ddd} z&t t j| | |fV| | |t|| jgWt| jdSt| jw)a Page through text by invoking a program on a temporary file. Used as the primary pager strategy on Windows (where piping to ``more`` adds spurious ``\r\n``), and as a fallback on other platforms. The command is resolved to an absolute path with :func:`shutil.which`. rFNrwb)modedelete)rrrrrrrtempfiler rrNamedTemporaryFilerKrLrrtrcallr/rrunlink) rr?rrrrrrrrfrrrr.s2       rt.TextIO&t.Iterator[tuple[t.TextIO, str, bool]]ccs&t|}|dur d}|||fVdS)z>Simply print unformatted text. This is the ultimate fallback.NFr )rr?rrrrras rc@s^eZdZ    dd ddZd!ddZd"ddZejd#ddZejd$ddZd%ddZdS)&EditorNT.txteditorr<rcabc.Mapping[str, str] | None require_saver4 extensionr/rrCcCs||_||_||_||_dSrg)r rr r )r[r rr r rrrr\ms zEditor.__init__cCsh|jdur|jSdD]}tj|}|r|Sq trdSddlm}dD] }||dur1|Sq%dS)N)VISUALEDITORnotepadr)r)zsensible-editorvimnanovi)r rrrrrr)r[keyrrr rrr get_editorys    zEditor.get_editor filenamescabc.Iterable[str]c Csddl}ddl}|}d}|jrtj}||jz#|j| |t ||d}| }|dkr=t t dj|dWdStyW}z t t dj||d|d}~ww)z Open files in the user's editor.rN)argsrz{editor}: Editing failed)r z{editor}: Editing failed: {e})r r)rrrrrrcopyrrrrrrrrOSError) r[rrrr rr exit_coderrrr edit_filess4  zEditor.edit_filesrbytes | bytearray bytes | NonecCdSrgrrrrreditz Editor.editcCrrgrrrrrrr str | bytes | bytearray | Nonestr | bytes | Nonec Csddl}|dur d}n&t|ttfr|}n|r |ds |d7}tr,|ddd}n|d}|jd|j d\}}zt |d  }| |Wdn1sRwYt |t j|t j|d ft j|}||f|jrt j||krWt |dSt|d  }|}Wdn1swYt|ttfr|Wt |S|dddWt |St |w) Nrr#z z utf-8-sigrzeditor-)prefixsuffixrrb)rrbytes bytearrayendswithrrencodemkstempr rfdopenrrutimepathgetatimegetmtimerr ropenreaddecode) r[rrdatafdrr timestamprrrrrs:  $    )NNTr) r r<rr r r4r r/rrCr)rrrrC)rrrr)rr<rr<)rr!rr") rrrr\rrrKoverloadrrrrrrls   rFurlr/rr4locaterAc Csddl}ddd}tjdkr?dg}|r|d |r|d |||td d }z|j||d W|S|wtrr|r_||}dd|g}z| |WSt y^YdSwzt |WdSt yqYdSwt r|r||}dt j|g}ndg}|r|d||z| |WSt yYdSwz"|rt j||pd}n||}|d|g}|r|WSWdSt y|dr|s|sddl}||YdSYdSw)Nrr9r/rcSs*ddlm}|dr||dd}|S)Nr)unquotezfile://) urllib.parser;r)r9r;rrr _unquote_files  zopen_url.._unquote_filedarwinr2z-Wrz /dev/nullw)stderrexplorerz/select,cygstartz-w.zxdg-open)zhttp://zhttps://r)r9r/rr/)rrrrr2rrrrrrr startfiler r/dirnamer webbrowser) r9rr:rr>rnullrrHrrropen_urlsp                rJchrCcCs:|dkrt|dkrtst|dkrtrtdSdS)N)rrEOFError)rKrrr_translate_ch_to_exc&s  rPwin32cabc.Iterator[int]ccs dVdS)NrFrrrrr raw_terminal4s rSrcCsZ|rttjgtftj}n ttjgtftj}|}|dvr'||7}t||S)N)à)rKrLCallabler/msvcrtgetwchegetwchrP)rfuncrrrrgetchar8s r[c csttjstd}|}ntj}d}zAt|}z$t||VWt |tj |tj |dur>| WdSWdSt |tj |tj |durW| wwtjybYdSw)Nz/dev/tty)r rrr2filenotermios tcgetattrttysetraw tcsetattr TCSADRAINrrtrerror)rr6 old_settingsrrrrSjs0        cCsjt(}t|dttjd}|rttjrtj |t ||WdS1s.wYdS)N r) rSrr3r4r rrr rrrrP)rr6rKrrrr[s $)rrrrrg)r?r7rr)r?r7rr)rrr?r7rr)rrr?r7rr)FF)r9r/rr4r:r4rrA)rKr/rrC)rrR)rr4rr/)=__doc__ __future__rcollections.abcabccabc contextlibiorrrrrQtypingrKrrrpathlibrtypesr_compatrr r r r rr exceptionsrutilsrTypeVarrProtocolrr rrrsGenericr$ TextIOWrapperrrcontextmanagerrrrrrrJrPrrWrSr[r]r_rrrrs|                  N g2 t F  /