��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_type�PY2�isatty� open_stream�
strip_ansi�term_len�get_best_encoding�WIN)�echo)�ClickException�nt�
�
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� TypeError�type�__length_hint__�AttributeError�NotImplemented�
isinstance�int�long)�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_template�info_sep�show_eta� show_percent�show_pos�item_show_func�labelr �file�color�width� autowidthr r r �iter�length� length_known�pos�avg�time�start�last_eta� eta_known�finished� max_width�entered� current_itemr � is_hidden�
_last_line)�self�iterabler1 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_value�tbr 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_BAR�flush)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+ �bar�info)r( r2 r rP r. r# r$ r) r9 �listrR �math�cosr3 �join�appendrY 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_width�lineZline_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" ZTERM�dumb�emacsZos2zmore <�systemz(less) 2>/dev/nullr �lessz more "%s"Zmore)rx ry )r r �sys�stdin�
_nullpager�os�environ�get�stripr �_tempfilepager�
_pipepager�platform�
startswith�hasattrrz �tempfile�mkstemp�close�unlink)�textr- �stdoutZ pager_cmdr� �fd�filenamer 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-RT�r�R�shellr} �env�replace���)�
subprocess�dictr r� �rsplit�splitr� ra r �Popen�PIPEr
r} rK �encoder� �IOError�KeyboardInterrupt�wait) r� �cmdr- r� r� Z
cmd_detailZ
less_flags�c�encodingr 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 N�wbz "�") 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) N�VISUAL�EDITORZnotepad�vim�nanozwhich %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_code�er 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� �path�getmtimer� r� �open�read�decoder� ) 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�| r�d 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_file�darwinr� z-Wz-Rz /dev/null�w�stderrzexplorer /select,"%s"r� r" zstart %s "" "%s"z/WAIT�.zxdg-open�http://�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� �fileno�termiosZ tcgetattr�ttyZsetrawr 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