File: //lib/python3/dist-packages/awscli/customizations/__pycache__/awslambda.cpython-310.pyc
o
�s�aG � @ 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-function�Codez4building-argument-table.lambda.publish-layer-version�Contentz3building-argument-table.lambda.update-function-codez+process-cli-arg.lambda.update-function-code)�register�ZipFileArgumentHoister�hoist�_modify_zipfile_docstring�validate_is_zip_file)�cli� r
�A/usr/lib/python3/dist-packages/awscli/customizations/awslambda.py� register_lambda_create_function! s ����r c K s | j dkr t|� d S d S �N�zip-file)�name�_should_contain_zip_content)� 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)�
param_typer �blob)� help_text�
cli_type_namer �ZipFileF�
event_emitter)r �argument_model�operation_model� is_requiredr" r ) �
ZIP_DOCSTRING�formatr �ZipFileArgumentr �copy�deepcopyr# �members�ReplacedZipFileArgument�_operation_model�
get_component)r �session�argument_tabler r �argument�modelr
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|v r t |d _d S d S r )r& �
documentation)r/ r0 r r
r
r r
Q s �r
c C sx t | t�s
| �d�} t�| �}z tt�|���}|�� W d � W d S 1 s(w Y W d S tj y; t
t ��w )Nzutf-8) �
isinstance�bytes�encode�six�BytesIOr �zipfiler! �infolist�
BadZipfile�
ValueError� ERROR_MSG)r �fileobj�fr
r
r r V s
&��r c �( 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|i |�� d S )Nr )�pop�_param_to_replace�superr( r �r �argsr �� __class__r
r r n s zZipFileArgument.__init__c C sJ |d u rd S t |� d|i}|�| j�r || j �|� d S ||| j< d S )Nr! )r �getrF �update)r �
parametersr �zip_file_paramr
r
r �
add_to_paramsr s z ZipFileArgument.add_to_params�r3 r4 r5 r6 r rP �
__classcell__r
r
rJ r r( e s r( c rD )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 s2 t t| �j|i |�� d|d | _|d | _d S )Nz--%sr r )rG r, r � _cli_namerF rH rJ r
r r � s z ReplacedZipFileArgument.__init__c C sZ |d u rd S | � |�}d|v rtd| j ��|�| j�r&|| j �|� d S ||| 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.)�_unpack_argumentr@ rS rL rF rM )r rN r �unpackedr
r
r rP � s
�� z%ReplacedZipFileArgument.add_to_paramsrQ r
r
rJ r r, } s r, )r= r) �
contextlibr r; �awscli.argumentsr r rA r& r r �objectr r
r r( r, r
r
r
r �<module> s ��