U
e([VR � @ s d dl Z d dlZd dlZddlmZmZmZmZmZm Z m
Z
m Z ddl m
Z
ddlmZmZ ddlmZ ddlmZ eZdZd Zd
d � Zd2d
d�Zd3dd�Zd4dd�Zdd� Z d5dd�Z d6d d �Z d d!� Z d7d"d#�Z d$d%� Z!d8d&d'�Z"d9d)d*�Z#d:d+d,�Z$dZ%d;d-d.�Z&d<d0d1�Z'dS )=� N� )� raw_input� text_type� string_types�isatty�
strip_ansi�get_winterm_size�DEFAULT_COLUMNS�WIN)�echo)�Abort�
UsageError)� convert_type)�resolve_color_default) ZblackZredZgreenZyellowZblueZmagentaZcyanZwhite�resetz[0mc C s dd l }|� | �S )Nr )�getpass)�promptr � r �
/termui.py�hidden_prompt_func s r Fc C s$ | }|d k r |r d||f }|| S )Nz%s [%s]r )�text�suffix� show_default�defaultr r r r �
_build_prompt s r �: Tc
s� d} � �fdd�}
|dkr$t ||�}t| |||�} |
| �} | rBqPq2|dk r2|S q2z || �} W n< tk
r� }
z td|
j � d� W Y �q2W 5 d}
~
X Y nX |s�| S |
d�}|r�q�q�| |kr�| S td� d� q2dS )a� Prompts a user for input. This is a convenience function that can
be used to prompt a user for input later.
If the user aborts the input by sending a interrupt signal, this
function will catch it and raise a :exc:`Abort` exception.
.. versionadded:: 6.0
Added unicode support for cmd.exe on Windows.
.. versionadded:: 4.0
Added the `err` parameter.
:param text: the text to show for the prompt.
:param default: the default value to use if no input happens. If this
is not given it will prompt until it's aborted.
:param hide_input: if this is set to true then the input value will
be hidden.
:param confirmation_prompt: asks for confirmation for the value.
:param type: the type to use to check the value against.
:param value_proc: if this parameter is provided it's a function that
is invoked instead of the type conversion to
convert a value.
:param prompt_suffix: a suffix that should be added to the prompt.
:param show_default: shows or hides the default value in the prompt.
:param err: if set to true the file defaults to ``stderr`` instead of
``stdout``, the same as with echo.
Nc sX �rt p
t}zt| d� d� |d�W S ttfk
rR �rHtd � d� t� �Y nX d S )NF��nl�err� �r )r �visible_prompt_funcr �KeyboardInterrupt�EOFErrorr )r �f�r �
hide_inputr r � prompt_funcC s
zprompt.<locals>.prompt_funcz Error: %sr zRepeat for confirmation: z*Error: the two entered values do not match)r r r
r �message)r r r& Zconfirmation_prompt�typeZ
value_proc�
prompt_suffixr r �resultr' r �value�eZvalue2r r% r r " s.
r c C s� t | |||rdpd�}z"t|d|d� td��� �� }W n ttfk
rV t� �Y nX |dkrfd}q�|dkrtd}q�|dkr�|}q�td |d
� qq�q|r�|s�t� �|S ) a� Prompts for confirmation (yes/no question).
If the user aborts the input by sending a interrupt signal this
function will catch it and raise a :exc:`Abort` exception.
.. versionadded:: 4.0
Added the `err` parameter.
:param text: the question to ask.
:param default: the default for the prompt.
:param abort: if this is set to `True` a negative answer aborts the
exception by raising :exc:`Abort`.
:param prompt_suffix: a suffix that should be added to the prompt.
:param show_default: shows or hides the default value in the prompt.
:param err: if set to true the file defaults to ``stderr`` instead of
``stdout``, the same as with echo.
zY/nzy/NFr r )�yZyesT)�nZnozError: invalid inputr )r r r! �lower�stripr" r# r ) r r �abortr* r r r r, �rvr r r �confirmq s(
� r4 c C s� t jdkr4ddl} t| dd�}|r4|� }|j|jfS tdk rBt� S dd� }|d�p`|d�p`|d�}|s�z0t�t� � tj
�}z ||�}W 5 t� |� X W n t k
r� Y nX |r�|d r�|d s�tj
�d d
�tj
�d t�f}t|d �t|d �fS ) zoReturns the current size of the terminal as tuple in the form
``(width, height)`` in columns and rows.
)� r5 r N�get_terminal_sizec S sH z,dd l }dd l}t�d|�| |jd��}W n tk
rB Y d S X |S )Nr ZhhZ1234)�fcntl�termios�struct�unpackZioctlZ
TIOCGWINSZ� Exception)�fdr7 r8 �crr r r � ioctl_gwinsz� s �z'get_terminal_size.<locals>.ioctl_gwinszr � ZLINES� ZCOLUMNS)�sys� version_info�shutil�getattr�columns�linesr �os�open�ctermid�O_RDONLY�closer; �environ�getr �int)rC Zshutil_get_terminal_sizeZszr> r= r< r r r r6 � s.
�r6 c C s4 t |�}t| t�st| �} ddlm} || d |�S )a3 This function takes a text and shows it via an environment specific
pager on stdout.
.. versionchanged:: 3.0
Added the `color` flag.
:param text: the text to page.
:param color: controls if the pager supports ANSI colors or not. The
default is autodetection.
r )�pager�
)r �
isinstancer r � _termui_implrO )r �colorrO r r r �echo_via_pager� s
rT �#�-� %(label)s [%(bar)s] %(info)s� �$ c C s8 ddl m} t|
�}
|| |||||||| |
| || |
d�S )a� This function creates an iterable context manager that can be used
to iterate over something while showing a progress bar. It will
either iterate over the `iterable` or `length` items (that are counted
up). While iteration happens, this function will print a rendered
progress bar to the given `file` (defaults to stdout) and will attempt
to calculate remaining time and more. By default, this progress bar
will not be rendered if the file is not a terminal.
The context manager creates the progress bar. When the context
manager is entered the progress bar is already displayed. With every
iteration over the progress bar, the iterable passed to the bar is
advanced and the bar is updated. When the context manager exits,
a newline is printed and the progress bar is finalized on screen.
No printing must happen or the progress bar will be unintentionally
destroyed.
Example usage::
with progressbar(items) as bar:
for item in bar:
do_something_with(item)
Alternatively, if no iterable is specified, one can manually update the
progress bar through the `update()` method instead of directly
iterating over the progress bar. The update method accepts the number
of steps to increment the bar with::
with progressbar(length=chunks.total_bytes) as bar:
for chunk in chunks:
process_chunk(chunk)
bar.update(chunks.bytes)
.. versionadded:: 2.0
.. versionadded:: 4.0
Added the `color` parameter. Added a `update` method to the
progressbar object.
:param iterable: an iterable to iterate over. If not provided the length
is required.
:param length: the number of items to iterate over. By default the
progressbar will attempt to ask the iterator about its
length, which might or might not work. If an iterable is
also provided this parameter can be used to override the
length. If an iterable is not provided the progress bar
will iterate over a range of that length.
:param label: the label to show next to the progress bar.
:param show_eta: enables or disables the estimated time display. This is
automatically disabled if the length cannot be
determined.
:param show_percent: enables or disables the percentage display. The
default is `True` if the iterable has a length or
`False` if not.
:param show_pos: enables or disables the absolute position display. The
default is `False`.
:param item_show_func: a function called with the current item which
can return a string to show the current item
next to the progress bar. Note that the current
item can be `None`!
:param fill_char: the character to use to show the filled part of the
progress bar.
:param empty_char: the character to use to show the non-filled part of
the progress bar.
:param bar_template: the format string to use as template for the bar.
The parameters in it are ``label`` for the label,
``bar`` for the progress bar and ``info`` for the
info section.
:param info_sep: the separator between multiple info items (eta etc.)
:param width: the width of the progress bar in characters, 0 means full
terminal width
:param file: the file to write to. If this is not a terminal then
only the label is printed.
:param color: controls if the terminal supports ANSI colors or not. The
default is autodetection. This is only needed if ANSI
codes are included anywhere in the progress bar output
which is not the case by default.
r )� ProgressBar)�iterable�length�show_eta� show_percent�show_pos�item_show_func� fill_char�
empty_char� bar_template�info_sep�file�label�widthrS )rR rZ r )r[ r\ rf r] r^ r_ r` ra rb rc rd rg re rS rZ r r r � progressbar� s S �rh c C s. t tj�sdS tr t�d� n tj�d� dS )z�Clears the terminal screen. This will have the effect of clearing
the whole visible space of the terminal and moving the cursor to the
top left. This does not do anything if not connected to a terminal.
.. versionadded:: 2.0
N�clsz
[2J[1;1H)r rA �stdoutr
rG �system�writer r r r �clear5 s
rm c
C sH g } |rFz | � dt�|�d � W n tk
rD td| ��Y nX |r�z | � dt�|�d � W n tk
r� td| ��Y nX |dk r�| � d|r�dnd � |dk r�| � d|r�dnd � |dk r�| � d|r�d nd
� |dk �r| � d|r�d nd � |dk �r$| � d|�r d
nd � | � | � |�r>| � t� d�| �S )aH Styles a text with ANSI styles and returns the new string. By
default the styling is self contained which means that at the end
of the string a reset code is issued. This can be prevented by
passing ``reset=False``.
Examples::
click.echo(click.style('Hello World!', fg='green'))
click.echo(click.style('ATTENTION!', blink=True))
click.echo(click.style('Some things', reverse=True, fg='cyan'))
Supported color names:
* ``black`` (might be a gray)
* ``red``
* ``green``
* ``yellow`` (might be an orange)
* ``blue``
* ``magenta``
* ``cyan``
* ``white`` (might be light gray)
* ``reset`` (reset the color code only)
.. versionadded:: 2.0
:param text: the string to style with ansi codes.
:param fg: if provided this will become the foreground color.
:param bg: if provided this will become the background color.
:param bold: if provided this will enable or disable bold mode.
:param dim: if provided this will enable or disable dim mode. This is
badly supported.
:param underline: if provided this will enable or disable underline.
:param blink: if provided this will enable or disable blinking.
:param reverse: if provided this will enable or disable inverse
rendering (foreground becomes background and the
other way round).
:param reset: by default a reset-all code is added at the end of the
string which means that styles do not carry over. This
can be disabled to compose styles.
z[%dm� zUnknown color %r�( Nr � r? � � � r@ � � r )�append� _ansi_colors�index�
ValueError� TypeError�_ansi_reset_all�join)
r ZfgZbgZboldZdimZ underlineZblink�reverser �bitsr r r �styleG s2 *
r c C s t | �S )a Removes ANSI styling information from a string. Usually it's not
necessary to use this function as Click's echo function will
automatically remove styling if necessary.
.. versionadded:: 2.0
:param text: the text to remove style information from.
)r )r r r r �unstyle� s r� c K s t t| f|�||||d�S )as This function combines :func:`echo` and :func:`style` into one
call. As such the following two calls are the same::
click.secho('Hello World!', fg='green')
click.echo(click.style('Hello World!', fg='green'))
All keyword arguments are forwarded to the underlying functions
depending on which one they go with.
.. versionadded:: 2.0
)re r r rS )r r )r re r r rS Zstylesr r r �secho� s r� �.txtc C s< ddl m} |||||d�}|dkr.|�| �S |�|� dS )a� Edits the given text in the defined editor. If an editor is given
(should be the full path to the executable but the regular operating
system search path is used for finding the executable) it overrides
the detected editor. Optionally, some environment variables can be
used. If the editor is closed without changes, `None` is returned. In
case a file is edited directly the return value is always `None` and
`require_save` and `extension` are ignored.
If the editor cannot be opened a :exc:`UsageError` is raised.
Note for Windows: to simplify cross-platform usage, the newlines are
automatically converted from POSIX to Windows and vice versa. As such,
the message here will have ``\n`` as newline markers.
:param text: the text to edit.
:param editor: optionally the editor to use. Defaults to automatic
detection.
:param env: environment variables to forward to the editor.
:param require_save: if this is true, then not saving in the editor
will make the return value become `None`.
:param extension: the extension to tell the editor about. This defaults
to `.txt` but changing this might change syntax
highlighting.
:param filename: if provided it will edit this file instead of the
provided text contents. It will not use a temporary
file as an indirection in that case.
r )�Editor)�editor�env� require_save� extensionN)rR r� �editZ edit_file)r r� r� r� r� �filenamer� r r r r� � s �
r� c C s ddl m} || ||d�S )ap This function launches the given URL (or filename) in the default
viewer application for this file type. If this is an executable, it
might launch the executable in a new session. The return value is
the exit code of the launched application. Usually, ``0`` indicates
success.
Examples::
click.launch('http://click.pocoo.org/')
click.launch('/my/downloaded/file', locate=True)
.. versionadded:: 2.0
:param url: URL or filename of the thing to launch.
:param wait: waits for the program to stop.
:param locate: if this is set to `True` then instead of launching the
application associated with the URL it will attempt to
launch a file manager with the file located. This
might have weird effects if the URL does not point to
the filesystem.
r )�open_url)�wait�locate)rR r� )Zurlr� r� r� r r r �launch� s r� c C s t }|dkrddlm} || �S )a� Fetches a single character from the terminal and returns it. This
will always return a unicode character and under certain rare
circumstances this might return more than one character. The
situations which more than one character is returned is when for
whatever reason multiple characters end up in the terminal buffer or
standard input was not actually a terminal.
Note that this will always read from the terminal, even if something
is piped into the standard input.
.. versionadded:: 2.0
:param echo: if set to `True`, the character read will also show up on
the terminal. The default is to not show it.
Nr )�getchar)�_getcharrR r� )r r$ r r r r� � s r� � Press any key to continue ...c C sh t tj�rt tj�sdS z:| r,t| d|d� z
t� W n ttfk
rN Y nX W 5 | rbt|d� X dS )a� This command stops execution and waits for the user to press any
key to continue. This is similar to the Windows batch "pause"
command. If the program is not run through a terminal, this command
will instead do nothing.
.. versionadded:: 2.0
.. versionadded:: 4.0
Added the `err` parameter.
:param info: the info string to print before pausing.
:param err: if set to message goes to ``stderr`` instead of
``stdout``, the same as with echo.
Nr Fr )r rA �stdinrj r r� r" r# )�infor r r r �pause s
r� )FN)NFFNNr TF)FFr TF)N)NNNTNFNrU rV rW rX rY NN)NNNNNNNT)NTFN)NNNTr� N)FF)F)r� F)(rG rA r9 Z_compatr r r r r r r r
Zutilsr �
exceptionsr r
�typesr �globalsr r! rw r{ r r r r4 r6 rT rh rm r r� r� r� r� r� r� r� r r r r �<module> sl (
�
O �
,)
�
] �
E
�
%