W? @ s d Z d d l Z d d l Z d d l Z d d l Z d d l m Z m Z m Z m Z m
Z
m Z m Z m
Z
m Z d d l m Z d d l m Z e j d k r d Z d Z n d
Z d Z d d
Z Gd d d e Z d d d Z d d Z d d Z d d Z Gd d d e Z d d d d Z d d Z e rjd d l Z d d Z! n$ d d l" Z" d d l# Z# d! d Z! d S)"aY
click._termui_impl
~~~~~~~~~~~~~~~~~~
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.
:copyright: (c) 2014 by Armin Ronacher.
:license: BSD, see LICENSE for more details.
N ) _default_text_stdout
range_typePY2isatty open_stream
strip_ansiterm_lenget_best_encodingWIN)echo)ClickExceptionnt
z
[?25lz[?25h
c C s y t | SWn t k
r y t | j } Wn t k
rI d SYn Xy | | } Wn t k
rr d SYn X| t k s t | t t f s | d k r d S| SYn Xd S)z%Returns the length hint of an object.Nr ) len TypeErrortype__length_hint__AttributeErrorNotImplemented
isinstanceintlong)objZget_hintZhint r /_termui_impl.py _length_hint s
r c @ s$ e Z d Z d d d d d d d d d d d d d d d
Z d d Z d
d Z d d Z d d Z e d d Z e d d Z
e d d Z d d Z d d Z
d d Z d d Z d! d" Z d# d$ Z d% d& Z d' d( Z d) d* Z e s e Z [ d S)+ ProgressBarN# z%(bar)sz TF c C sz | | _ | | _ | | _ | | _ | | _ | | _ | | _ |
| _ | pQ d | _ | d k rl t } | | _
|
| _ | | _ | d k | _
| d k r t | } | d k r | d k r t d t | } t | | _ | | _ | d k | _ d | _ g | _ t j | _ | _ d | _ d | _ d | _ d | _ d | _ t | j
| _ d | _ d S)N r z iterable or length is requiredF)! fill_char
empty_char bar_templateinfo_sepshow_eta show_percentshow_positem_show_funclabelr filecolorwidth autowidthr r r iterlength length_knownposavgtimestartlast_eta eta_knownfinished max_widthentered current_itemr is_hidden
_last_line)selfiterabler1 r# r$ r% r& r' r( r) r* r+ r, r- r. r r r __init__4 sD zProgressBar.__init__c C s d | _ | j | S)NT)r; render_progress)r? r r r __enter__\ s
zProgressBar.__enter__c C s | j d S)N)
render_finish)r? exc_type exc_valuetbr r r __exit__a s zProgressBar.__exit__c C s# | j s t d | j | S)Nz.You need to use progress bars in a with block.)r; RuntimeErrorrB )r? r r r __iter__d s
zProgressBar.__iter__c C s. | j r
d S| j j t | j j d S)N)r= r, write AFTER_BARflush)r? r r r rD j s zProgressBar.render_finishc C s0 | j r
d St | j t | j p( d d S)Ng ?r )r9 minr3 floatr1 )r? r r r pctp s zProgressBar.pctc C s- | j s
d St | j t t | j S)Ng )r4 sumrO r )r? r r r time_per_iterationv s z ProgressBar.time_per_iterationc C s, | j r( | j r( | j | j | j Sd S)Ng )r2 r9 rR r1 r3 )r? r r r eta| s zProgressBar.etac C s | j r | j d } | d } | d } | d } | d } | d } | d } | d k rx | } d | | | | f Sd | | | f Sd S)Nr < r z%dd %02d:%02d:%02dz%02d:%02d:%02dr" )r8 rS )r? tZsecondsZminutesZhoursZdaysr r r
format_eta s
zProgressBar.format_etac C s- t | j } | j r) | d | j 7} | S)Nz/%s)strr3 r2 r1 )r? r3 r r r
format_pos s zProgressBar.format_posc C s d t | j d d d S)Nz% 4d%%d r )r rP )r? r r r
format_pct s zProgressBar.format_pctc C s | j } g } | j rl t | j | j } | j | } | | j | j | 7} | d k r | j } n | j r | j | j } no t | j | j p d } | j
d k r | j | t t j | j
| j
d d | j <d j | } | j r| j | j | r,| j | j | j r[| j r[| j r[| j | j | j d k r| j | j } | d k r| j | | j d | j d | d | j j | i j S) Nr r g @g ?r" r+ barinfo)r( r2 r rP r. r# r$ r) r9 listrR mathcosr3 joinappendrY r[ r' r8 rW r* r< r% r+ r& rstrip)r? r( Z info_bitsZ
bar_lengthr\ Z item_infor r r format_progress_line s8
5
z ProgressBar.format_progress_linec
C s d d l m } d } | j r4 | j g } d } ng } | j r | j } d | _ t | j } t d | d | } | | k r | j t
| j d | j | | _ | | _ | j } | j d k r | j } | j t
| j } t | } | j d k s$| j | k r-| | _ | j | | j d | | d j | } | | j
k r| | _
t | d | j d | j d
| | j j d S) Nr )get_terminal_sizeFTr r r" r, r- nl)Ztermuire r= r+ r/ r. r rd maxrb
BEFORE_BARr: ra r> r r, r- rM )
r? re rf bufZ old_widthZclutter_lengthZ new_widthZ clear_widthlineZline_lenr r r rB s>
"zProgressBar.render_progressc C s | j | 7_ | j r3 | j | j k r3 d | _ t j | j d k rP d St j | _ | j d d | j t j | j g | _ | j | _ d S)NTg ? i) r3 r2 r1 r9 r5 r7 r4 r6 r8 )r? n_stepsr r r make_step s 2zProgressBar.make_stepc C s | j | | j d S)N)rm rB )r? rl r r r update s
zProgressBar.updatec C s d | _ d | _ d | _ d S)Nr T)r8 r< r9 )r? r r r finish s zProgressBar.finishc C sy | j r t | j Sy t | j } | | _ Wn/ t k
rc | j | j t Yn X| j d | Sd S)Nr )r= nextr0 r<
StopIterationro rB rn )r? rvr r r rp s
zProgressBar.next)__name__
__module__ __qualname__rA rC rH rJ rD propertyrP rR rS rW rY r[ rd rB rm rn ro rp r __next__r r r r r 2 s. % % )
r c C s t } t t j s& t | r6 t | | | St j j d d pN d j } | r t rs t
| | | St | | | St j j d d k r t | | | St s t j j
d r t
| d | St t d rt j d
d k rt | d | Sd d l } | j \ } } t j | zL t t d rmt j d
| d k rmt | d | St | | | SWd t j | Xd S)z2Decide what method to use for paging through text.ZPAGERNr" ZTERMdumbemacsZos2zmore <systemz(less) 2>/dev/nullr lessz more "%s"Zmore)rx ry )r r sysstdin
_nullpagerosenvirongetstripr _tempfilepager
_pipepagerplatform
startswithhasattrrz tempfilemkstempcloseunlink)textr- stdoutZ pager_cmdr fdfilenamer r r pager s, !$
(r c C sr d d l } t t j } | j d d d j } | d k r | d d k r t j j d d d j | d d } | s d | d <d
} n d | k s d | k r d
} | s t | } | j | d
d
d | j
d | } t | j } y- | j j
| j | d | j j Wn t t f k
rBYn Xx( y | j Wn t k
rhYqFXPqFWd S)zqPage through text by feeding it to another program. Invoking a
pager through this might support colors.
r N/r r{ LESSr" r z-RTrRshellr} envreplace)
subprocessdictr r rsplitsplitr ra r PopenPIPEr
r} rK encoder IOErrorKeyboardInterruptwait) r cmdr- r r Z
cmd_detailZ
less_flagscencodingr r r r . s4 ,
r c C s d d l } | j } | s* t | } t t j } t | d d } | j | j | Wd QRXz t j
| d | d Wd t j | Xd S)z<Page through text by invoking a program on a temporary file.r Nwbz "") r mktempr r
r| r r rK r r rz r )r r r- r r r fr r r r ] s r c C s# | s t | } | j | d S)z>Simply print unformatted text. This is the ultimate fallback.N)r rK )streamr r- r r r r~ l s r~ c @ sL e Z d Z d d d d d d Z d d Z d d Z d
d Z d S) EditorNTz.txtc C s( | | _ | | _ | | _ | | _ d S)N)editorr require_save extension)r? r r r r r r r rA u s zEditor.__init__c C s | j d k r | j Sx* d D]" } t j j | } | r | Sq Wt rM d Sx+ d
D]# } t j d | d k rT | SqT Wd S) NVISUALEDITORZnotepadvimnanozwhich %s >/dev/null 2>&1r Zvi)r r )r r )r r r r r rz )r? keyrr r r r r
get_editor| s
zEditor.get_editorc C s d d l } | j } | j rC t j j } | j | j n d } yQ | j d | | f d | d d } | j } | d k r t d | Wn; t
k
r } z t d | | f WYd d } ~ Xn Xd S)Nr z%s "%s"r r Tz%s: Editing failed!z%s: Editing failed: %s) r r r r r copyrn r r r
OSError)r? r r r r r Z exit_codeer r r edit_file s zEditor.edit_filec C s[ d d l } | p d } | r8 | j d r8 | d 7} | j d d d | j \ } } z t r} d } | j d d } n d } | j | } t j | d
} | j | | j
t j j | } | j
| | j r t j j | | k r d St | d } z | j } Wd | j
X| j d j d d SWd t j | Xd S) Nr r" r prefixzeditor-suffixz utf-8-sigz
zutf-8r rb)r endswithr r r r r r fdopenrK r pathgetmtimer r openreaddecoder ) r? r r r namer r timestamprr r r r edit s2
!
z Editor.edit)rs rt ru rA r r r r r r r r s s
r Fc C s d d l } d d } t j d k r d g } | rC | j d | rV | j d | j | | t d d } z | j | d
| j SWd | j Xnl t r| r | | } d | | j d d
} n( d | r d p d
| j d d
f } t
j | Sy` | r<t
j j
| | p6d } n | | } | j d | g } | rm| j Sd SWnP t k
r| j d r| r| rd d l } | j | d Sd SYn Xd S)Nr c S s] y d d l } Wn t k
r0 d d l } Yn X| j d rY | j | d d } | S)Nr zfile:// )urllib ImportErrorr unquote)urlr r r r
_unquote_file s
z open_url.<locals>._unquote_filedarwinr z-Wz-Rz /dev/nullwstderrzexplorer /select,"%s"r r" zstart %s "" "%s"z/WAIT.zxdg-openhttp://https://r )r r )r r| r rb r r r r r r r rz r dirnamer r
webbrowser) r r Zlocater r argsZnullr r r r r open_url sF
%
!
r c C s. | d k r t | d k r* t d S)N)r EOFError)chr r r _translate_ch_to_exc s r c C s{ t j } | r t j | t | t rw t t j d d } | d k re | j | d } n | j d d } | S)Nr r cp1252) msvcrtZgetchZputcharr r getattrr| r} r )r rr encr r r getchar s
r c C s t t j s* t d } | j } n t j j } d } y t j | } zH t j | t j
| d } | r t t j r t j j | Wd t j
| t j | t j j | d k r | j XWn t j k
r Yn Xt | | j t t j d S)Nz/dev/tty r )r r| r} r filenotermiosZ tcgetattrttyZsetrawr r r rK Z tcsetattrZ TCSADRAINrM r errorr r r
)r r r Z old_settingsr r r r r
s(
)$__doc__r r| r5 r_ Z_compatr r r r r r r r
r Zutilsr
exceptionsr
r rh rL r objectr r r r r~ r r r r r r r r r r r <module> s6 @ / M3