o .&ßa<ã@sDddlmZddlmZdd„Zdd„Zdd„ZGd d „d eƒZdS)é)ÚShape)ÚBaseCLIArgumentcKs,t|ƒrt|ƒ}t|||dd|d<dSdS)NÚoutfile)Úresponse_keyÚoperation_modelÚsessionÚname)Ú_has_streaming_outputÚ_get_streaming_argument_nameÚStreamingOutputArgument)Úargument_tablerrÚkwargsÚstreaming_argument_name©rúJ/usr/lib/python3/dist-packages/awscli/customizations/streamingoutputarg.pyÚadd_streaming_output_argsýþrcCó|jS©N)Úhas_streaming_output©Úmodelrrrr sr cCs|jjdS)NÚpayload)Úoutput_shapeÚ serializationrrrrr "sr c@sveZdZdZdZ ddd„Zedd„ƒZedd „ƒZed d„ƒZ e j dd„ƒZ ed d„ƒZdd„Zdd„Z dd„ZdS)ri€z(Filename where the content will be savedNcCsR||_tdddiƒ|_|dur|j}||_||_d|_||_d|_||_||_ dS)NrÚtypeÚstringT) Ú_namerÚargument_modelÚBUFFER_SIZEÚ_buffer_sizeÚ _response_keyÚ_output_fileÚ _requiredÚ_operation_modelÚ_session)ÚselfrrrrÚbuffer_sizerrrÚ__init__+sÿ z StreamingOutputArgument.__init__cCrr)r©r%rrrÚcli_name<sz StreamingOutputArgument.cli_namecCsdS)Nrrr(rrrÚ cli_type_nameCsz%StreamingOutputArgument.cli_type_namecCrr©r"r(rrrÚrequiredGóz StreamingOutputArgument.requiredcCs ||_dSrr+)r%Úvaluerrrr,Ks cCrr)ÚHELPr(rrrÚ documentationOr-z%StreamingOutputArgument.documentationcCs|j|j|j|jddS)N)ÚmetavarÚhelp)Úadd_argumentrÚpy_namer/)r%ÚparserrrrÚ add_to_parserSs ÿz%StreamingOutputArgument.add_to_parsercCs8||_|jjj ¡}|jj}|j d||f|j¡dS)Nzafter-call.%s.%s) r!r#Ú service_modelÚ service_idÚ hyphenizerr$ÚregisterÚ save_file)r%Ú parametersr.r8Úoperation_namerrrÚ add_to_paramsWsÿÿz%StreamingOutputArgument.add_to_paramscKs||j|vrdS||j}|j}t|jdƒ}| |¡}|r)| |¡| |¡}|sWdƒn1s3wY||j=dS)NÚwb)r rÚopenr!ÚreadÚwrite)r%Úparsedr Úbodyr&ÚfpÚdatarrrr;^s þ€þz!StreamingOutputArgument.save_filer)Ú__name__Ú __module__Ú__qualname__rr/r'Úpropertyr)r*r,Úsetterr0r6r>r;rrrrr&s$ ÿ rN)Úbotocore.modelrÚawscli.argumentsrrr r rrrrrÚs