o +ca[ @sLdZddlZddlZddlZddlZddlZddlZddlmZ ddl m Z ddl m Z ddl m Z ddl mZdd l mZdd l mZdd l mZdd l mZdd lmZddlmZedZejdkrkdZdZndZdZGdddejeZd6dejedej e!ddfddZ"dejededej e!ddfddZ#dejededej e!ddfdd Z$d!ej%dejedej e!ddfd"d#Z&Gd$d%d%Z'd7d'ed(e!d)e!de(fd*d+Z)d,edej e*fd-d.Z+erddl,Z,ej-dej.e(fd/d0Z/d1e!defd2d3Z0dSddl1Z1ddl2Z2ej-dej.e(fd4d0Z/d1e!defd5d3Z0dS)8z This module contains implementations for the termui module. To keep the import time of Click down, some infrequently used functionality is placed in this module and only imported as needed. N)gettext)_default_text_stdout)CYGWIN)get_best_encoding)isatty) open_stream) strip_ansi)term_len)WIN)ClickException)echoVnt  z [?25lz[?25h c!@seZdZ              dBd ejejed ejed ed ededede deje de dejej ejegejefdejedejej deje dededdf ddZ dCddZ ddZdejefd d!Zdefd"d#ZdDd$d%Zedefd&d'Zedefd(d)Zedefd*d+Zdefd,d-Zdefd.d/Zdefd0d1Zdefd2d3Zdefd4d5ZdDd6d7Zd8eddfd9d:ZdEd8ed;ejeddfdd?Z dejefd@dAZ!dS)F ProgressBarN# %(bar)s TFriterablelength fill_char empty_char bar_templateinfo_sepshow_eta show_percentshow_positem_show_funclabelfilecolorupdate_min_stepswidthreturncCs*||_||_||_||_||_||_| |_| |_| pd|_| dur$t } | |_ | |_ ||_ d|_ ||_|dk|_|durMddlm}||d}|dkrMd}|durd|durYtdttjtt|}t||_||_d|_g|_t|_|_d|_d|_ d|_!d|_"d|_#t$|j  |_%d|_&dS)Nr) length_hintziterable or length is requiredF)'rrrrrrr r!r"rr#r$r%_completed_intervalsr& autowidthoperatorr) TypeErrortcastIterablerrangeiterrposavgtimestartlast_eta eta_knownfinished max_widthentered current_itemr is_hidden _last_line)selfrrrrrrrrr r!r"r#r$r%r&r)rA4/usr/lib/python3/dist-packages/click/_termui_impl.py__init__$sL      zProgressBar.__init__cCsd|_||S)NT)r<render_progressr@rArArB __enter__`szProgressBar.__enter__cCs |dSN) render_finish)r@exc_type exc_valuetbrArArB__exit__es zProgressBar.__exit__cCs|jstd||S)N.You need to use progress bars in a with block.)r< RuntimeErrorrD generatorrErArArB__iter__hszProgressBar.__iter__cCs tt|SrG)nextr3rErArArB__next__ns zProgressBar.__next__cCs$|jrdS|jt|jdSrG)r>r#write AFTER_BARflushrErArArBrHvs zProgressBar.render_finishcCs(|jrdSt|jt|jp dpddS)N?r)r:minr4floatrrErArArBpct|szProgressBar.pctcCs"|jsdSt|jtt|jSNg)r5sumrXlenrErArArBtime_per_iterationszProgressBar.time_per_iterationcCs&|jdur|js|j|j|jSdSrZ)rr:r]r4rErArArBetaszProgressBar.etacCs|jrAt|j}|d}|d}|d}|d}|d}|d}|dkr4|d|dd|dd|dS|dd|dd|dSdS)N<rzd 02:r()r9intr^)r@r/secondsminuteshoursrArArB format_etas  zProgressBar.format_etacCs(t|j}|jdur|d|j7}|S)N/)strr4r)r@r4rArArB format_poss  zProgressBar.format_poscCst|jdddddS)Ndz 4%r)rcrYrErArArB format_pctszProgressBar.format_pctcCs|jdurt|j|j}|j|}||j|j|7}|S|jr)|j|j}|St|j|jp0d}|jdkrM|j|tt |j |jdd|j<d |}|S)Nrrg@g?r() rrcrYr&rrr:listr]mathcosr4join)r@ bar_lengthbarcharsrArArB format_bars(       zProgressBar.format_barcCs|j}g}|jdur|dur|j }|jr|||r%|||jr5|jr5|js5|| |j durI| |j }|durI|||j |j ||j|dS)N)r"rsinfo)rrr appendrjrmrr9r:rgr!r=rr"rurrqrstrip)r@r info_bits item_inforArArBformat_progress_lines,    z ProgressBar.format_progress_linec CsDddl}|jr|j|jkr|j|_t|j|j|jddSg}|jrO|j}d|_t | }t d| j |}||krL|t|d|j||_||_|j}|jdurZ|j}|t| }t |}|jdusq|j|krt||_|||d||d|}||jkr||_t||j|jdd|jdSdS)Nr)r#r$rr(F)r#r$nl)shutilr>r?r"r r#r$r,r&r r{maxget_terminal_sizecolumnsrw BEFORE_BARr;rqrU) r@r}buf old_widthclutter_length new_width clear_widthlineline_lenrArArBrDsB        zProgressBar.render_progressn_stepscCs|j|7_|jdur|j|jkrd|_t|jdkr dSt|_|jr3t|j|j}nt|j}|jdd|g|_|jdu|_dS)NTrVi)r4rr:r6r8r7r5r9)r@rsteprArArB make_steps zProgressBar.make_stepr=cCsJ|dur||_|j|7_|j|jkr#||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=r+r%rrD)r@rr=rArArBupdates   zProgressBar.updatecCsd|_d|_d|_dS)NFT)r9r=r:rErArArBfinish1s zProgressBar.finishccsn|jstd|jr|jEdHdS|jD]}||_|jdkr$||V|dq||dS)zReturn a generator which yields the items added to the bar during construction, and updates the progress bar *after* the yielded block returns. rMNrr) r<rNr>r3r=r+rDrr)r@rvrArArBrO6s     zProgressBar.generator)NrrrrTNFNNNNrr)r'r)r'NrG)"__name__ __module__ __qualname__r/Optionalr1rrcriboolCallableTextIOrCrFrLIteratorrPrRrHpropertyrXrYr]r^rgrjrmrur{rDrrrrOrArArArBr#s      <  - rrOr$r'c Cs2t}ttjr t|st|||Stjddpd}|r-t r't |||St |||Stjddvr;t|||St sCtj drIt |d|Sttdr[td d kr[t |d |Sd dl}|\}}t|z(ttdrtd |d d krt |d|Wt|St|||Wt|St|w)z2Decide what method to use for paging through text.PAGERNr(TERM)dumbemacsos2zmore <systemz(less) 2>/dev/nullrlesszmore ""more)rrsysstdin _nullpagerosenvirongetstripr _tempfilepager _pipepagerplatform startswithhasattrrtempfilemkstempcloseunlink)rOr$stdout pager_cmdrfdfilenamerArArBpagerXs.            rcmdc Cs"ddl}ttj}|ddd}|durB|ddkrBtjddd |dd}|s8d |d<d }n d |vs@d |vrBd }|j|d |j |d}t t j |j }t|} z|D]} |sct| } || | dq[Wn ttfyyYnw| z|WdStyYnwq)zqPage through text by feeding it to another program. Invoking a pager through this might support colors. rNrhrr*rLESSr(r-RTrR)shellrenvreplace) subprocessdictrrrsplitsplitrrqPopenPIPEr/r0BinaryIOrrr rSencodeOSErrorKeyboardInterruptrwait) rOrr$rr cmd_detail less_flagscrencodingtextrArArBrus> $   rc Csddl}|\}}d|}|st|}ttj}t|dd}|| |Wdn1s4wYzt |d|dWt |t |dSt |t |w)zSimply print unformatted text. This is the ultimate fallback.N)r rS)rrOr$rrArArBrs  rc @seZdZ    ddejedejejeefdededdf d d Zdefd d Z d eddfddZ dejej dejej fddZ dS)EditorNT.txteditorr require_save extensionr'cCs||_||_||_||_dSrG)rrrr)r@rrrrrArArBrCs zEditor.__init__cCsf|jdur|jSdD]}tj|}|r|Sq trdSdD]}td|ddkr0|SqdS)N)VISUALEDITORnotepad)zsensible-editorvimnanozwhich z >/dev/null 2>&1rvi)rrrrr r)r@keyrrrArArB get_editors  zEditor.get_editorrc Csddl}|}d}|jrtj}||jz#|j|d|d|dd}|}|dkr9t t dj |dWdSt yS}z t t dj ||d |d}~ww) NrrrT)rrz{editor}: Editing failed)rz{editor}: Editing failed: {e})re) rrrrrcopyrrrr _formatr)r@rrrrr exit_coderrArArB edit_files,  zEditor.edit_filerc Csddl}|s d}n&t|ttfr|}n|r|ds|d7}tr*|ddd}n|d}|jd|j d\}}zt |d  }| |Wdn1sPwYt |t j|t j|d ft j|}|||jrt j||krWt |dSt|d  }|}Wdn1swYt|ttfr|Wt |S|dddWt |St |w) Nrrz z utf-8-sigzutf-8zeditor-)prefixsuffixrrb)r isinstancebytes bytearrayendswithr rrrrrfdopenrSutimepathgetatimegetmtimerrropenreaddecode) r@rrdatarnamer timestamprrArArBedits:  $    z Editor.edit)NNTr) rrrr/rriMappingrrCrrAnyStrrrArArArBrs&  &rFurlrlocatec Csddl}dtdtfdd}tjdkrCdg}|r|d|r#|d |||td d }z|j||d W|S|wt rp|rV|| d d}d|d }n| d d}|r`dnd}d|d|d }t |St r|rt j|| d d}d|d }n| d d}|rdnd}d|d|d }t |Sz"|rt j||pd}n||}|d|g}|r|WSWdSty|dr|s|sddl} | |YdSYdSw)Nrrr'cSs*ddlm}|dr||dd}|S)Nr)unquotezfile://) urllib.parserr)rrrArArB _unquote_file*s  zopen_url.._unquote_filedarwinrz-Wrz /dev/nullw)stderrrr(zexplorer /select,"z/WAITzstart z "" "z cygstart "z-wz cygstart r.zxdg-open)zhttp://zhttps://r)rrirrrwrrrrr rrrrrdirnamerr webbrowser) rrrrrargsnullwait_strrrrArArBopen_url'sX             r chcCs6|dkrt|dkrtst|dkrtrtdS)N)rr EOFError)r rArArB_translate_ch_to_exccs  rccs dVdS)Nr*rArArArArB raw_terminalss rr cCs6|rtj}ntj}|}|dvr||7}t||S)N)à)msvcrtgetwchegetwchr)r funcrrArArBgetcharws  rc csttjstd}|}ntj}d}zAt|}z$t||VWt |tj |tj |dur>| WdSWdSt |tj |tj |durW| wwtjybYdSw)Nz/dev/tty)rrrrfilenotermios tcgetattrttysetraw tcsetattr TCSADRAINrrUrerror)rr old_settingsrArArBrs0        cCsjt(}t|dttjd}|rttjrtj |t ||WdS1s.wYdS)N r) rrrrrrrrrrSr)r rr rArArBrs $rG)FF)3__doc__ contextlibrorrr6typingr/rr_compatrrrrrr r r exceptionsr utilsr TypeVarrrrrTGenericrr1rirrrrrrrrrcr  BaseExceptionrrcontextmanagerrrrrrrArArArBsx             $7&2   a< 1