M@cD PddlZddlZddlZddlZddlZddlmZ ddlZddlZddlZdZ ddl m Z m Z GddeZGd d ZGd d ZGd dZeeZeeZeeZdZdZdZGddeZGddeZdZdZdZdZd8dZd9dZ dZ!d:dZ"ejFdfdZ$ddejFfdZ%e&d k(rejOdd!Z(d"e)eDD]\Z*Z+ejFjYe+e*d#ejZd$e+d$d%e)eDD]f\Z.Z/ejFjYe/d&e)eDD]4\Z0Z1ejFjYe1e0d$e2ed'e0zd$e+6hejFjYd(ejfzejFjYd)eejhd*z5Z5e6d+D]Z.ejnd,e8e5 dddeejrd-zejZzd.5Z5e6d+D]Z.ejnd,e8e5 ddded/dd05Z5e6d+D]Z.e8e5ejnd, ddded1dd235Z5e6d+D]Z.e8e5ejnd, ddde6d4Dcgc]}|c}Z:ee;e:5Z<e=e:D]$Z.e.code_to_charss>>CI-3 3_RESET_%sRESETz%s)dir startswithgetattrsetattrhasattrlowerr_mkfunc)selfcodesrnamevalue reset_name reset_values r__init__zAnsiCodes.__init__s 4J D??3't,dM%$89$.#4g">Juj1&-eZ&@ dTZZ\&9 ) 1 1-2F2? 2L!NO  rcfd}|S)Nctjs|Sd}|D]}||z } ||zzz }|r|tjz }|SN)r _USE_ANSIStyle RESET_ALL)textstylessstcolorresets r_cwrapz!AnsiCodes._mkfunc.._cwrap+sW&& A R  % %AU__$Hr)r.r/r0s`` rrzAnsiCodes._mkfunc)s  rctt| |}t|dr0|j tj rtj sy|S)Nrr&)superr__getattribute__rrr r')rrattr __class__s rr4zAnsiCodes.__getattribute__8s?Y6t< D, ' /''Krc6t||jSN)rupper)rrs r __getitem__zAnsiCodes.__getitem__AstTZZ\**rc6|xr|j|_yr8)_term_supports_colorr')cls allow_colorss rinitzAnsiCodes.initDs$C)A)A)C rctjjddk(s"tjjddk(ryttj dxrtj j S)NTERMdumbOS Windows_NTFisatty)osenvirongetrsysstdoutrEr1rrr<zAnsiCodes._term_supports_colorHsR JJNN6 "f , t$ 4szz8,D1B1B1DDr)__name__ __module__ __qualname__r'r r" staticmethodrr4r: classmethodr?r< __classcell__)r6s@rrrs\I D$  +DDEErrc0eZdZdZdZdZdZdZdZdZ dZ d Z y ) AnsiFore !"#$%'N rKrLrMGREYREDGREENYELLOWBLUEMAGENTACYANWHITErr1rrrRrRP/DCEFDGDEErrRc0eZdZdZdZdZdZdZdZdZ dZ d Z y ) AnsiBack()*+,-./1Nr\r1rrrgrg\rerrgcPeZdZdZdZdZdZeZdZdZ dZ dZ d Z d Z d Zd Zd ZdZdZdZy) AnsiStylerr N)rKrLrMr)BRIGHT RESET_BRIGHTDIM RESET_DIMITALICS RESET_ITALICS UNDERLINERESET_UNDERLINE BLINK_SLOWRESET_BLINK_SLOW BLINK_FASTRESET_BLINK_FASTINVERSE RESET_INVERSE STRIKE_THRURESET_STRIKE_THRUr1rrrrrrhsZIFLC$IGMIOJJGMKrrrc"tjSr8)Forer^r1rr ERROR_COLORrs 88Orc"tjSr8)rr`r1rr WARNING_COLORrs ;;rc"tjSr8)rr_r1rr HEADER_COLORrs ::rc<eZdZdZdZdZ d dZdZdZdZ d Z y) Spinnerz A class to display a spinner in the terminal. It is designed to be used with the `with` statement:: with Spinner("Reticulating splines", "green") as s: for item in enumerate(items): s.next() u ◓◑◒◐z-/|\Nc||_|xstj|_||_|s|r |j }n |j }||_|jj |_ ||_ yr8) _msgrIrJ_file_step_default_unicode_chars_default_ascii_chars_charsrE_silent _print_done)rmsgfilestepchars use_unicode print_dones rr"zSpinner.__init__sd 'SZZ  3311 ::,,.. %rc#XK|j}d}|jj}|jj} |d||j|d||||dt |j D]}d|dz }|t|k(rd}lw)Nr  r)rrwriteflushrrangerlen)rrindexrrrs r _iteratorzSpinner._iterators        $K $))  #J %,  G 4::&   QJEE "sB(B*cZ|jr|jS|jSr8)r_silent_iteratorrrs r __enter__zSpinner.__enter__s% <<((* *>># #rcJ|jj}|jj}|js|d||j|j r6||t jdzn"|t jdzn|d|y)Nrz [Done] z [Failed]  ) rrrrrrrr_r^)rexc_type exc_value tracebackrrs r__exit__zSpinner.__exit__sw      || $K $))    djj;./dhh./ $K rc#K|jj|j|jj dwr8)rrrrrs rrzSpinner._silent_iterators7 #  sAA)NrNTT) rKrLrM__doc__rrr"rrrrr1rrrrs4,",-:>& *$  rrceZdZdZddeeejffdZddZ dZ dZ dZ d Z dd Zd Zdd Zedd Zy) ProgressBaraP A class to display a progress bar in the terminal. It is designed to be used either with the `with` statement:: with ProgressBar(len(items)) as bar: for item in enumerate(items): bar.update() or as a generator:: for item in ProgressBar(items): item.process() Ntotal_or_itemsc|xstj|_|jjs|j|_d|_nd|_ t||_t||_ tj|_t xr|jj|_|j%|j"r6t'j&t&j(|j$d|_nd|_|j dy#t$rQ t||_ tt|j|_n#t$r tdwxYwYwxYw)a total_or_items : int or sequence If an int, the number of increments in the process being tracked. If a sequence, the items to iterate over. file : writable file-like object, optional The file to write the progress bar to. Defaults to `sys.stdout`. If `file` is not a tty (as determined by calling its `isatty` member, if any), the scrollbar will be completely silent. TFz&First argument must be int or sequencerN)rIrJrrE_silent_updateupdateriter_itemsr_total TypeErrorintrtime _start_time_CAN_RESIZE_TERMINAL_should_handle_resize_handle_resizesignalSIGWINCH _signal_set)rrrs rr"zProgressBar.__init__s0'SZZ zz  "--DKDL DL J~.DKn-DK 99; ! 8TZZ%6%6%8 "   % % MM&//4+>+> ?#D $D  A% J J!.1 "5#56  J HII J Js* D E:*3EE:E33E:9E:cTt|j|jd|_y)Nr) getTtySizerr_terminal_width)rsignumframes rrzProgressBar._handle_resizes()$***.*D*D FFG Irc|Sr8r1rs rrzProgressBar.__enter__ rc@|js||j|j|jj d|jj |j r3tjtjtjyyyNr) rrrrrrrrrSIG_DFL)rrrrs rrzProgressBar.__exit__sj|| DKK( JJ  T " JJ    foov~~> rc|Sr8r1rs r__iter__zProgressBar.__iter__'rrc t|j}|j|S#t$r|j dddwxYwr8)nextrr StopIterationr)rrvs r__next__zProgressBar.__next__*sI dkk"B KKMI   MM$d +  s )Ac||jdzx}|_n||_|jdk(rd}n!t|t|jz }|j}|j}|j ||}|j dz |_tt|j|z}|d|tjd|zztjz||jkrE|tjdztjz|d |j|z dz z|d |||jjy) zq Update the progress bar to the given value (out of the total given to the constructor). Nrr?rZz |=>-|)_current_valuerfloatrr _formatSuffixr _bar_lengthrrrarr_r)rrfracrrsuffixbar_fills rrzProgressBar.update4s* =*.*=*=*A AED'"'D  ;;! D<% "44Dzz ##E40//"4uT--.56 e  dii#.(4::56 d&& & $**s"TZZ/ 0 #))H4q89 : c  f  rc||jk\r$tj|jz }d}n6|dkrd}d}n,tj|jz d|z z|z }d}||t|dz }dj t |dt |jd}|d j d j |d zz }||z }|S) Nz rr&rz ETA T)shortz {0:>4s}/{1:>4s}z ({0:>6s}%)z{0:.2f}gY@)rrrrformatr)rrrttime_strrs rrzProgressBar._formatSuffixSs DKK  d...AH aZAH))+ 0 00S4Z@DHAH =  1D1 1H#**:e4+H+5dkk+NP-&&y'7'7u 'EFF( rcyr8r1)rrs rrzProgressBar._silent_updatehs rc $g}|tj}|t||5}td|j}tt t t||z d}|s6t|D]'\} } || | |zdk(s|j| )nZddl } | j} t| j|||D]'\} } |j| |j| )ddd|S#1swY|SxYw)as Does a `map` operation while displaying a progress bar with percentage complete. :: def work(i): print(i) ProgressBar.map(work, range(50)) Parameters: function : function Function to call for each step items : sequence Sequence where each element is a tuple of arguments to pass to *function*. multiprocess : bool, optional If `True`, use the `multiprocessing` module to distribute each task to a different processor core. file : writeable file-like object, optional The file to write the progress bar to. Defaults to `sys.stdout`. If `file` is not a tty (as determined by calling its `isatty` member, if any), the scrollbar will be completely silent. Nrrr) rIrJrmaxrrr enumeratermultiprocessingPoolimap_unorderedappend)r=functionitems multiprocessrresultsbar step_sizestepsiitemrpresults rmapzProgressBar.mapks@ <::D U$ ' +3C1IE#e*- 9:A>E(/&GAtTNE a' 1 & '#((*!*1+;+;He* 44rrc  tj}t|tsJ |j d|z|jy#t $r1|j dt|j dddzYJwxYw)N%s zutf-8replace)rIrJ isinstancer rUnicodeEncodeErrorencoder)r,fps rprintMsgrsr B a  F !HHJ F #ahhw :GDDEFsA 7BBcJt|ttjyr8)_printWithColorrrIstderrr,s r printErrorrsA{}cjj1rcJt|ttjyr8)rrrIrJrs r printWarningrsA} 3rcJt|ttjyr8)rrrIrJrs r printHeaderrsA|~szz2rct|d|S)NT)bc) formatText)r,r!s rboldTextr#s a41 %%rc|rtjnd|xsdz|z|rtjndz|rtjzSdzSr%)r(rrrr)r,r r!s rr"r"sYU\\B W"   TZZ" &$%U    ./+-  ./rct|tsJ|j||ztjzdz|j yr)rr rrrr)r,r.rs rrrs9 a  JJuqy4::%,-JJLrc|xsd}|xsd}dj|xsg}|s|s|r2tjtjztj znd}dt z}|S)zFormat ``msg`` with foreground and optional background. Optional ``styles`` lists will also be applied. The formatted string is returned.r&z&%(fg)s%(bg)s%(styles)s%(msg)s%(reset)s)joinrrBackr(r)locals)rfgbgr+r/outputs rcformatr-sb rB rB WWV\r "F:<fDJJ #eoo 5;= 6 @F Mrcd}|r< tj|tjd}t j d|}|sS ttjjdttjjdf}|S|S#t tf$rYhwxYw#ttf$rd}Y|SwxYw)NzhhLINESCOLUMNS)Nrz)fcntlioctltermios TIOCGWINSZstructunpackOSError NameErrorrrFrGrHr ValueError)fd check_ttyhwdatas rrrs B ;;r7#5#5x@DtT*B  bjjnnW-.bjjnnY/02B I2I#    :& B I s$;BAB-B*)B*-CCc8tt|||||y)zBCalls ``cformat`` and prints the result to output stream ``file``.)r+r+rN)printr-)rr*r+r+rs rcprintrBs '#rb 0trIs,A!"9Q< q!12A')7rc#VK|]!}t|r|tt|f#ywr8)rErrrGs rrIrIs,E%&y|wtQ/0ErJc#VK|]!}t|r|tt|f#ywr8)rErr(rGs rrIrIs,%J)*IaL'():%;%JrJrrrzPhase #12g?zPhase #2)rzPhase #3)rrzPhase #4u.oO°Oo.)rrriih[gMbP?r8r r )BrFr7rIrr rr3r5rr ImportErrorr&rrobjectrrRrgrrrr(r(rrrrrrrrrr#r"rr-rJrrBrKr?rErbg_namebg_coderrrfg_codest_namest_coderr)r_spinnerrsleeprr^rrrrrrprogressmax_val)xs0rr[sl !%:E:Ez    ()MfM`C&CL243&/ **"Dszz=  z NN4 A&)$iA 3 !7DJJA BE*-d)E ,JAw JJ  W %%J.1%j%J ,    ")7")%g1E"F")"+, , , %//12 3JJT j( )Wr A DJJsO M  J&3! #&-r A DJJsO M  5 AWr A M DJJsO  Z @Gr A M DJJsO  c #1Q #E SZ C5! A JJL DJJsO  +e$ % 3HG W   H JJx DJJt  oC! !v  $s_O ?(O(O$(O1 (O> P !0P 3P OOO!$O.1O;>PPP%