File: //lib/python3.6/site-packages/awscli/__pycache__/argprocess.cpython-36.pyc
3
�T_�Q � @ s4 d Z ddlZddlZddlmZ ddlmZmZ ddlm Z m
Z
ddlm Z ddl m
Z
ddlmZ ejd �ZG d
d � d e�ZG d d
� d
e�ZG dd� de�ZG dd� de�Zdd� Zdd� Zdd� Zdd� Zdd� Zd d � Z d d � Z d d!� Z d-d#d$�Z d%d&� Z G d'd(� d(e!�Z"G d)d*� d*e"�Z#G d+d,� d,e"�Z$dS ).z Module for processing CLI args.� N)�six)� OrderedDict�json)� SCALAR_TYPES�
COMPLEX_TYPES)� shorthand)�%find_service_and_method_in_event_name)�is_json_value_headerzawscli.argprocessc s e Zd Z� fdd�Z� ZS )�
ParamErrorc s, d||f }t t| �j|� || _|| _dS )aB
:type cli_name: string
:param cli_name: The complete cli argument name,
e.g. "--foo-bar". It should include the leading
hyphens if that's how a user would specify the name.
:type message: string
:param message: The error message to display to the user.
z Error parsing parameter '%s': %sN)�superr
�__init__�cli_name�message)�selfr
r � full_message)� __class__� � /usr/lib/python3.6/argprocess.pyr s
zParamError.__init__)�__name__�
__module__� __qualname__r �
__classcell__r r )r r r
s r
c @ s e Zd ZdS )�ParamSyntaxErrorN)r r r r r r r r 0 s r c s e Zd Z� fdd�Z� ZS )�ParamUnknownKeyErrorc s* dj |�}d||f }tt| �j|� d S )Nz, z'Unknown key '%s', valid choices are: %s)�joinr r r )r �keyZ
valid_keysr )r r r r 5 s
z ParamUnknownKeyError.__init__)r r r r r r r )r r r 4 s r c @ s e Zd ZdS )�TooComplexErrorN)r r r r r r r r = s r c C s: t |dd�}| jd|||f ||||d�}|dk r6|}|S )a
Unpack an argument's value from the commandline. This is part one of a two
step process in handling commandline arguments. Emits the load-cli-arg
event with service, operation, and parameter names. Example::
load-cli-arg.ec2.describe-instances.foo
�nameZ anonymouszload-cli-arg.%s.%s.%s)�param�value� service_name�operation_nameN)�getattrZ emit_first_non_none_response)Zsessionr r! � cli_argumentr Z
param_nameZvalue_overrider r r �unpack_argumentA s r$ c C s g }t | |�S )N)�_detect_shape_structure)r �stackr r r �detect_shape_structureY s r' c s� | j � krdS � j| j � z�| jtkr*dS | jdkr�� fdd�| jj� D �}t|�dkrntdd� |D ��rnd S t|�dkr�td
d� |D ��r�d S d d
jt t
|��� S nD| jdkr�dt | j � � S | jdkr�| j
jtkr�dS dt | j
� � S W d � j� X d S )N� recursive�scalar� structurec s g | ]}t |� ��qS r )r% )�.0�p)r& r r �
<listcomp>g s z+_detect_shape_structure.<locals>.<listcomp>� c s s | ]}|d kV qdS )r) Nr )r+ r, r r r � <genexpr>m s z*_detect_shape_structure.<locals>.<genexpr>zstructure(scalar)c s s | ]}|d kV qdS )r) Nr )r+ r, r r r r/ o s zstructure(scalars)z
structure(%s)z, �listzlist-%s�mapz
map-scalarzmap-%s)r �append� type_namer �members�values�len�allr �sorted�setr% �memberr �pop)r r& Z sub_typesr )r& r r% ^ s*
r% c C s t | j|| j�S )a1
Parses and unpacks the encoded string command line parameter
and returns native Python data structures that can be passed
to the Operation.
:type cli_argument: :class:`awscli.arguments.BaseCLIArgument`
:param cli_argument: The CLI argument object.
:param value: The value of the parameter. This can be a number of
different python types (str, list, etc). This is the value as
it's specified on the command line.
:return: The "unpacked" argument than can be sent to the `Operation`
object in python.
)�_unpack_cli_arg�argument_modelr
)r# r r r r �unpack_cli_arg~ s r> c C s. | j jd�r*| j jd�dkr*| jdkr*dS dS )NZ jsonvalue�location�header�stringTF)�
serialization�getr3 )�modelr r r �
_special_type� s
rE c C sN t | �rt| ||�S | jtkr*t| ||�S | jtkr@t| ||�S tj|�S d S )N) r �_unpack_json_cli_argr3 r �unpack_scalar_cli_argr �_unpack_complex_cli_argr Z text_type)r= r r
r r r r< � s
r< c C sH yt j|td�S tk
rB } zt|d||f ��W Y d d }~X nX d S )N)�object_pairs_hookz"Invalid JSON: %s
JSON received: %s)r �loadsr �
ValueErrorr
)r= r r
�er r r rF � s rF c sJ | j }|dks|dkrz|j� d dkrjytj|td�S tk
rh } zt� d||f ��W Y d d }~X nX t� d| ��n�|dk�rFt|tj �r�|j� d d kr�tj|td�S nDt|t
�r�t |�d
kr�|d j � }|r�|d d kr�tj|d td�S y| j
�� �fd d �|D �S ttfk
�rD } zt� |d ��W Y d d }~X nX d S )
Nr* r1 r �{)rI z"Invalid JSON: %s
JSON received: %szInvalid JSON:
%sr0 �[r. c s g | ]}t �|� ��qS r )r< )r+ �v)r
�member_shape_modelr r r- � s z+_unpack_complex_cli_arg.<locals>.<listcomp>)r3 �lstripr rJ r rK r
�
isinstancer � string_typesr0 r6 �stripr: � TypeError)r= r r
r3 rL Z single_valuer )r
rP r rH � s0
rH � c C s� | j dks| j dkr t|�S | j dks0| j dkr8t|�S | j dkr�| jjd�r�tjj|�}tjj|�}tjj |�s�d}t
||��t |d�S | j d kr�t |t
j�r�|j� d
kr�d S t|�S |S d S ) NZintegerZlong�floatZdoubleZblobZ streamingz%Blob values must be a path to a file.�rbZbooleanZfalseF)r3 �intrW rB rC �os�path�
expandvars�
expanduser�isfiler
�openrR r rS �lower�bool)r= r r
Z file_path�msgr r r rG � s"
rG c C s, | j dkrdS | j dkr(| jj dkr(dS dS )Nr* r0 r1 FT)r* r0 r1 )r* r0 r1 )r3 r: )rD r r r �_is_complex_shape� s
rc c @ s e Zd Zdd� ZdS )�ParamShorthandc C sN ddgidgdgdgd�dgdgdgdgd �d
�}|j |i �j |g �}||kS ) z�
Determines whether a given operation for a service needs to use the
deprecated shorthand parsing case for lists of structures that only have
a single member.
zput-record-batchZrecordszreboot-workspace-requestszrebuild-workspace-requestsz terminate-workspace-requests)zreboot-workspaceszrebuild-workspaceszterminate-workspacesZtagsZ instances)z remove-tagszdescribe-instance-healthz'deregister-instances-from-load-balancerz%register-instances-with-load-balancer)ZfirehoseZ
workspaceszelastic-load-balancing)rC )r �
service_idr! Z
argument_nameZcasesr r r �_uses_old_list_case� s z"ParamShorthand._uses_old_list_caseN)r r r rf r r r r rd � s rd c @ s4 e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd S ) �ParamShorthandParserc C s t j� | _t j� | _d S )N)r ZShorthandParser�_parserZBackCompatVisitor�_visitor)r r r r r s
z ParamShorthandParser.__init__c K s0 | j ||�sdS t|�\}}| j||||�S dS )a� Attempt to parse shorthand syntax for values.
This is intended to be hooked up as an event handler (hence the
**kwargs). Given ``param`` object and its string ``value``,
figure out if we can parse it. If we can parse it, we return
the parsed value (typically some sort of python dict).
:type cli_argument: :class:`awscli.arguments.BaseCLIArgument`
:param cli_argument: The CLI argument object.
:type param: :class:`botocore.parameters.Parameter`
:param param: The parameter object (includes various metadata
about the parameter).
:type value: str
:param value: The value for the parameter type on the command
line, e.g ``--foo this_value``, value would be ``"this_value"``.
:returns: If we can parse the value we return the parsed value.
If it looks like JSON, we return None (which tells the event
emitter to use the default ``unpack_cli_arg`` provided that
no other event handlers can parsed the value). If we
run into an error parsing the value, a ``ParamError`` will
be raised.
N)�_should_parse_as_shorthandr �_parse_as_shorthand)r r# r Z
event_name�kwargsre r! r r r �__call__ s
z ParamShorthandParser.__call__c s� yxt jd|j� � j||||�}|d k r,|S t|t�rZ� fdd�|D �}� jj||j� n � j j
|�}� jj||j� W nh t j k
r� } zt
|jt|���W Y d d }~X n6 t
tfk
r� } zt
|jt|���W Y d d }~X nX |S )Nz Parsing param %s as shorthandc s g | ]}� j j|��qS r )rh �parse)r+ rO )r r r r- K s z<ParamShorthandParser._parse_as_shorthand.<locals>.<listcomp>)�LOG�debugr
�_handle_special_casesrR r0 ri Zvisitr= rh rn r ZShorthandParseErrorr
�strr )r r# r re r! Z
handled_valueZparsedrL r )r r rk < s"
"z(ParamShorthandParser._parse_as_shorthandc s� |j }|jdkrf|jjdkrft|jj�dkrf| j|||j�rft|jjj� �d � � fdd�|D �}|S |jdkr�t|j�dkr�d|jkr�|jd jdkr�d |kr�d|iS d S )
Nr0 r* r. r c s g | ] }� |i�qS r r )r+ rO )�key_namer r r- n s z>ParamShorthandParser._handle_special_cases.<locals>.<listcomp>�ValuerA �=) r= r3 r: r6 r4 rf r r0 �keys)r r# r re r! rD Z
new_valuesr )rs r rq [ s
z*ParamShorthandParser._handle_special_casesc C sV |rt |t�r|d }n|}t |tj�rH|j� jd�rHtjd|j� dS |j }t
|�S )Nr rN rM z=Param %s looks like JSON, not considered for param shorthand.F)rN rM ) rR r0 r rS rT �
startswithro rp Zpy_namer= rc )r r# r Z check_valrD r r r rj ~ s
z/ParamShorthandParser._should_parse_as_shorthandN)r r r r rm rk rq rj r r r r rg s
$ #rg c @ sZ e Zd ZdZe� ZdZdd� Zdd� Zdd� Z d d
� Z
d d � Z d
d� Z dd� Z
dd� ZdS )�ParamShorthandDocGenz3Documentation generator for param shorthand syntax.� c C s |dk rt |�S dS )z3Checks if a CLI argument supports shorthand syntax.NF)rc )r r= r r r �supports_shorthand� s z'ParamShorthandDocGen.supports_shorthandc C sx | j |||�}|| jkr dS |r$|S g }y6|jjdkrN|jj}| j||�d S | j|j|�S W n tk
rr dS X dS )a� Generate documentation for a CLI argument.
:type cli_argument: awscli.arguments.BaseCLIArgument
:param cli_argument: The CLI argument which to generate
documentation for.
:return: Returns either a string or ``None``. If a string
is returned, it is the generated shorthand example.
If a value of ``None`` is returned then this indicates
that no shorthand syntax is available for the provided
``argument_model``.
Nr0 z ...rV )rq � _DONT_DOCr= r3 r: �_shorthand_docsr )r r# re r! Z docstringr& r= r r r �generate_shorthand_example� s
z/ParamShorthandDocGen.generate_shorthand_examplec C s� |j }|jdkr�|jjdkr�t|jj�dkr�| j|||j�r�t|jj�d }|j}|j d�dkr||j d�dkr|d|j
|f S d|j
|||f S |jdkr�t|j�dkr�d |jkr�|jd jd
kr�| j S d S ) Nr0 r* r. r �min�maxz%s %s1z%s %s1 %s2 %s3rt rA rV ) r= r3 r: r6 r4 rf r r0 �metadatarC r
r{ )r r# re r! rD � member_namer� r r r rq � s$
z*ParamShorthandDocGen._handle_special_casesc C s` t |�| jkrt� �|jdkr*| j||�S |jdkr@| j||�S |jdkrV| j||�S |jS d S )Nr* r0 r1 )r6 �
_MAX_STACKr r3 �_structure_docs�
_list_docs� _map_docs)r r= r& r r r r| � s
z$ParamShorthandDocGen._shorthand_docsc C sb |j }|j|j� z| j|j |�}W d |j� X |jtksFt|�dkrRd||f S d||f S d S )Nr. z[%s,%s]z%s,%s)r: r2 r r| r; r3 r r6 )r r= r& Z list_memberZ element_docsr r r r� � s
z ParamShorthandDocGen._list_docsc C sb |j }| j|j|�}d||f }|jrR| rR|d7 }x(|jD ]}|d| 7 }q<W n |r^d| }|S )NzKeyName1=%s,KeyName2=%sz
Where valid key names are:
z %s
z{%s})r r| r �enum)r r= r& �kZ
value_docs�startr� r r r r� � s z ParamShorthandDocGen._map_docsc C sJ g }x*|j j� D ] \}}|j| j|||�� qW dj|�}|sB|S d| S )N�,z{%s})r4 �itemsr2 � _member_docsr )r r= r& �partsr Z member_shapeZ
inner_partr r r r� � s
z$ParamShorthandDocGen._structure_docsc C sH |j |j�dkrdS |j|j� z| j||�}W d |j� X d||f S )Nr z( ... recursive ... )z%s=%s)�countr r2 r| r; )r r �shaper& Z value_docr r r r� s
z!ParamShorthandDocGen._member_docsN)r r r �__doc__�objectr{ r� rz r} rq r| r� r� r� r� r r r r rx � s " rx )rV )%r� rZ ZloggingZ
awscli.compatr Zbotocore.compatr r Zawsclir r r Z awscli.utilsr Zbotocore.utilsr Z getLoggerro � Exceptionr
r r r r$ r' r% r> rE r< rF rH rG rc r� rd rg rx r r r r �<module>
s6
$