d jt dZddlmZddlmZddlZddlZddlZddl Z ddl Z ddl Z ddl Z ddl ZddlmZddlmZddlmZddlmZdd lmZdd lmZdd lmZdd lmZdd lmZddlmZddlmZddlmZddl m!Z!ej"dZ#Gddej$Z%dEdZ&e j'dkrdZ(dZ)ndZ(dZ)Gdd ej*e#Z+Gd!d"ej,Z- dFdGd&Z.ej/dFdHd(Z0ej/ dFdId,Z1ej/ dFdId-Z2ej/ dFdJd0Z3Gd1d2Z4dKdLd:Z5dMd=Z6e j7d>krddl8Z8ej/dNd@Z9dOdBZ:dSddl;Z;ddlr@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__9s&#$(     (  ,+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.updateCst  # ,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.generatorbs| 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)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]rDr9kwargst.Anyc Jtj|fi|||_dSro)superrbrD)rarrDr __class__s rrbzMaybeStripAnsi.__init__s-**6*** rtextr4r rFcr|jst|}t|Sro)rDrrr|)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#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. )rD)rDencodingN)rr$rr isinstancerQBinaryIOr~)rDrrs 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-8FNrrrLESSrJr+-RTrRreplace)shellrenverrorsr)r rrwhichrabsolutename subprocessdictrrrrPopenr4PIPErQrRrrr BrokenPipeError Exception terminateclosewaitKeyboardInterrupt)rrDrrcmd 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 rrNamedTemporaryFilerQrRrr~r callr4rrunlink) rrDrrrrrrrrfs rrr.s %''58::gu$$$$MMM A,C<<$$L %''58::gu$$$$ L!!**,,HOOO ,,H } ##e#<Kt|}|d}|||fVdS)z>Simply print unformatted text. This is the ultimate fallback.NFr )rrDrs rrras8 !((H } (E !!!!!!rc|eZdZ dddZddZddZejddZejd dZd!dZdS)"EditorNT.txteditorrArcabc.Mapping[str, str] | None require_saver9 extensionr4r rHc>||_||_||_||_dSro)r'rr)r*)rar'rr)r*s rrbzEditor.__init__ms% ("rc|j|jSdD]'}tj|}|r|cS(trdSddlm}dD]}|||cSdS)N)VISUALEDITORnotepadr)r)zsensible-editorvimnanovi)r'rrrrrr)rakeyrrr's r get_editorzEditor.get_editorys ; "; %  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)rrr4rrrcopyrrrrrrrrOSError) rar5rrr'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=@SrcdSrorr@s rrAz 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 bytearrayendswithrrencodemkstempr*rfdopenr|utimepathgetatimegetmtimer<r)ropenreaddecode) rarrdatafdrr  timestamprs rrAz 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)r5r6r rH)rr=r r>)rrAr rA)rrCr rD) rrrrbr4r<rQoverloadrArrrr%r%ls"-1! # # # # #"<Z@@@Z@Z777Z7------rr%Furlr4rr9locaterFcddl}dd}tjdkrdg}|r|d|r|d |||t d d } ||| |S#|wxYwtrd|r9||}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.parser_r)r[r_s r _unquote_filezopen_url.._unquote_filesD(((((( >>) $ $ #'#abb'""C rdarwinrTz-Wrz /dev/nullw)stderrexplorerz/select,cygstartz-w.zxdg-open)zhttp://zhttps://r)r[r4r r4)rrrrrTrrr rrr:r startfiler rQdirnamer webbrowser) r[rr\rrbr8nullrrls ropen_urlrns  |xx   KK      KK    MM#&&'''K%% ##D#66;;== JJLLLLDJJLLLL   -$$C 03 0 01D !t,,,   ss   S!!!!   ss 1    -$$C 4 45DD>1 2 2 6 $      OOC 11qq sU9(B66C +D D DD'' D54D5 F55 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 rrrTfilenotermios tcgetattrttysetraw tcsetattr TCSADRAINrr~r error)r rX old_settingss rryryjs6 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) ryrrUrVr rrr rr|ru)rrXros 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)ror4r rH)r rw)rr9r r4)=__doc__ __future__rcollections.abcabccabc contextlibiorrrrrWtypingrQrrrpathlibrtypesr_compatr r r r rrr exceptionsrutilsrTypeVarrProtocolrr$rrr}Genericr) TextIOWrapperrrcontextmanagerrrrrr%rnrurr~ryrrrrrrrs( #"""""  ))))))&&&&&&&&&&&& AIcNNqz77777d??JIIJII#I#I#I#I#!)A,I#I#I#X #####R%###'''''> . /3dddddN /3/////d +/"""""qqqqqqqqhCCCCCL<7MMM,,,,,,^NNNJJJ4r