o !d_@sddlZddlZddlmZddlmZddlmZmZdZ dZ ddZ d d Z Gd d d e Zd dZddZGdddeZGdddeZdS)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

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

cCs@|dtdj|dtdj|dt|dtdS)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)registerZipFileArgumentHoisterhoist_modify_zipfile_docstringvalidate_is_zip_file)clir Cusr/lib/python3.10/site-packages/awscli/customizations/awslambda.pyregister_lambda_create_function!srcKs|jdkr t|dSdSNzip-file)name_should_contain_zip_content)Z cli_argumentvaluekwargsr r r r ,s  r c@s eZdZdZddZddZdS)rarHoists 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. Replaces the old ZipFile argument with an instance of ReplacedZipFileArgument to prevent its usage and recommend the new top-level injected parameter. cCs||_||_dS)N)_serialized_namelower_name)selfserialized_namer r r __init__:szZipFileArgumentHoister.__init__cKsltj|jd}td|d|jd|d<||j}t|j}|jd=t |j||j d| d|jd||j<dS) N)Z param_typerZblob) help_textZ cli_type_namerZipFileF event_emitter)rargument_modelZoperation_modelZ is_requiredrr) ZIP_DOCSTRINGformatrZipFileArgumentrcopydeepcopyrmembersReplacedZipFileArgumentZ_operation_modelZ get_component)rsessionargument_tablerrZargumentmodelr r r r>s   zZipFileArgumentHoister.hoistN)__name__ __module__ __qualname____doc__rrr r r r r1s rcKsd|vr t|d_dSdSr)r documentation)r&r'rr r r r Qsr cCsxt|ts |d}t|}z tt|}|WdWdS1s(wYWdStj y;t t w)Nzutf-8) isinstancebytesencoderBytesIOrzipfilerinfolist BadZipfile ValueError ERROR_MSG)rfileobjfr r r rVs    &rc(eZdZdZfddZddZZS)r!aA 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 serialized as { 'Code': { 'ZipFile': } }. cs&|d|_tt|j|i|dS)Nr)pop_param_to_replacesuperr!rrargsr __class__r r rns zZipFileArgument.__init__cCsJ|durdSt|d|i}||jr||j|dS|||j<dS)Nr)rgetr;update)r parametersrZzip_file_paramr r r add_to_paramsrs zZipFileArgument.add_to_paramsr)r*r+r,rrD __classcell__r r r?r r!es r!cr9)r%aA 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. cs2tt|j|i|d|d|_|d|_dS)Nz--%srr)r<r%r _cli_namer;r=r?r r rsz ReplacedZipFileArgument.__init__cCsZ|durdS||}d|vrtd|j||jr&||j|dS|||j<dS)NrzyZipFile cannot be provided as part of the %s argument. Please use the '--zip-file' option instead to specify a zip file.)Z_unpack_argumentr5rGrAr;rB)rrCrZunpackedr r r rDs  z%ReplacedZipFileArgument.add_to_paramsrEr r r?r r%}s r%)r2r" contextlibrZbotocore.vendoredrZawscli.argumentsrrr6rrr objectrr rr!r%r r r r s