ijttdZddlmZddlmZddlZddlZddlZddl 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- d. d/dZ.ej^d.d0d Z0ej^ d. d1d!Z1ej^ d. d1d"Z2ej^ d. d2d#Z3Gd$d%Z4d3d4d&Z5d5d'Z6e jnd(k(rddl8Z8ej^d6d)Z9d7d*Z:yddl;Z;ddl , 2.F|  ~ @AAvv0%-@H&*8n   "  %)YY[0 T]$# %)" &*tyy) &*rc4d|_|j|S)NT)rJrender_progressrNs r __enter__zProgressBar.__enter__s   rc$|jyN) render_finish)rNexc_type exc_valuetbs r__exit__zProgressBar.__exit__s rcp|js td|j|jS)N.You need to use progress bars in a with block.)rJ RuntimeErrorrR generatorrSs r__iter__zProgressBar.__iter__s.||OP P ~~rc*tt|SrV)nextr@rSs r__next__zProgressBar.__next__s DJrc|js |jsy|jjt|jj yrV)r/rLr5write AFTER_BARflushrSs rrWzProgressBar.render_finishs0 ;;dmm   " rc|jryt|jt|jxsdxsdz dS)N?r)rHminrBfloatrArSs rpctzProgressBar.pcts5 ==488uT[[%5A6;!>DHH A"fG "HA"fG "HAFE "HA1u<CCf D 1WRL'">>rclt|j}|j|d|jz }|S)N/)strrBrA)rNrBs r format_poszProgressBar.format_poss3$((m ;; " Qt{{m$ $C rc@t|jdzddddS)Ndz 4%r)r~rlrSs 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() rAr~rlr9r+r,rHlistrqmathcosrBjoin)rN 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)r4rinfo)r1rAr2appendrrr0rGrHrr3rKr-r4rr.rrstrip)rNr1 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)r5r6r r(F)r5r6nl)r/rLrMr4rr5r6r:shutilr9rrmaxget_terminal_sizecolumnsrIr BEFORE_BARrrg) rNbufr old_widthclutter_length new_width clear_widthlineline_lens rrRzProgressBar.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)NTrii)rBrArHrDrFrErCrG)rNn_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)rKr8r7rrR)rNrrKs rupdatezProgressBar.updateCsb  # ,D  !!W,!  $ $(=(= = NN444 5  "()D % >rc.d|_d|_d|_y)NFT)rGrKrHrSs 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) rJr^rLr@rKr8rRrr)rNrvs rr_zProgressBar.generatorbs||OP P}}yy ii$&! ,,1((* A  KKM  " !s3B*B(A3B*)N#rz%(bar)sz FTNFNNNNr)"rOzcabc.Iterable[V] | NonerAz int | Noner+rr,rr-rr.rr/boolr0rr1 bool | Noner2rr3z)t.Callable[[V | None], str | None] | Noner4 str | Noner5zt.TextIO | Noner6rr7r~r9r~returnNone)rzProgressBar[V])rXztype[BaseException] | NonerYzBaseException | NonerZzTracebackType | Nonerr)rzcabc.Iterator[V])rr)rr)rrkrr)rr~rrrV)rr~rKzV | Nonerr)rrrrPrTr[r`rcrWpropertyrlrqrsrrrrrrRrrrr_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|||_yrV)superrPr6)rNr r6kwargs __class__s rrPzMaybeStripAnsi.__init__s *6* rcP|js t|}t| |SrV)r6rrre)rNtextrs rrezMaybeStripAnsi.writes#zzd#Dw}T""r)r z t.IO[bytes]r6rrzt.Any)rrrr~)rrrrPre __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)r6stdoutpager_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. )r6)r6encodingN)rr!rr isinstancer=BinaryIOrg)r6r 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_partsr6rrcmd 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>rrgrcallrrrunlink) rr6rrrrrrrrfs rrr.s %'58:gu$$ A,C<<$L %'58:gu$$ L!**,H ,H } ##e# E">!EE"c#8Kt|}|d}|||fyw)z>Simply print unformatted text. This is the ultimate fallback.NFr )r r6rs rrras) !(H } (E !!sceZdZ d d dZd dZd dZejd dZejd dZddZy)EditorNc<||_||_||_||_yrV)editorr require_save extension)rNrrrrs rrPzEditor.__init__ms! ("rc|j |jSdD]'}tjj|}|s%|cStryddlm}dD]}|| |cSy)N)VISUALEDITORnotepadr)r)zsensible-editorvimnanovi)rrrrrrr)rNkeyrrrs r get_editorzEditor.get_editorysc ;; ";; %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)rrrrrrcopyrrrrrrrrOSError) rN filenamesrrrrr exit_coders r edit_fileszEditor.edit_filess")- 88jjoo'G NN488 $    U[[(4 ?:!AIA~$01888G  129919M  s!A(C C>&C99C>cyrVrrNrs reditz Editor.edits=@rcyrVrr%s rr&z 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)rrbytes bytearrayendswithrrencodemkstemprrfdopenreutimepathgetatimegetmtimer#rr openreaddecode) rNrrdatafdrr  timestamprs rr&z 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=B._unquote_files&( >>) $#ab'"C rdarwinr7z-Wrz /dev/nullw)stderrexplorerz/select,cygstartz-w.zxdg-open)zhttp://zhttps://r)rCrrr)rrrrr7rrrrr r r startfiler r4dirnamer webbrowser) rCrlocaterrDrnullrrNs ropen_urlrQs ||xx  KK   KK  M#&'K% ##D#6;;= JJL $C(3% 01D !t,,  S!  $C 45D>1 26$  OOC  sU) F 2F5G4GAG" F25 GG GG GG"3HHcx|dk(r t|dk(rts t|dk(rtr tyy)N)rrEOFError)chs r_translate_ch_to_excrX&s; V|!! V|Cj V|j|rwin32c#Kdyw)Nr*rrrr raw_terminalr[4s sc8|rCallablermsvcrtgetwchegetwchrX)rfuncrs rgetcharrd8ss> 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 rrr7filenotermios tcgetattrttysetraw tcsetattr TCSADRAINrrgrerror)r r; old_settingss rr[r[js 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) r[rr8r9r rrr rrerX)rr;rWs rrdrdsd ^rR''(9#))(DiPBszz*   $  $^^s BBB")r zt.BinaryIO | t.TextIOrz%t.TypeGuard[_BufferedTextPagerStream]rV)r6rrz9t.ContextManager[tuple[t.BinaryIO | t.TextIO, str, bool]])r6rrz!t.Generator[t.TextIO, None, None])rz list[str]r6rrz3t.Iterator[tuple[t.BinaryIO | t.TextIO, str, bool]])r zt.TextIOr6rrz&t.Iterator[tuple[t.TextIO, str, bool]])FF)rCrrrrOrrr~)rWrrr)rzcabc.Iterator[int])rrrr)=__doc__ __future__rcollections.abcabccabc contextlibiorrrrrDtypingr=rrrpathlibrtypesr_compatr r r r rrr exceptionsrutilsrTypeVarrProtocolrr!rrrfGenericr& TextIOWrapperrrcontextmanagerrrrrrrQrXrr`r[rdrgrirrrrs #  )&& AIIcNqzz7 !7*777d?JIJII#!))A,I#X #R%%#' '>'> . /3dd!,d8ddN /3//!,/8//d +/" "("+""qqhCL<<7,^4r