U
��W�? � @ s d Z ddlZddlZddlZddlZddlmZmZmZm Z m
Z
m Z m Z m
Z
mZ ddlmZ ddlmZ ejdkr|dZd Znd
Zd Zd d
� ZG dd� de�Zd"dd�Zdd� Zdd� Zdd� Z G dd� de�Z d#dd �Z d d � Z er�ddl Z d d � Z!nddl"Z"ddl#Z#d!d � Z!dS )$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� z
t | �W S tk
r� zt| �j}W n tk
r@ Y Y dS X z || �}W n tk
rf Y Y dS X |tks�t|ttf�r�|dk r�Y dS | Y S X dS )z%Returns the length hint of an object.Nr ) �len� TypeError�type�__length_hint__�AttributeError�NotImplemented�
isinstance�intZlong)�objZget_hintZhint� r �/_termui_impl.py� _length_hint s$
��r c
@ s� e Zd Zd+d d
�Zd d � Zd
d� Zdd� Zdd� Zedd� �Z edd� �Z
edd� �Z dd� Z dd � Z
d d � Zd d � Zd!d"� Zd#d$� Zd%d&� Zd'd(� Zd)d*� Zes�eZ[dS ),� ProgressBarN�#� �%(bar)s� TF� c C s || _ || _|| _|| _|| _|| _| | _|
| _| p6d| _| d krHt � } | | _
|
| _ || _ |dk| _
|d krtt|�}|d kr�|d kr�td��t|�}t|�| _|| _|d k | _d| _g | _t�� | _| _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�iterabler2 r$ r% r& r' r( r) r* r+ r, r- r. r/ r r r �__init__4 sD
zProgressBar.__init__c C s d| _ | �� | S )NT)r< �render_progress�r@ r r r � __enter__\ s zProgressBar.__enter__c C s | � � d S �N)�
render_finish)r@ �exc_type� exc_value�tbr r r �__exit__a s zProgressBar.__exit__c C s | j std��| �� | S )Nz.You need to use progress bars in a with block.)r< � RuntimeErrorrC rD r r r �__iter__d s zProgressBar.__iter__c C s$ | j r
d S | j�t� | j�� d S rF )r> r- �write� AFTER_BAR�flushrD r r r rG j s zProgressBar.render_finishc C s$ | j r
dS t| jt| j�p d d�S )N� �?r )r: �minr4 �floatr2 rD r r r �pctp s zProgressBar.pctc C s" | j s
dS t| j �tt| j �� S �Ng )r5 �sumrS r rD r r r �time_per_iterationv s z ProgressBar.time_per_iterationc C s" | j r | js | j| j| j S dS rU )r3 r: rW r2 r4 rD r r r �eta| s zProgressBar.etac C sn | j rj| jd }|d }|d }|d }|d }|d }|d }|dkr\|}d||||f S d|||f S dS )Nr �< � r z%dd %02d:%02d:%02dz%02d:%02d:%02dr# )r9 rX )r@ �tZsecondsZminutesZhoursZdaysr r r �
format_eta� s
zProgressBar.format_etac C s" t | j�}| jr |d| j 7 }|S )Nz/%s)�strr4 r3 r2 )r@ r4 r r r �
format_pos� s
zProgressBar.format_posc C s dt | jd � dd � S )Nz% 4d%%�d r )r rT rD r r r �
format_pct� s zProgressBar.format_pctc C sH | j }g }| jrPt| j| j �}| j| }|| j| j| 7 }|d kr�| j }nf| jrd| j| j }nRt | j| jprd �}| j
dkr�| j|tt � | j
| j
�d d | j �< d�|�}| jr�|�| �� � |r�|�| �� � | jr�| jr�| js�|�| �� � | jd k �r(| �| j�}|d k �r(|�|� | j| j|| j�|�d� �� S )Nr r g @g �?r# )r, �bar�info)r) r3 r rT r/ r$ r% r* r: �listrW �mathZcosr4 �join�appendr^ r` r( r9 r\ r+ r= r& r, r' �rstrip)r@ r) Z info_bitsZ
bar_lengthra Z item_infor r r �format_progress_line� sH
����
�z ProgressBar.format_progress_linec
C s, ddl m} d}| jr$| jg}d}n�g }| jr�| j}d| _t| �� �}td|� d | �}||k r�|� t
� |� d| j � || _ || _| j}| j d k r�| j }|� t
� | �� }t|�} | j d ks�| j | k r�| | _ |� |� |� d|| � d� |�}|| j
k�r(|| _
t|| j| j|d� | j�� d S ) Nr )�get_terminal_sizeFTr r r# )r- r. �nl)Ztermuiri r> r, r0 r/ r rh �maxrf �
BEFORE_BARr; re r? r r- r. rP )
r@ ri rj ZbufZ old_widthZclutter_lengthZ new_widthZ clear_width�lineZline_lenr r r rC � s>
zProgressBar.render_progressc C sz | j |7 _ | jr&| j | jkr&d| _t�� | j dk r<d S t�� | _| jdd � | jt�� | j g | _| j| _d S )NTrQ i����) r4 r3 r2 r: r6 r8 r5 r7 r9 �r@ Zn_stepsr r r � make_step� s
(zProgressBar.make_stepc C s | � |� | �� d S rF )ro rC rn r r r �update� s
zProgressBar.updatec C s d| _ d | _d| _d S )Nr T)r9 r= r: rD r r r �finish� s zProgressBar.finishc C sb | j rt| j�S zt| j�}|| _W n* tk
rN | �� | �� t� �Y nX | �d� |S d S )Nr )r> �nextr1 r= �
StopIterationrq rC rp )r@ �rvr r r rr � s
zProgressBar.next)
Nr r r r! TNFNNNNr" )�__name__�
__module__� __qualname__rB rE rK rM rG �propertyrT rW rX r\ r^ r` rh rC ro rp rq rr r �__next__r r r r r 2 sD �
(
%)
r c C s& t � }ttj�rt|�s$t|| |�S tj�dd�p4d�� }|rZt rNt
| ||�S t | ||�S tj�d�dkrvt|| |�S t s�tj �
d�r�t
| d|�S ttd�r�t�d �d
kr�t | d |�S d
dl}|�� \}}t�|� z@ttd��rt�d | �d
k�rt | d
|�W �S t|| |�W �S t�|� X dS )z2Decide what method to use for paging through text.ZPAGERNr# ZTERM)ZdumbZemacsZos2zmore <�systemz(less) 2>/dev/nullr �lessz more "%s"Zmore)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 s ddl }ttj�}|�dd�d �� }|dkr�|d dkr�tj�dd�d �|dd� � }|sld
|d< d }nd |ks|d
|kr�d }|s�t| �} |j |d |j
|d�}t |j �}z"|j �
| �|d�� |j �� W n ttfk
r� Y nX z |�� W n tk
�r Y q�X �qq�dS )zqPage through text by feeding it to another program. Invoking a
pager through this might support colors.
r N�/r ���r{ �LESSr# r �-RT�r�R)�shellr} �env�replace)�
subprocess�dictr r� �rsplit�splitr� re r �Popen�PIPEr
r} rN �encoder� �IOError�KeyboardInterrupt�wait) r� �cmdr. r� r� Z
cmd_detailZ
less_flags�c�encodingr r r r� . s4
�
r� c C s| ddl }|�� }|s t| �} ttj�}t|d�d �}|�| �|�� W 5 Q R X zt � |d | d � W 5 t �
|� X dS )z<Page through text by invoking a program on a temporary file.r N�wbz "�") r� Zmktempr r
r| r� r rN r� r r� rz )r� r� r. r� r� r� �fr r r r� ] s
r� c C s |s t |�}| �|� dS )z>Simply print unformatted text. This is the ultimate fallback.N)r rN )�streamr� r. r r r r~ l s r~ c @ s. e Zd Zd dd�Zdd� Zdd � Zd
d � ZdS )
�EditorNT�.txtc C s || _ || _|| _|| _d S rF )�editorr� � require_save� extension)r@ r� r� r� r� r r r rB u s zEditor.__init__c C sb | j d k r| j S dD ] }tj�|�}|r| S qtr:dS dD ] }t�d| �dkr>| S q>dS )N)ZVISUALZEDITORZnotepad)ZvimZnanozwhich %s >/dev/null 2>&1r Zvi)r� r r� r� r rz )r@ �keyrt r� r r r �
get_editor| s
zEditor.get_editorc
C s� dd l }| �� }| jr.tj�� }|�| j� nd }z8|jd||f |dd�}|�� }|dkrht d| ��W n2 t
k
r� } zt d||f ��W 5 d }~X Y nX d S )Nr z%s "%s"T)r� r� z%s: Editing failed!z%s: Editing failed: %s) r� r� r� r r� �copyrp r� r� r
�OSError)r@ r� r� r� r� r� Z exit_code�er r r � edit_file� s
�zEditor.edit_filec C s� dd l }|pd}|r&|�d�s&|d7 }|jd| jd�\}}z�trRd}|�dd�}nd}|�|�}t� |d �}|�
|� |� � tj �
|�}| �|� | jr�tj �
|�|kr�W �<d S t|d
�}z |�� }W 5 |� � X |�d��dd�W �S t�|� X d S ) Nr r# r zeditor-)�prefix�suffixz utf-8-sigz
zutf-8r� �rb)r� �endswithr� r� r r� r r� r� �fdopenrN r� �path�getmtimer� r� �open�read�decode) r@ r� r� r� �namer� r� Z timestamprt r r r �edit� s4
�
z Editor.edit)NNTr� )ru rv rw rB r� r� r� r r r r r� s s �
r� Fc C sX dd l }dd� }tjdkrxdg}|r.|�d� |r<|�d� |�|| �� tdd �}z|j||d
��� W �S |�� X nLtr�|r�|| �} d || � d d
�� }n d|r�dp�d
| � d d
�f }t
� |�S zF|r�t
j �
|| ��p�d} n|| �} |�d| g�}|�r|�� W S W dS tk
�rR | �d��rL|�sL|�sLdd l}|�| � Y dS Y dS X d S )Nr c S sJ z dd l }W n tk
r( dd l }Y nX | �d�rF|�| dd � �} | S )Nr zfile://� )�urllib� ImportErrorr� Zunquote)�urlr� r r r �
_unquote_file� s
z open_url.<locals>._unquote_file�darwinr� z-Wr� z /dev/null�w)�stderrzexplorer /select,"%s"r� r# zstart %s "" "%s"z/WAIT�.zxdg-open)zhttp://zhttps://r )r� r| r� rf r� r� r� r� r r� r rz r� �dirnamer� r� �
webbrowser) r� r� Zlocater� r� �argsZnullr� r� r r r �open_url� sL
�
�
r� c C s | dkrt � �| dkr t� �d S )N��)r� �EOFError)�chr r r �_translate_ch_to_exc� s r� c C sV t �� }| rt �|� t|� trRttjdd �}|d k rF|�|d�}n |�dd�}|S )Nr� r� �cp1252) �msvcrtZgetchZputcharr� r �getattrr| r} r� )r rt �encr r r �getchar� s
r� c C s� t tj�s td�}|�� }ntj�� }d }zpt�|�}z4t �
|� t�|d�}| rht tj �rhtj �|� W 5 t�|tj|� tj �
� |d k r�|� � X W n tjk
r� Y nX t|� |�ttj�d�S )Nz/dev/tty� r� )r r| r} r� �fileno�termiosZ tcgetattrZ tcsetattrZ TCSADRAINr� rP r� �ttyZsetrawr r� rN �errorr� r� r
)r r� r� Z old_settingsr� r r r r�
s(
)N)FF)$�__doc__r r| r6 rd Z_compatr r r r r r r r
r Zutilsr �
exceptionsr
r� rl rO r �objectr r� r� r� r~ r� r� r� r� r� r� r� r r r r �<module> s8 ,
a
/M
3