o !d<@sDddlmZddlmZddZddZddZGd d d eZd S) )Shape)BaseCLIArgumentcKs,t|rt|}t|||dd|d<dSdS)Noutfile) response_keyoperation_modelsessionname)_has_streaming_output_get_streaming_argument_nameStreamingOutputArgument)Zargument_tablerrkwargsZstreaming_argument_namer Lusr/lib/python3.10/site-packages/awscli/customizations/streamingoutputarg.pyadd_streaming_output_argsrcC|jSN)Zhas_streaming_outputmodelr r rr sr cCs |jjdS)Npayload)Z output_shapeZ serializationrr r rr "s r c@sveZdZdZdZ dddZeddZedd Zed d Z e j d d Z ed dZ ddZ ddZ ddZdS)r iz(Filename where the content will be savedNcCsR||_tdddi|_|dur|j}||_||_d|_||_d|_||_||_ dS)Nr typestringT) _namerZargument_model BUFFER_SIZE _buffer_size _response_key _output_file _required_operation_model_session)selfrrrr buffer_sizer r r__init__+s z StreamingOutputArgument.__init__cCrr)rrr r rcli_name<sz StreamingOutputArgument.cli_namecCsdS)Nrr r"r r r cli_type_nameCsz%StreamingOutputArgument.cli_type_namecCrrrr"r r rrequiredGz StreamingOutputArgument.requiredcCs ||_dSrr%)rvaluer r rr&Ks cCrr)HELPr"r r r documentationOr'z%StreamingOutputArgument.documentationcCs|j|j|j|jddS)N)metavarhelp) add_argumentrZpy_namer))rparserr r r add_to_parserSs  z%StreamingOutputArgument.add_to_parsercCs8||_|jjj}|jj}|jd||f|jdS)Nzafter-call.%s.%s) rrZ service_model service_idZ hyphenizerrregister save_file)r parametersr(r0Zoperation_namer r r add_to_paramsWsz%StreamingOutputArgument.add_to_paramscKs||j|vrdS||j}|j}t|jd}||}|r)||||}|sWdn1s3wY||j=dS)Nwb)rropenrreadwrite)rparsedr bodyr fpdatar r rr2^s      z!StreamingOutputArgument.save_filer)__name__ __module__ __qualname__rr)r!propertyr#r$r&setterr*r/r4r2r r r rr &s$       r N)Zbotocore.modelrZawscli.argumentsrrr r r r r r rs