#tj{,dZddlmZddlmZddlZddlZddlZddl Z ddl Z ddl Z ddl Z ddl ZddlmZddlmZddlmZddlmZdd lmZdd lmZdd lmZdd lmZdd lmZddlmZddlmZddlmZddl m!Z!ddl m"Z"ej#dZ$Gddej%Z&dFdZ'e j(dkrdZ)dZ*ndZ)dZ*Gd d!ej+e$Z,Gd"d#ej-Z. dGdHd'Z/ej0dGdId)Z1ej0 dGdJd-Z2ej0 dGdJd.Z3ej0 dGdKd1Z4Gd2d3Z5dLdMd;Z6dNd>Z7e j8d?krddl9Z9ej0dOdAZ:dPdCZ;dSddlr@r rrBrDrE_completed_intervalsrG autowidthoperatorrK TypeErrortcastrangeiterr1posavgtimestartlast_eta eta_knownfinished max_widthentered current_itemr _is_atty _last_line)selfr/r1r3r5r6r7r8r:r;r=r>r@rBrDrErGrKs r__init__zProgressBar.__init__:s&#$(     (  ,+2 <'))D|zz  0$%! $z > , , , , , , [2..F||  ~ @AAAv0%--@@H&*8nn   "  %)Y[[0 T]$# %)" &*ty)) &*rProgressBar[V]c<d|_||S)NT)r]render_progressras r __enter__zProgressBar.__enter__s    rexc_typetype[BaseException] | None exc_valueBaseException | NonetbTracebackType | Nonec.|dSN) render_finish)rarhrjrls r__exit__zProgressBar.__exit__s rcabc.Iterator[V]c~|jstd||S)N.You need to use progress bars in a with block.)r] RuntimeErrorre generatorrfs r__iter__zProgressBar.__iter__s>| QOPP P ~~rrc:tt|Sro)nextrTrfs r__next__zProgressBar.__next__s DJJrc|js|jsdS|jt|jdSro)r8r_rBwrite AFTER_BARflushrfs rrpzProgressBar.render_finishsF ; dm  F  """ rfloatcp|jrdSt|jt|jpdpdz dS)N?r)r[minrUrr1rfs rpctzProgressBar.pcts< = 348uT[%5A66;! ?DH A"fG "HA"fG "HAFE "HA1uu<CCff D >>>W>>>'>>>>rrcVt|j}|j |d|jz }|S)N/)r4rUr1)rarUs r format_poszProgressBar.format_poss1$(mm ; " $t{$$ $C rcHt|jdzddddS)Ndz 4%r)rFrrfs 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)r@rinfo)r;r1r=appendrrr:rZr[rr>r^r6r@rr7rrstrip)rar; 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)rBrDrr+rJF)rBrDnl)r8r_r`r@rrBrDrNshutilrGrrmaxget_terminal_sizecolumnsr\r BEFORE_BARrr~) rabufr old_widthclutter_length new_width clear_widthlineline_lens rrezProgressBar.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)rUr1r[rWrYrXrVrZ)rarsteps 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^rMrErre)rarr^s rupdatezProgressBar.updateDst  # ,D  !!W,!!  $(= = = NN44 5 5 5  " " "()D % % % > =rc0d|_d|_d|_dS)NFT)rZr^r[rfs 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. rtNrr) r]rur_rTr^rMrerr)rarvs rrvzProgressBar.generatorcs| QOPP P} #y i  $&! ,11((*** A KKMMM  " " " " "r)Nr*r+r,r-FTNFNNNNrr.)"r/r0r1r2r3r4r5r4r6r4r7r4r8r9r:r9r;r<r=r9r>r?r@rArBrCrDr<rErFrGrFr rH)r rc)rhrirjrkrlrmr rH)r rr)r r)r rH)r rr r4)rrFr rHro)rrFr^rr rH)rrrrbrgrqrwrzrppropertyrrrrrrrrrerrrrvrrrr)r)9s"%$(DH $! !#E+E+E+E+E+N          CCCXC 444X4 X * 0000&6,,,,\1111,*****4 ######rr)c,eZdZd fd Zdfd ZxZS)MaybeStripAnsir t.IO[bytes]rDr9kwargst.Anyc Jtj|fi|||_dSro)superrbrD)rarrDr __class__s rrbzMaybeStripAnsi.__init__s-**6*** rtextr4r rFcr|jst|}t|Sro)rDr rr|)rarrs rr|zMaybeStripAnsi.writes0z $d##Dww}}T"""r)rrrDr9rr)rr4r rF)rrrrbr| __classcell__)rs@rrrs[##########rrrDr<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.NPAGERrJTERM)dumbemacsos2moreless)r rr sysstdin _nullpagershlexsplitosenvirongetr_tempfilepager _pipepagerplatform startswith)rDstdoutpager_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#&Kt|5\}}}d}t|rt|j||}|} t jtj|V |||ni#||wwxYw# |||ww#||wwxYwxYwddddS#1swxYwYdS)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. )rDN)rDencoding) rr$rrrQrRTextIOr~detach)rDrrwrappers rget_pager_filersz U + + +%/H%*. f % % $V]%(SSSGF %&6** * * * % &NN$$$&NN$$$$' % &NN$$$$'7&NN$$$$'/%%%%%%%%%%%%%%%%%%sS/D!B,%B9DB))D,C6.CC6C3 3C66DD  D  cmd_parts list[str]3t.Iterator[tuple[t.BinaryIO | t.TextIO, str, bool]]c#HK|sJtp t}t||5}|Vdddn #1swxYwYdSddl}|d}|dd}||}|sJtp t}t||5}|Vdddn #1swxYwYdSt |}|j} ddl} ttj } |Q| dkrKtj ddd |} | sd| d<d }n d | vsd | vrd }|d }| t|g|zd | j| d d } t#jt"j| j}t+|} |||fVn3#t,$rYn't.$r}| |d}~wwxYw |n#t,$rYnwxYw | dS#t6$rYnwxYw(# |n#t,$rYnwxYw | n#t6$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. NrrrLESSrJr+-RTrRFreplace)shellrenverrorsr)r rrrwhichrabsolutename subprocessdictrrrrPopenr4PIPErQrRBinaryIOrr BrokenPipeError Exception terminateclosewaitKeyboardInterrupt)rrDrrrcmd cmd_params cmd_filepathcmd_pathcmd_namerr less_flagscrres rrrs &''58::  & & "HHH               MMM A,C122J<<$$L &''58::  & & "HHH                L!!**,,H}H rz  C }V++ vr22JCHHZ4H4HJJ  CKEE J  #"3"3E } X*$o     A F1:qw ' 'E ''H#Xu$$$$$          KKMMMM    D   %       KKMMMM    D   %         sAA A(B99B=B=GI G?I G?$G::G??IH H%$H%*I I  I J!I('J!( I52J!4I55J!:JJ! JJ!JJ!c#K|sJtp t}t||5}|Vdddn #1swxYwYdSddl}ddl}|d}||}|sJtp t}t||5}|Vdddn #1swxYwYdSt |}ddl} ttj } |d}| dd} tjtj| | |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`. NrFwb)modedelete)r rrrrrrrtempfiler rrNamedTemporaryFilerQrRr r~rcallr4rrunlink) rrDrrrrrrrrrfs rrrCs9 &''58::  & & "HHH               MMM A,C<<$$L &''58::  & & "HHH                L!!**,,HOOO ,,H } ##e#<||_||_||_||_dSro)r(rr*r+)rar(rr*r+s rrbzEditor.__init__s% ("rc|j|jSdD]'}tj|}|r|cS(trdSddlm}dD]}|||cSdS)N)VISUALEDITORnotepadr)r)zsensible-editorvimnanovi)r(rrrrrr)rakeyrrr(s 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)r(z{editor}: Editing failed: {e})r(r)rrr5rrrcopyrrrrrrrrOSError) rar6rrr(rr exit_coders r edit_fileszEditor.edit_filess; "")- 8 %joo''G NN48 $ $ $    U[((4 ??:!AIA~~$011888GG~    1229919MM  sA?C D!*2DD!rbytes | bytearray bytes | NonecdSrorrars reditz Editor.edits=@SrcdSrorrAs rrBz 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-sigzutf-8zeditor-)prefixsuffixrrb)r isinstancebytes bytearrayendswithrrencodemkstempr+rfdopenr|utimepathgetatimegetmtimer=r*r openreaddecode) rarrdatafdrr! timestamprs rrBz 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)NNTr') r(rArr)r*r9r+r4r rHr)r6r7r rH)rr>r r?)rrAr rA)rrDr rE) rrrrbr5r=rQoverloadrBrrrr&r&s"-1! # # # # #"<Z@@@Z@Z777Z7------rr&Furlr4rr9locaterFcddl}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)Nrr]r4r cbddlm}|dr||dd}|S)Nr)unquotezfile://) urllib.parserar)r]ras r _unquote_filezopen_url.._unquote_filesD(((((( >>) $ $ #'#abb'""C rdarwinrVz-Wrz /dev/nullw)stderrexplorerz/select,cygstartz-w.zxdg-open)zhttp://zhttps://r)r]r4r r4)rrrrrVrrrrrr;r startfiler rSdirnamer webbrowser) r]rr^rrdr9nullrrns ropen_urlrps |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 rrrVfilenotermios tcgetattrttysetraw tcsetattr TCSADRAINrr~rerror)r!rZ old_settingss rr{r{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) r{rrWrXr rrr rr|rw)rrZrqs rrrs ^^ rR''(9#)(D(DiPPB %sz** %   $$$  $ $ $                  sB B''B+.B+)rrr r!ro)rDr<r r)rDr<r r)rrrDr<r r)rr"rDr<r r#)FF)r]r4rr9r^r9r rF)rqr4r rH)r ry)rr9r r4)>__doc__ __future__rcollections.abcabccabc contextlibiorrrrrWtypingrQrrrpathlibrtypesr_compatr r r r r rr exceptionsrutilsrrTypeVarrProtocolrr$rrr}Genericr) TextIOWrapperrrcontextmanagerrrrrr&rprwrrr{rrrrrrrs; #"""""  ))))))&&&&&&&&&&&& AIcNNqz77777d??JIIJII#I#I#I#I#!)A,I#I#I#X #####R%###'''''> "%"%"%"%"%J /3jjjjjZ /355555p +/00000&qqqqqqqqhCCCCCL<7MMM,,,,,,^NNNJJJ4r