V.jvdZddlmZddlmZddlZddlZddlZddl Z ddl Z ddl Z ddl Z ddl ZddlmZddlmZddlmZddlmZdd lmZdd lmZdd lmZdd lmZdd lmZddlmZddlmZddlmZddl m!Z!ejDdZ#GddejHZ% d/dZ&e jNdk(rdZ(dZ)ndZ(dZ)GddejTe#Z+GddejXZ- d0 d1dZ.ej^d0d2d Z0ej^ d0 d3d!Z1ej^ d0 d3d"Z2Gd#d$Z3ej^ d0 d4d%Z4Gd&d'Z5d5d6d(Z6d7d)Z7e jpd*k(rddl9Z9ej^d8d+Z:d9d,Z;yddl , 2.F|  ~ @AAvv0%-@H&*8n   "  %)YY[0 T]$# %)" &*tyy) &*rc4d|_|j|S)NT)rKrender_progressrOs r __enter__zProgressBar.__enter__s   rc$|jyN) render_finish)rOexc_type exc_valuetbs r__exit__zProgressBar.__exit__s rcp|js td|j|jS)N.You need to use progress bars in a with block.)rK RuntimeErrorrS generatorrTs r__iter__zProgressBar.__iter__s.||OP P ~~rc*tt|SrW)nextrArTs r__next__zProgressBar.__next__s DJrc|js |jsy|jjt|jj yrW)r0rMr6write AFTER_BARflushrTs rrXzProgressBar.render_finishs0 ;;dmm   " rc|jryt|jt|jxsdxsdz dS)N?r)rIminrCfloatrBrTs rpctzProgressBar.pcts5 ==488uT[[%5A6;!secondsminuteshourss r format_etazProgressBar.format_etas >>DHH A"fG "HA"fG "HAFE "HA1u<CCf D 1WRL'">>rclt|j}|j|d|jz }|S)N/)strrCrB)rOrCs r format_poszProgressBar.format_poss3$((m ;; " Qt{{m$ $C rc@t|jdzddddS)Ndz 4%r)rrmrTs r format_pctzProgressBar.format_pcts&dhhn%b)+AB//rcF|jRt|j|jz}|j|z}||j |j|z zz }|S|j r|j|jz}|St|j |jxsdz}|jdk7rU|j|ttj|j|jzdz dz|jz<dj|}|S)Nrrg@g?r)) rBrrmr:r,r-rIlistrrmathcosrCjoin)rO bar_lengthbarcharss r format_barzProgressBar.format_bars ;; "TXX 23J..:-C 4??djj:&=> >C ]]..4::-C DJJO!<=E&&!+ NN $((T-D-D"DEKcQ**% ''%.C rc|j}g}|j| |j }|jr|j|j |r|j|j |j r7|jr+|js|j|j|j.|j|j}||j||j|j|j|jj!|dzj#S)N)r5rinfo)r2rBr3appendrrr1rHrIrr4rLr.r5rr/rrstrip)rOr2 info_bits item_infos rformat_progress_linez ProgressBar.format_progress_lines((  ;; "|';#}},L ==   T__. /    T__. / ==T^^DMM   T__. /    *++D,=,=>I$  +   ( **95  &( rcX|jry|jsW|j|jk7r=|j|_t |j|j |j yg}|jrddl}|j}d|_ t|j}td|jj|z }||krF|j:|j!t"|j!d|jz||_||_ |j}|j |j}|j!t"|j}t|}|j|j|kr||_|j!||j!d||z zdj%|}||jk7rE||_t ||j |j d|j j'yy)N)r6r7r r)F)r6r7nl)r0rMrNr5rr6r7r;shutilr:rrmaxget_terminal_sizecolumnsrJr BEFORE_BARrrh) rObufr old_widthclutter_length new_width clear_widthlineline_lens rrSzProgressBar.render_progresss ;; }}$**,"&**TZZdiitzzB  >>  IDJ%d&?&?&ABNAv779AANRSI9$)C :& 3/0!*"DJjj >> %..K :((*D> >> !T^^h%>%DN 4 3+012wws| 4?? ""DO DIITZZE B IIOO  #rc|xj|z c_|j |j|jk\rd|_tj|jz dkrytj|_|jr/tj|j z |jz }n!tj|j z }|j dd|gz|_|jdu|_y)NTrji)rCrBrIrErGrFrDrH)rOn_stepssteps r make_stepzProgressBar.make_step-s G ;; "txx4;;'> DM IIK$-- '3 .   88IIK$**,8D99;+D88BC=D6)D0rc|||_|xj|z c_|j|jk\r3|j|j|j d|_yy)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)rLr9r8rrS)rOrrLs rupdatezProgressBar.updateCsb  # ,D  !!W,!  $ $(=(= = NN444 5  "()D % >rc.d|_d|_d|_y)NFT)rHrLrIrTs rfinishzProgressBar.finish]s  rc#XK|js td|js|jEd{y|jD]=}||_|j dk(r|j ||jd?|j|j y7rw)zReturn a generator which yields the items added to the bar during construction, and updates the progress bar *after* the yielded block returns. r^Nrr) rKr_rMrArLr9rSrr)rOrvs rr`zProgressBar.generatorbs||OP P}}yy ii$&! ,,1((* A  KKM  " !s3B*B(A3B*)N#rz%(bar)sz FTNFNNNNr)"rPzcabc.Iterable[V] | NonerBz int | Noner,rr-rr.rr/rr0boolr1rr2 bool | Noner3rr4z)t.Callable[[V | None], str | None] | Noner5 str | Noner6zt.TextIO | Noner7rr8rr:rreturnNone)rzProgressBar[V])rYztype[BaseException] | NonerZzBaseException | Noner[zTracebackType | Nonerr)rzcabc.Iterator[V])rrrr)rrlrr)rrrrrW)rrrLzV | Nonerr)rrrrQrUr\rardrXpropertyrmrrrtrrrrrrSrrrr`rrrr'r'8s"%$(DH $! !#E+)E+E+ E+  E+  E+E+E+E+"E+E+BE+E+E+E+ !E+"#E+$ %E+N ,( !      CC 44  * 0&6,\1,*4 #rr'c,eZdZdfd Zdfd ZxZS)MaybeStripAnsic 4t||fi|||_yrW)superrQr7)rOr!r7kwargs __class__s rrQzMaybeStripAnsi.__init__s *6* rcP|js t|}t| |SrW)r7r rrf)rOtextrs rrfzMaybeStripAnsi.writes#zzd#Dw}T""r)r!z t.IO[bytes]r7rrt.Any)rrrr)rrrrQrf __classcell__)rs@rrrs##rrc t}| t}ttjr t|s t ||St jtjjdd}|rtr t||St||Stjjddvr t ||Ststjjdr tdg|Stdg|S)z2Decide what method to use for paging through text.PAGERr)TERM)dumbemacsos2moreless)r rr sysstdin _nullpagershlexsplitosenvirongetr_tempfilepager _pipepagerplatform startswith)r7stdoutpager_cmd_partss r_pager_contextmanagerrs" #F~ #)) F6N&%(( kk"**..""=>O !/59 9/511 zz~~f!22&%(( cll%%e,vh.. vh &&rc#DKt|5\}}}t|rt|j||}n(t |t j rt|||} ||j dddy#|jwxYw#1swYyxYww)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. )r7)r7encodingN)rr"rr isinstancer>BinaryIOrh)r7r!rs rget_pager_filers U +/H% f %#FMMRF  +#F%(KF L LLN , + LLN , +s5 B AB!A?%B6 B ?BBBB c#>K|stxs t}|ddfyddl}|d}|dd}|j|}|stxs t}|ddfyt |j }|j }ddl} ttj} |L|dk(rGtjjddd j|} | sd | d<d }n d | vsd | vrd }|d}| jt|g|zd| j| dd } t!j"t j$| j&} t)| } | ||f | j1 | j3y#t*$rY/t,$r}| j/|d}~wwxYw#t*$rYKwxYw#t4$rYnwxYw^# | j1n#t*$rYnwxYw | j3w#t4$rYnwxYw"xYww)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-8FNrrrLESSr)r-RTrRreplace)shellrenverrorsr)r rrwhichrabsolutename subprocessdictrrrrPopenrPIPEr>r?rrr BrokenPipeError Exception terminateclosewaitKeyboardInterrupt) cmd_partsr7rrcmd cmd_params cmd_filepathcmd_pathcmd_namerr less_flagscrres rrrsF %'58:gu$$ A,C12J<<$L %'58:gu$$ L!**,H}}H rzz C }V+ vr23CHHZ4H3IJ CKE J #"3E } X*$oo    A FF1::qww 'E 'H#Xu$$  KKM C        %    KKM    %  sEHF &F66H8GH F3GF3F..F33G6 G?HGH GHGHHG('H( G41H3G44H9H  H HHHHHc#K|stxs t}|ddfyddl}ddl}|d}|j |}|stxs t}|ddfyt |j }ddl}ttj} |d}|jdd} tjtj| | |f| j| j!|j#t%|| j&gt)j*| j&y#t)j*| j&wxYww)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 rrNamedTemporaryFiler>r?rrhrcallrrrunlink) rr7rrrrrrr rfs rrr.s %'58:gu$$ A,C<<$L %'58:gu$$ L!**,H ,H } ##e# E">!EE"c6eZdZddZddZeddZd dZy) _SkipClosec||_yrWr )rOr!s rrQz_SkipClose.__init__bs  rc.t|j|SrW)rr!)rOrs r __getattr__z_SkipClose.__getattr__est{{D))rc@t|jjSrW)rr!rrTs rrz_SkipClose.bufferhs$++,,--rcyrWrrTs rrz_SkipClose.closels rN)r!z t.IO[t.Any]rr)rrrr)rrr)rrrrQrrrrrrrrras%*.. rrc#JKt|}|d}t|||fyw)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)r!r7rs rrrps.!(H } V h --s!#ceZdZ d d dZd dZd dZejd dZejd dZddZy)EditorNc<||_||_||_||_yrW)editorr require_save extension)rOrrrrs rrQzEditor.__init__s! ("rc|j |jSdD]'}tjj|}|s%|cStryddlm}dD]}|| |cSy)N)VISUALEDITORnotepadr)r)zsensible-editorvimnanovi)rrrrrrr)rOkeyrrrs r get_editorzEditor.get_editorsc ;; ";; %C$B &  6FV}( 7rcddl}ddl}|j}d}|jr9tj j }|j|j |j|j|t|z|}|j}|dk7r$ttdj|y#t$r+}ttdj|||d}~wwxYw)z Open files in the user's editor.rN)argsrz{editor}: Editing failed)rz{editor}: Editing failed: {e})rr)rrr&rrrcopyrrrrrrrrOSError) rO filenamesrrrrr exit_coders r edit_fileszEditor.edit_filess")- 88jjoo'G NN488 $    U[[(4 ?:!AIA~$01888G  129919M  s!A(C C>&C99C>cyrWrrOrs reditz Editor.edits=@rcyrWrr/s rr0z Editor.edits47rcddl}|d}njt|ttfr|}nQ|r|j ds|dz }t r"|j ddjd}n|jd}|jd|j\}} tj|d 5}|j|dddtj|tjj|tjj!|d z ftjj!|}|j#|f|j$r9tjj!||k(r tj&|yt)|d 5}|j+}dddt|ttfrtj&|Sj-dj ddtj&|S#1swY]xYw#1swYyxYw#tj&|wxYw) Nrrr%z z utf-8-sigrzeditor-)prefixsuffixrrb)r rbytes bytearrayendswithrrencodemkstemprrfdopenrfutimepathgetatimegetmtimer-rropenreaddecode) rOrr datafdrr timestamprs rr0z Editor.edits <&)D ui0 1DDMM$/ ||D&188E{{7+##9T^^#LD 2t$ % HHTBGG,,T2BGG4D4DT4JQ4NO P((.I OOTG $  RWW%5%5d%;y%H IIdOdD!QVVX"$ 23 IIdO99[)11&$? IIdO3%$""! IIdOsIH1+H=Boverloadr0rrrrrs|"-1! # #+ # #  #  #"<ZZ@@ZZ77-rrc6ddl}dd}tjdk(r~dg}|r|jd|r|jd|j||t dd} |j || j |jStr8|r||}d d |g} |j|S tj|ytre|r*||}d tjj|g}n'd g}|r|jd|j| |j|S |r*tjj||xsd}n||}|j d|g}|r|j Sy#|jwxYw#t$rYy wxYw#t$rYy wxYw#t$rYy wxYw#t$r/|j!dr|s|sddl}|j |YyYywxYw)NrcJddlm}|jdr ||dd}|S)Nr)unquotezfile://) urllib.parserJr)urlrJs r _unquote_filezopen_url.._unquote_files&( >>) $#ab'"C rdarwinrAz-Wrz /dev/nullw)stderrexplorerz/select,cygstartz-w.zxdg-open)zhttp://zhttps://r)rMrrr)rrrrrArrrrr r*r startfiler r>dirnamer webbrowser) rMrlocaterrNr(nullrrXs ropen_urlr[s ||xx  KK   KK  M#&'K% ##D#6;;= JJL $C C0D !t,,  S!  $C 45D>1 26$  OOC  sU) F0F3G2GAG F03 F?>F? G G GG 3HHcx|dk(r t|dk(rts t|dk(rtr tyy)N)rrEOFError)chs r_translate_ch_to_excrb:s; V|!! V|Cj V|j|rwin32c#Kdyw)Nr+rrrr raw_terminalreHs sc8|rr?Callablermsvcrtgetwchegetwchrb)rfuncrs rgetcharrnLss> 66!**RW-v~~>D66!**RW-v}}=D V ! ! $&LBR  rc#Kttjstd}|j }n tjj }d} t j |} tj||t j|t j|tjj||jyy#t j|t j|tjj||jwwxYw#t j$rYywxYww)Nz/dev/tty)r rrrAfilenotermios tcgetattrttysetraw tcsetattr TCSADRAINrrhrerror)rrE old_settingss rrere~s cii Z AB!!#BA ",,R0L  2!!"g&7&7F   "=GGI!!!"g&7&7F   "=GGI!}}   sDAED:/CAD:EAD77D::E EEEcJt5}tj|djt t j d}|r8tt jrt jj|t||cdddS#1swYyxYw)N r) rerrBrCr rrr rrfrb)rrEras rrnrnsd ^rR''(9#))(DiPBszz*   $  $^^s BBB")r!zt.BinaryIO | t.TextIOrz%t.TypeGuard[_BufferedTextPagerStream]rW)r7rrz9t.ContextManager[tuple[t.BinaryIO | t.TextIO, str, bool]])r7rrz!t.Generator[t.TextIO, None, None])rz list[str]r7rrz3t.Iterator[tuple[t.BinaryIO | t.TextIO, str, bool]])r!zt.TextIOr7rrz&t.Iterator[tuple[t.TextIO, str, bool]])FF)rMrrrrYrrr)rarrr)rzcabc.Iterator[int])rrrr)>__doc__ __future__rcollections.abcabccabc contextlibiorrrrrEtypingr>rrrpathlibrtypesr_compatr r r r r rr exceptionsrutilsrTypeVarrProtocolrr"rrrgGenericr' TextIOWrapperrrcontextmanagerrrrrrrr[rbrrjrernrqrsrrrrs  #  )&& AIIcNqzz7 !7*777d?JIJII#!))A,I#X #R%%#' '>'> . /3dd!,d8ddN /3//!,/8//d    +/ .  .( .+ . .qqhCL<<7,^4r