o L)jv@sPdZddlmZddlmZddlZddlZddlZddl Z ddl Z ddl Z ddl Z ddl ZddlmZddlmZddlmZddlmZdd lmZdd lmZdd lmZdd lmZdd lmZddlmZddlmZddlmZddl m!Z!e"dZ#Gdddej$Z%dQddZ&e j'dkrdZ(dZ)ndZ(dZ)Gd d!d!ej*e#Z+Gd"d#d#ej,Z- dRdSd'd(Z.ej/dRdTd*d+Z0ej/ dRdUd/d0Z1ej/ dRdUd1d2Z2Gd3d4d4Z3ej/ dRdVd7d8Z4Gd9d:d:Z5dWdXdBdCZ6dYdFdGZ7e j8dHkrddl9Z9ej/dZdJdKZ:d[dMdNZ;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/r1r2r3r4r6r7r9r:r<rrr>r@rA_completed_intervalsrC autowidthoperatorrF TypeErrortcastrangeiterr-posavgtimestartlast_eta eta_knownfinished max_widthentered current_itemr _is_atty _last_line)selfr+r-r/r1r2r3r4r6r7r9r:r<r>r@rArCrFrrr__init__9sV       zProgressBar.__init__ProgressBar[V]cCsd|_||S)NT)rXrender_progressr\rrr __enter__szProgressBar.__enter__exc_typetype[BaseException] | None exc_valueBaseException | NonetbTracebackType | NonecCs |dSN) render_finish)r\rbrdrfrrr__exit__ zProgressBar.__exit__cabc.Iterator[V]cCs|jstd||S)N.You need to use progress bars in a with block.)rX RuntimeErrorr_ generatorr`rrr__iter__szProgressBar.__iter__rcCs tt|Srh)nextrOr`rrr__next__rkzProgressBar.__next__cCs*|js|jsdS|jt|jdSrh)r4rZr>write AFTER_BARflushr`rrrris  zProgressBar.render_finishfloatcCs(|jrdSt|jt|jp dpddS)N?r)rVminrPrvr-r`rrrpctszProgressBar.pctcCs"|jsdSt|jtt|jSNg)rQsumrvlenr`rrrtime_per_iterationszProgressBar.time_per_iterationcCs&|jdur|js|j|j|jSdSrz)r-rVr}rPr`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:rE)rUrBr~format_)r\rLsecondsminuteshoursrrr format_etas$ zProgressBar.format_etacCs(t|j}|jdur|d|j7}|S)N/)r0rPr-)r\rPrrr format_poss  zProgressBar.format_poscCst|jdddddS)Ndz 4%r)rBryr`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?rE) r-rBryrCr/r1rVlistr}mathcosrPjoin)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)r7r-r9appendrrr6rUrVrr:rYr2r<rr3rrstrip)r\r7 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'rEF)r>r@nl)r4rZr[r<rr>r@rIshutilrCr rmaxget_terminal_sizecolumnsrWr BEFORE_BARrru) 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)NTrwi)rPr-rVrRrTrSrQrU)r\rsteprrr make_step-s zProgressBar.make_steprYV | 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)rYrHrArr_)r\rrYrrrupdateCs   zProgressBar.updatecCsd|_d|_d|_dS)NFT)rUrYrVr`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. rmNrr) rXrnrZrOrYrHr_rr)r\rvrrrrobs     zProgressBar.generator)Nr&r'r(r)FTNFNNNNrr*)"r+r,r-r.r/r0r1r0r2r0r3r0r4r5r6r5r7r8r9r5r:r;r<r=r>r?r@r8rArBrCrBrrD)rr^)rbrcrdrerfrgrrD)rrl)rrrrD)rrvrr0)rrBrrDrh)rrBrYrrrD)rrrr]rarjrprrripropertyryr}r~rrrrrr_rrrrorrrrr%8sL G              .  r%cs,eZdZdfdd Zdfd d ZZS)MaybeStripAnsir t.IO[bytes]r@r5kwargst.Anyc stj|fi|||_dSrh)superr]r@)r\rr@r __class__rrr]s zMaybeStripAnsi.__init__textr0rrBcs|jst|}t|Srh)r@r rrsr\rrrrrss zMaybeStripAnsi.write)rrr@r5rr)rr0rrB)rrrr]rs __classcell__rrrrrsrr@r89t.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.NPAGERrETERM)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 isinstancerLBinaryIOru)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-8FNrrrLESSrEr'-RTrRreplace)shellrenverrorsr)rrrwhichrabsolutename subprocessdictrrrrPopenr0PIPErLrMrrr 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 rrNamedTemporaryFilerLrMrrurcallr0rrunlink) rr@rrrrrrrrfrrrr.s2       rc@s8eZdZdddZdd d Zedd dZdddZdS) _SkipCloser t.IO[t.Any]rrDcCs ||_dSrhr )r\rrrrr]bs z_SkipClose.__init__rr0rcCs t|j|Srh)rr)r\rrrr __getattr__es z_SkipClose.__getattr__rcCs t|jjSrh)rrrr`rrrrhs z_SkipClose.buffercCdSrhrr`rrrrlz_SkipClose.closeN)rrrrD)rr0rr)rrr)rrrr]r rrrrrrrras    rt.TextIO&t.Iterator[tuple[t.TextIO, str, bool]]ccs*t|}|dur d}t|||fVdS)zSimply print unformatted text. This is the ultimate fallback. Don't close the output stream in this case, since it's coming from elsewhere rather than our internal helpers. NF)r r)rr@rrrrrps rc@s^eZdZ    dd ddZd!ddZd"ddZejd#ddZejd$ddZd%ddZdS)&EditorNT.txteditorr=rcabc.Mapping[str, str] | None require_saver5 extensionr0rrDcCs||_||_||_||_dSrh)rrrr)r\rrrrrrrr]s zEditor.__init__cCsh|jdur|jSdD]}tj|}|r|Sq trdSddlm}dD] }||dur1|Sq%dS)N)VISUALEDITORnotepadr)r)zsensible-editorvimnanovi)rrrrrrr)r\keyrrrrrr get_editors    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)rz{editor}: Editing failed: {e})rr)rrrrrrcopyrrrrrrrrOSError) r\rrrrrr exit_coderrrr edit_filess4  zEditor.edit_filesrbytes | bytearray bytes | NonecCr rhrrrrreditr z Editor.editcCr rhrrrrrr%r 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 bytearrayendswithrrencodemkstemprrfdopenrsutimepathgetatimegetmtimer"rropenreaddecode) r\rrdatafdrr timestamprrrrr%s:  $    )NNTr) rr=rrrr5rr0rrDr)rrrrD)rr#rr$)rr=rr=)rr&rr') rrrr]rr"rLoverloadr%rrrrrs   rFurlr0rr5locaterBc Csddl}ddd}tjdkr?dg}|r|d |r|d |||td d }z|j||d W|S|wtrp|r]||}dd|g}z| |WSt y\YdSwzt |WdSt yoYdSwt r|r||}dt j|g}ndg}|r|d||z| |WSt yYdSwz"|rt j||pd}n||}|d|g}|r|WSWdSt y|dr|s|sddl}||YdSYdSw)Nrr>r0rcSs*ddlm}|dr||dd}|S)Nr)unquotezfile://) urllib.parser@r)r>r@rrr _unquote_files  zopen_url.._unquote_filedarwinr7z-Wrz /dev/nullw)stderrexplorerz/select,cygstartz-w.zxdg-open)zhttp://zhttps://r)r>r0rr0)rrrrr7rrrrrr r startfiler r4dirnamer webbrowser) r>rr?rrCrnullrrMrrropen_urlsp                 rOchrDcCs:|dkrt|dkrtst|dkrtrtdSdS)N)rrEOFError)rPrrr_translate_ch_to_exc:s  rUwin32cabc.Iterator[int]ccs dVdS)NrGrrrrr raw_terminalHs rXrcCsZ|rttjgtftj}n ttjgtftj}|}|dvr'||7}t||S)N)à)rLrMCallabler0msvcrtgetwchegetwchrU)rfuncrrrrgetcharLs r`c csttjstd}|}ntj}d}zAt|}z$t||VWt |tj |tj |dur>| WdSWdSt |tj |tj |durW| wwtjybYdSw)Nz/dev/tty)r rrr7filenotermios tcgetattrttysetraw tcsetattr TCSADRAINrrurerror)rr; old_settingsrrrrX~s0        cCsjt(}t|dttjd}|rttjrtj |t ||WdS1s.wYdS)N r) rXrr8r9r rrr rrsrU)rr;rPrrrr`s $)rrrrrh)r@r8rr)r@r8rr)rrr@r8rr)rr r@r8rr )FF)r>r0rr5r?r5rrB)rPr0rrD)rrW)rr5rr0)>__doc__ __future__rcollections.abcabccabc contextlibiorrrrrRtypingrLrrrpathlibrtypesr_compatrr r r r r r exceptionsrutilsrTypeVarrProtocolrr!rrrtGenericr% TextIOWrapperrrcontextmanagerrrrrrrrOrUrr\rXr`rbrdrrrrs~                  N g2 t F  /