U
e([V�* � @ s� d dl Z d dlZd dlmZ ddlmZ ddlmZ ddlm Z ddl
m Z dd � Z d
d � Z
d$d
d�Zdd� Zd%dd�Zd&dd�Zdd� Zdd� Zdd� Zdd � Zd d � Zd'd d �Zd!d"� Zdd#lmZmZm Z m Z dS )(� N��update_wrapper� )� iteritems)�_check_for_unicode_literals)�echo��get_current_contextc s � fdd�}t |� �S )z]Marks a callback as wanting to receive the current context
object as first argument.
c s � t � f| �|�S �Nr ��args�kwargs��f� �/decorators.py�new_func s z pass_context.<locals>.new_funcr �r r r r r � pass_context s r c s � fdd�}t |� �S )z�Similar to :func:`pass_context`, but only pass the object on the
context onwards (:attr:`Context.obj`). This is useful if that object
represents the state of a nested system.
c s � t � jf| �|�S r
)r �objr r r r r s zpass_obj.<locals>.new_funcr r r r r �pass_obj s r Fc s � �fdd�}|S )a. Given an object type this creates a decorator that will work
similar to :func:`pass_obj` but instead of passing the object of the
current context, it will find the innermost context of type
:func:`object_type`.
This generates a decorator that works roughly like this::
from functools import update_wrapper
def decorator(f):
@pass_context
def new_func(ctx, *args, **kwargs):
obj = ctx.find_object(object_type)
return ctx.invoke(f, obj, *args, **kwargs)
return update_wrapper(new_func, f)
return decorator
:param object_type: the type of the object to pass.
:param ensure: if set to `True`, a new object will be created and
remembered on the context if it's not there yet.
c s �� �fdd�}t |� �S )Nc sR t � }� r|���}n
|���}|d kr6td�j ��|j�|f| dd � �|�S )NzGManaged to invoke callback without a context object of type %r existingr )r Z
ensure_objectZ find_object� RuntimeError�__name__Zinvoke)r r
�ctxr )�ensurer � object_typer r r 6 s
�z8make_pass_decorator.<locals>.decorator.<locals>.new_funcr r �r r r r � decorator5 s z&make_pass_decorator.<locals>.decoratorr )r r r r r r �make_pass_decorator s
r c C s� t | t�rtd��z| j}|�� | `W n tk
r@ g }Y nX |�d�}|d krtt�| �}t |t �r~|�
d�}n
t� |�}||d< t � |f |p�| j
�� | |d�|��S )Nz5Attempted to convert a callback into a command twice.�helpzutf-8)�name�callback�params)�
isinstance�Command� TypeError�__click_params__�reverse�AttributeError�get�inspectZgetdoc�bytes�decode�cleandocr r �lower)r r �attrs�clsr" r r r r �
_make_commandE s,
��r1 c s �dkr t �� ��fdd�}|S )a* Creates a new :class:`Command` and uses the decorated function as
callback. This will also automatically attach all decorated
:func:`option`\s and :func:`argument`\s as parameters to the command.
The name of the command defaults to the name of the function. If you
want to change that, you can pass the intended name as the first
argument.
All keyword arguments are forwarded to the underlying command class.
Once decorated the function turns into a :class:`Command` instance
that can be invoked as a command line utility or be attached to a
command :class:`Group`.
:param name: the name of the command. This defaults to the function
name.
:param cls: the command class to instantiate. This defaults to
:class:`Command`.
Nc s t | �� ��}| j|_|S r
)r1 �__doc__)r �cmd�r/ r0 r r r r r s zcommand.<locals>.decorator)r$ )r r0 r/ r r r4 r �command\ s r5 c K s |� dt� t| f|�S )z�Creates a new :class:`Group` with a function as callback. This
works otherwise the same as :func:`command` just that the `cls`
parameter is set to :class:`Group`.
r0 )�
setdefault�Groupr5 )r r/ r r r �groupy s r8 c C s8 t | t�r| j�|� n t| d�s(g | _| j�|� d S )Nr& )r# r$ r" �append�hasattrr& )r �paramr r r � _param_memo� s
r< c s � �fdd�}|S )a� Attaches an argument to the command. All positional arguments are
passed as parameter declarations to :class:`Argument`; all keyword
arguments are forwarded unchanged (except ``cls``).
This is equivalent to creating an :class:`Argument` instance manually
and attaching it to the :attr:`Command.params` list.
:param cls: the argument class to instantiate. This defaults to
:class:`Argument`.
c s" � � dt�}t| |�f� �� | S )Nr0 )�pop�Argumentr< )r Z
ArgumentClass�r/ � param_declsr r r � s zargument.<locals>.decoratorr �r@ r/ r r r? r �argument� s
rB c s � �fdd�}|S )a� Attaches an option to the command. All positional arguments are
passed as parameter declarations to :class:`Option`; all keyword
arguments are forwarded unchanged (except ``cls``).
This is equivalent to creating an :class:`Option` instance manually
and attaching it to the :attr:`Command.params` list.
:param cls: the option class to instantiate. This defaults to
:class:`Option`.
c s< d� krt �� d �� d< � �dt�}t| |�f� �� | S )Nr r0 )r* r- r= �Optionr< )r Z OptionClassr? r r r � s
zoption.<locals>.decoratorr rA r r? r �option� s
rD c s � �fdd�}|S )a� Shortcut for confirmation prompts that can be ignored by passing
``--yes`` as parameter.
This is equivalent to decorating a function with :func:`option` with
the following parameters::
def callback(ctx, param, value):
if not value:
ctx.abort()
@click.command()
@click.option('--yes', is_flag=True, callback=callback,
expose_value=False, prompt='Do you want to continue?')
def dropdb():
pass
c sV dd� }� � dd� � � d|� � � dd� � � dd � � � d
d � t�pLd � �| �S )
Nc S s |s | � � d S r
)�abort�r r; �valuer r r r! � s z8confirmation_option.<locals>.decorator.<locals>.callback�is_flagTr! � expose_valueF�promptzDo you want to continue?r z%Confirm the action without prompting.)z--yes�r6 rD �r r! r? r r r � s z&confirmation_option.<locals>.decoratorr rA r r? r �confirmation_option� s
rM c s � �fdd�}|S )aL Shortcut for password prompts.
This is equivalent to decorating a function with :func:`option` with
the following parameters::
@click.command()
@click.option('--password', prompt=True, confirmation_prompt=True,
hide_input=True)
def changeadmin(password):
pass
c s6 � � dd� � � dd� � � dd� t�p,d� �| �S )NrJ TZconfirmation_promptZ
hide_input)z
--passwordrK r r? r r r � s z"password_option.<locals>.decoratorr rA r r? r �password_option� s rN c s0 �dkrt �d�j�d��� ���fdd�}|S )ac Adds a ``--version`` option which immediately ends the program
printing out the version number. This is implemented as an eager
option that prints the version and exits the program in the callback.
:param version: the version number to show. If not provided Click
attempts an auto discovery via setuptools.
:param prog_name: the name of the program (defaults to autodetection)
:param message: custom message to show instead of the default
(``'%(prog)s, version %(version)s'``)
:param others: everything else is forwarded to :func:`option`.
Nr r c st �� dd ���� dd�� � ���fdd�}��dd� ��dd � ��d
d� ��d d � |�d
< t�pjd��| �S )N� prog_name�messagez %(prog)s, version %(version)sc
s� |r
| j rd S �}|d kr$| �� j}�}|d kr�z dd l}W n tk
rP Y nFX |jD ]<}|�� �d�pli }t|�D ] \}} | j �krv|j
} qXqvqX|d kr�t d��t � ||d� | j
d� | �� d S )Nr Zconsole_scriptszCould not determine version)�prog�version��color)�resilient_parsingZ find_rootZ info_name�
pkg_resources� ImportErrorZ working_setZ
get_entry_mapr) r Z module_namerR r r rT �exit)
r r; rG rQ ZverrV ZdistZscriptsZ script_nameZ entry_point)rP �modulerO rR r r r! � s4
��z3version_option.<locals>.decorator.<locals>.callbackrH TrI F�is_eagerr zShow the version and exit.r! )z --version)r= r6 rD rL �r/ rY r@ rR )rP rO r r � s z!version_option.<locals>.decorator)�sys� _getframe� f_globalsr) )rR r@ r/ r r r[ r �version_option� s %r_ c s � �fdd�}|S )al Adds a ``--help`` option which immediately ends the program
printing out the help page. This is usually unnecessary to add as
this is added by default to all commands unless suppressed.
Like :func:`version_option`, this is implemented as eager option that
prints in the callback and exits.
All arguments are forwarded to :func:`option`.
c sR dd� }� � dd� � � dd� � � dd� � � d d� |� d
< t�pHd � �| �S ) Nc S s( |r$| j s$t| �� | jd� | �� d S )NrS )rU r Zget_helprT rX rF r r r r! " s
z0help_option.<locals>.decorator.<locals>.callbackrH TrI Fr zShow this message and exit.rZ r! )z--helprK rL r? r r r ! s z help_option.<locals>.decoratorr rA r r? r � help_option s
r` )r$ r7 r> rC )F)NN)N)N) r\ r* � functoolsr Z_compatr Z _unicodefunr Zutilsr �globalsr r r r r1 r5 r8 r< rB rD rM rN r_ r` Zcorer$ r7 r> rC r r r r �<module> s(
&
6