File: //lib/python3.6/site-packages/awscli/customizations/__pycache__/awslambda.cpython-36.pyc
3
�T_^ � @ s� d dl Z d dlZd dlmZ d dlmZ d dlmZmZ dZ dZ
dd� Z d d
� Z G d d � d e
�Zd
d� Zdd� ZG dd� de�ZG dd� de�ZdS )� N)�closing)�six)�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@ | j dtd�j� | j dtd�j� | j dt� | j 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 � /usr/lib/python3.6/awslambda.py� register_lambda_create_function! s r
c K s | j dkrt|� d S )Nzzip-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 || _ |j� | _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 }tj|j�}|jd= t | j||j
d|j d�| jd�|| j< d S ) N)Z
param_typezzip-fileZblob)� 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 )Nzzip-file)r Z
documentation)r# r$ r r r r r Q s r c C sh t | t�s| jd�} tj| �}y&ttj|���}|j� W d Q R X W n tj k
rb t
t ��Y nX d S )Nzutf-8) �
isinstance�bytes�encoder �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" |j 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| j�r:|| j j|� n
||| j< d S )Nr )r �getr2 �update)r �
parametersr Zzip_file_paramr r r �
add_to_paramsr s z ZipFileArgument.add_to_params)r% r& r' r( r r9 �
__classcell__r r )r5 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 )r3 r" r � _cli_namer2 )r r4 r )r5 r r r � s z ReplacedZipFileArgument.__init__c C sX |d kr d S | j |�}d|kr,td| j ��|j| j�rJ|| j 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_argumentr. r; r6 r2 r7 )r r8 r Zunpackedr r r r9 � s
z%ReplacedZipFileArgument.add_to_params)r% r& r' r( r r9 r: r r )r5 r r" } s r" )r- r �
contextlibr Zbotocore.vendoredr Zawscli.argumentsr r r/ r r
r
�objectr r r r r" r r r r �<module>
s