jv:dZddlmZddlmZddlZddlZddlZddl Z ddl Z ddl Z ddl Z ddl ZddlmZddlmZddlmZddlmZdd lmZdd lmZdd lmZdd lmZdd lmZddlmZddlmZddlmZddl m!Z!ej"dZ#Gddej$Z%dGdZ&e j'dkrdZ(dZ)ndZ(dZ)Gdd ej*e#Z+Gd!d"ej,Z- dHdId&Z.ej/dHdJd(Z0ej/ dHdKd,Z1ej/ dHdKd-Z2Gd.d/Z3ej/ dHdLd2Z4Gd3d4Z5dMdNd<Z6dOd?Z7e j8d@krddl9Z9ej/dPdBZ:dQdDZ;dSddlr?rAr rrCrErF_completed_intervalsrH autowidthoperatorrL TypeErrortcastrangeiterr2posavgtimestartlast_eta eta_knownfinished max_widthentered current_itemr _is_atty _last_line)selfr0r2r4r6r7r8r9r;r<r>r?rArCrErFrHrLs r__init__zProgressBar.__init__9s&#$(     (  ,+2 <'))D|zz  0$%! $z > , , , , , , [2..F||  ~ @AAAv0%--@@H&*8nn   "  %)Y[[0 T]$# %)" &*ty)) &*rProgressBar[V]c<d|_||S)NT)r^render_progressrbs r __enter__zProgressBar.__enter__s    rexc_typetype[BaseException] | None exc_valueBaseException | NonetbTracebackType | Nonec.|dSN) render_finish)rbrirkrms r__exit__zProgressBar.__exit__s rcabc.Iterator[V]c~|jstd||S)N.You need to use progress bars in a with block.)r^ RuntimeErrorrf generatorrgs r__iter__zProgressBar.__iter__s>| QOPP P ~~rrc:tt|Srp)nextrUrgs r__next__zProgressBar.__next__s DJJrc|js|jsdS|jt|jdSrp)r9r`rCwrite AFTER_BARflushrgs rrqzProgressBar.render_finishsF ; dm  F  """ rfloatcp|jrdSt|jt|jpdpdz dS)N?r)r\minrVrr2rgs rpctzProgressBar.pcts< = 348uT[%5A66;! ?DH A"fG "HA"fG "HAFE "HA1uu<CCff D >>>W>>>'>>>>rrcVt|j}|j |d|jz }|S)N/)r5rVr2)rbrVs r format_poszProgressBar.format_poss1$(mm ; " $t{$$ $C rcHt|jdzddddS)Ndz 4%r)rGrrgs r format_pctzProgressBar.format_pcts+dhn%%++++ABB//rc|j >CC ] !.4:-CCDJO!<==E&!++ N $(T-D"DEEKcQ*% ''%..C rcj|j}g}|j ||j }|jr'|||r'|||jr5|jr.|js'|| |j 1| |j }||||j |j ||j|dzS)N)rArinfo)r<r2r>appendrrr;r[r\rr?r_r7rArr8rrstrip)rbr< info_bits item_infos rformat_progress_linez ProgressBar.format_progress_lines3(  ; "|';#},L = 0   T__.. / / /  0   T__.. / / / = 0T^ 0DM 0   T__.. / / /   *++D,=>>I$  +++  (( **955  &(( rc|jrdS|js?|j|jkr-|j|_t |j|j|jdSg}|jrddl}|j }d|_ t| }td| j|z }||krE|j>|t"|d|jz||_||_ |j }|j|j}|t"| }t|}|j |j|kr||_|||d||z zd|}||jkr?||_t ||j|jd|jdSdS)N)rCrErr,rKF)rCrEnl)r9r`rarArrCrErOshutilrHrrmaxget_terminal_sizecolumnsr]r BEFORE_BARrr) rbbufr old_widthclutter_length new_width clear_widthlineline_lens rrfzProgressBar.render_progresss ;  F} $*,,"&*TZditzBBBB F > # MMM IDJ%d&?&?&A&ABBNAv7799ANRSSI9$$)C :&&& 3/000!*"DJj > %.K :((**D>> > !T^h%>%>%DN 4 3+01222wws|| 4? " ""DO DITZE B B B B IOO      # "rn_stepsc|xj|z c_|j|j|jkrd|_tj|jz dkrdStj|_|jr$tj|jz |jz }ntj|jz }|jdd|gz|_|jdu|_dS)NTri)rVr2r\rXrZrYrWr[)rbrsteps r make_stepzProgressBar.make_step-s G ; "tx4;'>'> DM IKK$- '3 . . F  8 ,IKK$*,8DD9;;+D8BCC=D6)D0rr_V | Nonec|||_|xj|z c_|j|jkr7||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)r_rNrFrrf)rbrr_s rupdatezProgressBar.updateCst  # ,D  !!W,!!  $(= = = NN44 5 5 5  " " "()D % % % > =rc0d|_d|_d|_dS)NFT)r[r_r\rgs rfinishzProgressBar.finish]s  rc#DK|jstd|js|jEd{VdS|jD]A}||_|jdkr||V|dB||dS)zReturn a generator which yields the items added to the bar during construction, and updates the progress bar *after* the yielded block returns. ruNrr) r^rvr`rUr_rNrfrr)rbrvs rrwzProgressBar.generatorbs| QOPP P} #y i  $&! ,11((*** A KKMMM  " " " " "r)Nr+r,r-r.FTNFNNNNrr/)"r0r1r2r3r4r5r6r5r7r5r8r5r9r:r;r:r<r=r>r:r?r@rArBrCrDrEr=rFrGrHrGr rI)r rd)rirjrkrlrmrnr rI)r rs)r rr rI)r rr r5)rrGr rIrp)rrGr_rr rI)rrrrcrhrrrxr{rqpropertyrrrrrrrrrfrrrrwrrrr*r*8s"%$(DH $! !#E+E+E+E+E+N          CCCXC 444X4 X * 0000&6,,,,\1111,*****4 ######rr*c,eZdZd fd Zdfd ZxZS)MaybeStripAnsir t.IO[bytes]rEr:kwargst.Anyc Jtj|fi|||_dSrp)superrcrE)rbrrEr __class__s rrczMaybeStripAnsi.__init__s-**6*** rtextr5r rGcr|jst|}t|Srp)rEr rr})rbrrs rr}zMaybeStripAnsi.writes0z $d##Dww}}T"""r)rrrEr:rr)rr5r rG)rrrrcr} __classcell__)rs@rrrs[##########rrrEr=9t.ContextManager[tuple[t.BinaryIO | t.TextIO, str, bool]]cVt}|t}ttjrt|st ||St jtj dd}|r'trt||St||Stj ddvrt ||Ststjdrtdg|Stdg|S) z2Decide what method to use for paging through text.NPAGERrKTERM)dumbemacsos2moreless)r rr sysstdin _nullpagershlexsplitosenvirongetr_tempfilepager _pipepagerplatform startswith)rEstdoutpager_cmd_partss r_pager_contextmanagerrs " # #F~ #)  )F6NN)&%((( k"*..""="=>>O2  :!/599 9/5111 z~~f!222&%((( /cl%%e,,/vh... vh & &&r!t.Generator[t.TextIO, None, None]c#rKt|5\}}}t|rt|j||}n,t |t jrt|||} |V|n#|wxYw ddddS#1swxYwYdS)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. )rE)rEencodingN)rr%rr isinstancerRBinaryIOr)rErrs rget_pager_filers U + + + /H% f % % L#FMRRRFF  + + L#F%(KKKF LLL LLNNNNFLLNNNNN                  s*AB,,B0B,BB,,B03B0 cmd_parts list[str]3t.Iterator[tuple[t.BinaryIO | t.TextIO, str, bool]]c#K|s%tp t}|ddfVdSddl}|d}|dd}||}|s%tp t}|ddfVdSt |}|j}ddl} ttj } |Q|dkrKtj ddd |} | sd | d<d }n d | vsd | vrd }|d}| t|g|zd| j| dd } t!jt j| j} t)| } | ||fVn3#t*$rYn't,$r}| |d}~wwxYw | n#t*$rYnwxYw | dS#t4$rYnwxYw(# | n#t*$rYnwxYw | n#t4$rYnwxYw'wxYw)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-8FNrrrLESSrKr,-RTrRreplace)shellrenverrorsr)r rrwhichrabsolutename subprocessdictrrrrPopenr5PIPErRrSrrr BrokenPipeError Exception terminateclosewaitKeyboardInterrupt)rrErrcmd cmd_params cmd_filepathcmd_pathcmd_namerr less_flagscrres rrrs% %''58::gu$$$$MMM A,C122J<<$$L %''58::gu$$$$ L!!**,,H}H rz  C }V++ vr22JCHHZ4H4HJJ  CKEE J  #"3"3E } X*$o     A F1:qw ' 'E ''H#Xu$$$$$          KKMMMM    D   %       KKMMMM    D   %         s=FH F5H F5F00F55H9G GG G66 HHI HI H+(I*H++I0II IIIIc#K|s%tp t}|ddfVdSddl}ddl}|d}||}|s%tp t}|ddfVdSt |}ddl}ttj } |d}| dd} tj tj| | |fV| | |t%|| jgt)j| jdS#t)j| jwxYw)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)r rrrrrrtempfiler rrNamedTemporaryFilerRrSrrrcallr5rrunlink) rrErrrrrrrrfs rrr.s %''58::gu$$$$MMM A,C<<$$L %''58::gu$$$$ L!!**,,HOOO ,,H } ##e#<eZdZddZdd Zedd Zdd Zd S) _SkipCloser t.IO[t.Any]r rIc||_dSrpr$)rbrs rrcz_SkipClose.__init__bs  rrr5rc,t|j|Srp)r#r)rbrs r __getattr__z_SkipClose.__getattr__est{D)))rrc4t|jjSrp)r$rrrgs rrz_SkipClose.bufferhs$+,---rcdSrprrgs rrz_SkipClose.closels rN)rr%r rI)rr5r r)r rr)rrrrcr(rrrrrrr$r$asp****...X.      rr$t.TextIO&t.Iterator[tuple[t.TextIO, str, bool]]c#XKt|}|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$)rrErs rrrps@!((H } V  h ------rc|eZdZ dddZddZddZejddZejd dZd!dZdS)"EditorNT.txteditorrBrcabc.Mapping[str, str] | None require_saver: extensionr5r rIc>||_||_||_||_dSrp)r1rr3r4)rbr1rr3r4s rrczEditor.__init__s% ("rc|j|jSdD]'}tj|}|r|cS(trdSddlm}dD]}|||cSdS)N)VISUALEDITORnotepadr)r)zsensible-editorvimnanovi)r1rrrrrr)rbkeyrrr1s r get_editorzEditor.get_editors ; "; %  C$$B     9      6  FuV}}( )tr filenamescabc.Iterable[str]cHddl}ddl}|}d}|jr8tj}||j ||j |t|z|}| }|dkr0ttd|dS#t$r7}ttd|||d}~wwxYw)z Open files in the user's editor.rN)argsrz{editor}: Editing failed)r1z{editor}: Editing failed: {e})r1r)rrr>rrrcopyrr rrrrrrOSError) rbr?rrr1rr exit_coders r edit_fileszEditor.edit_filess; "")- 8 %joo''G NN48 $ $ $    U[((4 ??:!AIA~~$011888GG~    1229919MM  sA?C D!*2DD!rbytes | bytearray bytes | NonecdSrprrbrs reditz Editor.edits=@SrcdSrprrJs rrKz Editor.edits47Crstr | bytes | bytearray | Nonestr | bytes | Nonecddl}|d}nt|ttfr|}nb|r|ds|dz }t r*|ddd}n|d}|d|j \}} tj |d 5}| |dddn #1swxYwYtj |tj|tj|d z ftj|}||f|jr:tj||kr tj|dSt)|d 5}|}dddn #1swxYwYt|ttfr|tj|S|dddtj|S#tj|wxYw) Nrrr(z z utf-8-sigrzeditor-)prefixsuffixrrb)rrbytes bytearrayendswithrrencodemkstempr4rfdopenr}utimepathgetatimegetmtimerFr3r!openreaddecode) rbrrdatafdrr" timestamprs rrKz Editor.edits <&)DD ui0 1 1 ,DD DMM$//   ,||D&1188EE{{7++##9T^#LLD 2t$$                  HTBG,,T22BG4D4DT4J4JQ4NO P P P((..I OOTG $ $ $  RW%5%5d%;%;y%H%H IdOOOOOdD!! QVVXX               $ 233  IdOOOO99[))11&$?? IdOOOOBIdOOOOsa+I"C" I""C&&I")C&*B6I"7I"G( I"(G,,I"/G,0 I"%(I""I8)NNTr0) r1rBrr2r3r:r4r5r rIr)r?r@r rI)rrGr rH)rrBr rB)rrMr rN) rrrrcr>rFrRoverloadrKrrrr/r/s"-1! # # # # #"<Z@@@Z@Z777Z7------rr/Furlr5rr:locaterGcddl}dd}tjdkrdg}|r|d|r|d |||t d d } ||| |S#|wxYwtrb|r7||}d d|g} | |S#t$rYdSwxYw tj |n#t$rYdSwxYwdStr|r-||}dtj|g}n/dg}|r|d|| | |S#t$rYdSwxYw |r+tj||pd}n ||}|d|g}|r|SdS#t$r9|dr |s|sddl}||YdSYdSwxYw)Nrrer5r cbddlm}|dr||dd}|S)Nr)unquotezfile://) urllib.parserir)reris r _unquote_filezopen_url.._unquote_filesD(((((( >>) $ $ #'#abb'""C rdarwinr^z-Wrz /dev/nullw)stderrexplorerz/select,cygstartz-w.zxdg-open)zhttp://zhttps://r)rer5r r5)rrrrr^r rrrr rDr startfiler r[dirnamer webbrowser) rerrfrrlrBnullrrvs ropen_urlrxs |xx   KK      KK    MM#&&'''K%% ##D#66;;== JJLLLLDJJLLLL   -$$C C0D !t,,,   ss   S!!!!   ss 1    -$$C 4 45DD>1 2 2 6 $      OOC 11qq sU9(B66C )C>> D  D D%% D32D3F33 GGA$H,,  >6!*RW-v~>>DD6!*RW-v}==D TVV ! ! ! $$&&LBR    rc#Kttjs$td}|}n tj}d} t j|} tj||Vt j |t j |tj || dSdS#t j |t j |tj || wwxYw#t j$rYdSwxYw)Nz/dev/tty)r rrr^filenotermios tcgetattrttysetraw tcsetattr TCSADRAINrrrerror)r"rb old_settingss rrr~s6 ci   Z  ABB!!##BA ",R00L  2!"g&7FFF   """=GGIIIII!=!"g&7FFF   """=GGIIII!}    DD s,!E6C&AE&AD==EEEcht5}tj|dt t jd}|r8tt jrt j |t||cdddS#1swxYwYdS)N r) rrr_r`r rrr rr}r)rrbrys rrrs ^^ rR''(9#)(D(DiPPB %sz** %   $$$  $ $ $                  sB B''B+.B+)rrr r!rp)rEr=r r)rEr=r r)rrrEr=r r)rr+rEr=r r,)FF)rer5rr:rfr:r rG)ryr5r rI)r r)rr:r r5)>__doc__ __future__rcollections.abcabccabc contextlibiorrrrrXtypingrRrrrpathlibrtypesr_compatr r r r r rr exceptionsrutilsrTypeVarrProtocolrr%rrr~Genericr* TextIOWrapperrrcontextmanagerrrrr$rr/rxrrrrrrrrrrrsP #"""""  ))))))&&&&&&&&&&&& AIcNNqz77777d??JIIJII#I#I#I#I#!)A,I#I#I#X #####R%###'''''> . /3dddddN /3/////d          +/ . . . . .qqqqqqqqhCCCCCL<7MMM,,,,,,^NNNJJJ4r