File: //lib/python3/dist-packages/awscli/customizations/__pycache__/awslambda.cpython-38.pyc
U
���^G � @ s� d dl Z d dlZd dlmZ d dlZd dlmZmZ dZdZ dd� Z
dd � Z G d
d � d e �Z
d d
� Zdd� ZG dd� de�ZG dd� de�ZdS )� N)�closing)�CustomArgument� CLIArgumentzk--zip-file must be a zip file with the fileb:// prefix.
Example usage: --zip-file fileb://path/to/file.zipz�<p>The path to the zip file of the {param_type} you are uploading. Specify --zip-file or --{param_type}, but not both. Example: fileb://{param_type}.zip</p>c C s@ | � dtd�j� | � dtd�j� | � dt� | � dt� d S )Nz.building-argument-table.lambda.create-functionZCodez4building-argument-table.lambda.publish-layer-versionZContentz3building-argument-table.lambda.update-function-codez+process-cli-arg.lambda.update-function-code)�register�ZipFileArgumentHoister�hoist�_modify_zipfile_docstring�validate_is_zip_file)Zcli� r
�A/usr/lib/python3/dist-packages/awscli/customizations/awslambda.py� register_lambda_create_function! s ����r c K s | j dkrt|� d S �N�zip-file)�name�_should_contain_zip_content)Z cli_argument�value�kwargsr
r
r r , s
r c @ s e Zd ZdZdd� Zdd� ZdS )r ar Hoists a ZipFile argument up to the top level.
Injects a top-level ZipFileArgument into the argument table which maps
a --zip-file parameter to the underlying ``serialized_name`` ZipFile
shape. Repalces the old ZipFile argument with an instance of
ReplacedZipFileArgument to prevent its usage and recommend the new
top-level injected parameter.
c C s || _ |�� | _d S )N)�_serialized_name�lower�_name)�self�serialized_namer
r
r �__init__: s z ZipFileArgumentHoister.__init__c K sl t j| jd�}td|d| jd�|d< || j }t�|j�}|jd= t | j||j
d|� d�| jd�|| j< d S ) N)Z
param_typer Zblob)� help_textZ
cli_type_namer �ZipFileF�
event_emitter)r �argument_modelZoperation_modelZ is_requiredr r ) �
ZIP_DOCSTRING�formatr �ZipFileArgumentr �copy�deepcopyr �members�ReplacedZipFileArgumentZ_operation_modelZ
get_component)r �session�argument_tabler r ZargumentZmodelr
r
r r > s$ �
�z ZipFileArgumentHoister.hoistN)�__name__�
__module__� __qualname__�__doc__r r r
r
r
r r 1 s r c K s d|krt |d _d S r
)r Z
documentation)r$ r% r r
r
r r Q s r c C sh t | t�s| �d�} t�| �}z&tt�|���}|�� W 5 Q R X W n tj k
rb t
t ��Y nX d S )Nzutf-8) �
isinstance�bytes�encode�six�BytesIOr �zipfiler ZinfolistZ
BadZipfile�
ValueError� ERROR_MSG)r Zfileobj�fr
r
r r V s
r c s( e Zd ZdZ� fdd�Zdd� Z� ZS )r a� A new ZipFile argument to be injected at the top level.
This class injects a ZipFile argument under the specified serialized_name
parameter. This can be used to take a top level parameter like --zip-file
and inject it into a nested different parameter like Code so
--zip-file foo.zip winds up being serilized as
{ 'Code': { 'ZipFile': <contents of foo.zip> } }.
c s" |� d�| _tt| �j||� d S )Nr )�pop�_param_to_replace�superr r �r �argsr �� __class__r
r r n s zZipFileArgument.__init__c C sH |d kr d S t |� d|i}|�| j�r:|| j �|� n
||| j< d S )Nr )r �getr4 �update)r �
parametersr Zzip_file_paramr
r
r �
add_to_paramsr s z ZipFileArgument.add_to_params�r& r' r( r) r r= �
__classcell__r
r
r8 r r e s r c s( e Zd ZdZ� fdd�Zdd� Z� ZS )r# a� A replacement arugment for nested ZipFile argument.
This prevents the use of a non-working nested argument that expects binary.
Instead an instance of ZipFileArgument should be injected at the top level
and used instead. That way fileb:// can be used to load the binary
contents. And the argument class can inject those bytes into the correct
serialization name.
c s. t t| �j||� d|d | _|d | _d S )Nz--%sr r )r5 r# r � _cli_namer4 r6 r8 r
r r � s z ReplacedZipFileArgument.__init__c C sX |d kr d S | � |�}d|kr,td| j ��|�| j�rJ|| j �|� n
||| j< d S )Nr zyZipFile cannot be provided as part of the %s argument. Please use the '--zip-file' option instead to specify a zip file.)Z_unpack_argumentr0 r@ r: r4 r; )r r< r Zunpackedr
r
r r= � s
�� z%ReplacedZipFileArgument.add_to_paramsr> r
r
r8 r r# } s r# )r/ r �
contextlibr r- Zawscli.argumentsr r r1 r r r �objectr r r r r# r
r
r
r �<module>
s ��