|
Server : Apache System : Linux iad1-shared-b8-43 6.6.49-grsec-jammy+ #10 SMP Thu Sep 12 23:23:08 UTC 2024 x86_64 User : dh_edsupp ( 6597262) PHP Version : 8.2.26 Disable Function : NONE Directory : /lib/python3/dist-packages/awscli/__pycache__/ |
Upload File : |
o
.&�aXS � @ sD d Z ddlZddlZddlmZ ddlmZmZ ddlm Z m
Z
ddlm Z ddl m
Z
mZmZ ddlmZ e�d �ZG d
d � d e�ZG d d
� d
e�ZG dd� de�ZG dd� de�Zdd� Zdd� Zdd� Zdd� Z dd� Z d d � Z d d � Z d d!� Z d/d#d$�Z!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 )0z Module for processing CLI args.� N)�six)� OrderedDict�json)� SCALAR_TYPES�
COMPLEX_TYPES)� shorthand)�%find_service_and_method_in_event_name�is_document_type�is_document_type_container)�is_json_value_headerzawscli.argprocessc � e Zd Z� fdd�Z� ZS )�
ParamErrorc s, d||f }t t| ��|� || _|| _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__� �3/usr/lib/python3/dist-packages/awscli/argprocess.pyr s �
zParamError.__init__��__name__�
__module__� __qualname__r �
__classcell__r r r r r
� r
c @ � e Zd ZdS )�ParamSyntaxErrorN�r r r r r r r r 3 � r c r )�ParamUnknownKeyErrorc s* d� |�}d||f }tt| ��|� d S )N�, z'Unknown key '%s', valid choices are: %s)�joinr r" r )r �key�
valid_keysr r r r r 8 s
��z ParamUnknownKeyError.__init__r r r r r r" 7 r r" c @ r )�TooComplexErrorNr r r r r r' @ r! r' c C s: t |dd�}| jd|||f ||||d�}|dur|}|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
�name� anonymouszload-cli-arg.%s.%s.%s)�param�value� service_name�operation_nameN)�getattr� emit_first_non_none_response)�sessionr, r- � cli_argumentr+ �
param_name�value_overrider r r �unpack_argumentD s ��r4 c C s g }t | |�S �N��_detect_shape_structure)r* �stackr r r �detect_shape_structure\ s
r9 c sH | j � v rdS � �| j � z�| jtv rW � �� dS | jdkrg� fdd�| j�� D �}t|�dkrAtdd� |D ��rAW � �� d S t|�dkrWtd
d� |D ��rWW � �� d S d d
� t
t |��� W � �� S | jdkrydt | j
� � W � �� S | jdkr�| jjtv r�W � �� dS dt | j� � W � �� S W � �� d S � �� w )N� recursive�scalar� structurec s g | ]}t |� ��qS r r6 ��.0�p�r8 r r �
<listcomp>j s �z+_detect_shape_structure.<locals>.<listcomp>� c s � � | ]}|d kV qdS �r; Nr r= r r r � <genexpr>p � � z*_detect_shape_structure.<locals>.<genexpr>zstructure(scalar)c s rC rD r r= r r r rE r rF zstructure(scalars)z
structure(%s)r# �listzlist-%s�mapz
map-scalarzmap-%s)r( �append� type_namer �pop�members�values�len�allr$ �sorted�setr7 �memberr+ )r* r8 � sub_typesr r@ r r7 a s<
�
�
� �
�
� �
�r7 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 )r1 r+ r r r �unpack_cli_arg� s �rV c C s. | j �d�r| j �d�dkr| jdkrdS dS )N� jsonvalue�location�header�stringTF)�
serialization�getrJ ��modelr r r �
_special_type� s
r_ c C sR t | �st| �rt| ||�S | jtv rt| ||�S | jtv r$t| ||�S t� |�S r5 )
r r �_unpack_json_cli_argrJ r �unpack_scalar_cli_argr �_unpack_complex_cli_argr � text_type)rU r+ r r r r rT � s �
�
�
rT c
C s< zt j|td�W S ty } z t|d||f ��d }~ww )N)�object_pairs_hookz"Invalid JSON: %s
JSON received: %s)r �loadsr �
ValueErrorr
)rU r+ r �er r r r` � s ����r` c
s | j }|dks |dkr |�� d dkrt| |� �S t� d| ��|dkr~t|tj�r9|�� d dkr8t| |� �S n!t|t�rZt|�dkrZ|d � � }|rZ|d dkrZt| |d � �S z| j
�� �fd d
�|D �W S t t fy} } zt� |d ��d }~ww d S ) Nr< rH r �{zInvalid JSON:
%srG �[rB c s g | ]}t �|� ��qS r )rT �r> �v�r �member_shape_modelr r rA � s �z+_unpack_complex_cli_arg.<locals>.<listcomp>)
rJ �lstripr` r
�
isinstancer � string_typesrG rN �striprR rf � TypeError)rU r+ r rJ � single_valuerg r rl r rb � s0 � ����rb � c C s� | j dks
| j dkrt|�S | j dks| j dkr t|�S | j dkrE| j�d�rEtj�|�}tj�|�}tj� |�s@d}t
||��t |d�S | j d kr\t |t
j�rX|�� d
krXd S t|�S |S ) N�integer�long�float�double�blob� streamingz%Blob values must be a path to a file.�rb�boolean�falseF)rJ �intrw r[ r\ �os�path�
expandvars�
expanduser�isfiler
�openro r rp �lower�bool)rU r+ r � file_path�msgr r r ra � s$
�
ra c C s t | �rdS t| �S )NF)r
�_is_complex_shaper] r r r �_supports_shorthand_syntax� s
r� c C s, | j dvrdS | j dkr| jj dvrdS dS )N)r< rG rH FrG T)rJ rR r] r r r r� � s
r� c @ s e Zd Zdd� ZdS )�ParamShorthandc C sN ddgidgdgdgd�dgdgdgdgd �d
�}|� |i �� |g �}||v S ) 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-batch�recordszreboot-workspace-requestszrebuild-workspace-requestsz terminate-workspace-requests)zreboot-workspaceszrebuild-workspaceszterminate-workspaces�tags� instances)z remove-tagszdescribe-instance-healthz'deregister-instances-from-load-balancerz%register-instances-with-load-balancer)�firehose�
workspaceszelastic-load-balancing)r\ )r �
service_idr- �
argument_name�casesr r r �_uses_old_list_case s ����z"ParamShorthand._uses_old_list_caseN)r r r r� r r r r r� s r� c @ s4 e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd S ) �ParamShorthandParserc C s t �� | _t �� | _d S r5 )r �ShorthandParser�_parser�BackCompatVisitor�_visitor�r r r r r " s
z ParamShorthandParser.__init__c K s, | � ||�sdS t|�\}}| �||||�S )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 r1 r+ �
event_name�kwargsr� r- r r r �__call__& s ��z ParamShorthandParser.__call__c
s� z@t �d|j� � �||||�}|d ur|W S t|t�r0� fdd�|D �}� j�||j� W |S � j �
|�}� j�||j� W |S t j yU } zt
|jt|���d }~w t
tfyj } zt
|jt|���d }~ww )Nz Parsing param %s as shorthandc s g | ]}� j �|��qS r )r� �parserj r� r r rA Y s z<ParamShorthandParser._parse_as_shorthand.<locals>.<listcomp>)�LOG�debugr �_handle_special_casesro rG r� �visitrU r� r� r �ShorthandParseErrorr
�strr" )r r1 r+ r� r- �
handled_value�parsedrg r r� r r� J s. ��
� ����z(ParamShorthandParser._parse_as_shorthandc s� |j }|jdkr3|jjdkr3t|jj�dkr3| �|||j�r3t|jj�� �d � � fdd�|D �}|S |jdkrTt|j�dkrVd|jv rX|jd jdkrZd |vr\d|iS d S d S d S d S d S )
NrG r< rB r c s g | ]}� |i�qS r r rj ��key_namer r rA | s z>ParamShorthandParser._handle_special_cases.<locals>.<listcomp>�ValuerZ �=) rU rJ rR rN rL r� r( rG �keys)r r1 r+ r� r- r^ �
new_valuesr r� r r� i s(
�
�z*ParamShorthandParser._handle_special_casesc C sV |r t |t�r |d }n|}t |tj�r$|�� �d�r$t�d|j� dS |j }t
|�S )Nr )ri rh z=Param %s looks like JSON, not considered for param shorthand.F) ro rG r rp rq �
startswithr� r� �py_namerU r� )r r1 r+ � check_valr^ r r r r� � s
��z/ParamShorthandParser._should_parse_as_shorthandN)r r r r r� r� r� r� r r r r r� s $ #r� 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 |durt |�S dS )z3Checks if a CLI argument supports shorthand syntax.NF)r� )r rU r r r �supports_shorthand� s z'ParamShorthandDocGen.supports_shorthandc C st | � |||�}|| ju rdS |r|S g }z|jjdkr(|jj}| �||�d W S | �|j|�W S ty9 Y dS w )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``.
NrG z ...rt )r� � _DONT_DOCrU rJ rR �_shorthand_docsr' )r r1 r� r- � docstringr8 rU r r r �generate_shorthand_example� s �
�z/ParamShorthandDocGen.generate_shorthand_examplec C s� |j }|jdkrG|jjdkrGt|jj�dkrG| �|||j�rGt|jj�d }|j}|� d�dkr>|� d�dkr>d|j
|f S d|j
|||f S |jdkrct|j�dkrcd |jv rc|jd jd
krc| j S d S ) NrG r< rB r �min�maxz%s %s1z%s %s1 %s2 %s3r� rZ rt ) rU rJ rR rN rL r� r( rG �metadatar\ r r� )r r1 r� r- r^ � member_namer� r r r r� � s*
�� �
z*ParamShorthandDocGen._handle_special_casesc C s\ t |�| jkr
t� �|jdkr| �||�S |jdkr | �||�S |jdkr+| �||�S |jS )Nr< rG rH )rN �
_MAX_STACKr' rJ �_structure_docs�
_list_docs� _map_docs)r rU r8 r r r r� � s
z$ParamShorthandDocGen._shorthand_docsc C sf |j }|�|j� z
| �|j |�}W |�� n|�� w |jtv s't|�dkr-d||f S d||f S )NrB z[%s,%s]z%s,%s)rR rI r( r� rK rJ r rN )r rU r8 � list_member� element_docsr r r r� � s z ParamShorthandDocGen._list_docsc C s^ |j }| �|j|�}d||f }|jr'|s'|d7 }|jD ]}|d| 7 }q |S |r-d| }|S )NzKeyName1=%s,KeyName2=%sz
Where valid key names are:
z %s
�{%s})r% r� r+ �enum)r rU r8 �k�
value_docs�startr� r r r r� � s
�z ParamShorthandDocGen._map_docsc C sP g }|j �� D ]\}}t|�rq|�| �|||�� qd�|�}|s$|S d| S )N�,r� )rL �itemsr
rI � _member_docsr$ )r rU r8 �partsr( � member_shape�
inner_partr r r r� s
z$ParamShorthandDocGen._structure_docsc C sP |� |j�dkr
dS |�|j� z | �||�}W |�� n|�� w d||f S )Nr z( ... recursive ... )z%s=%s)�countr( rI r� rK )r r( �shaper8 � value_docr r r r� s z!ParamShorthandDocGen._member_docsN)r r r �__doc__�objectr� r� r� r� r� r� r� r� r� r� r r r r r� � s " r� )rt )(r� r �logging�
awscli.compatr �botocore.compatr r �awsclir r r � awscli.utilsr r r
�botocore.utilsr � getLoggerr� � Exceptionr
r r" r' r4 r9 r7 rV r_ rT r` rb ra r� r� r� r� r� r� r r r r �<module> s8
}