File: //lib/python3/dist-packages/awscli/customizations/__pycache__/streamingoutputarg.cpython-310.pyc
o
.&�a< � @ sD d dl mZ d dlmZ dd� Zdd� Zdd� ZG d d
� d
e�Zd S ) � )�Shape)�BaseCLIArgumentc K s, t |�rt|�}t|||dd�| d< d S d S )N�outfile)� response_key�operation_model�session�name)�_has_streaming_output� _get_streaming_argument_name�StreamingOutputArgument)�argument_tabler r �kwargs�streaming_argument_name� r �J/usr/lib/python3/dist-packages/awscli/customizations/streamingoutputarg.py�add_streaming_output_arg s ��r c C � | j S �N)�has_streaming_output��modelr r r r s r c C s | j jd S )N�payload)� output_shape�
serializationr r r r r
" s r
c @ sv e Zd ZdZdZ ddd�Zedd� �Zedd � �Zed
d � �Z e j
d d � �Z ed
d� �Z dd� Z dd� Z
dd� ZdS )r i � z(Filename where the content will be savedNc C sR || _ tdddi�| _|d u r| j}|| _|| _d | _|| _ d| _|| _|| _ d S )Nr �type�stringT)
�_namer �argument_model� BUFFER_SIZE� _buffer_size�
_response_key� _output_file� _required�_operation_model�_session)�selfr r r r � buffer_sizer r r �__init__+ s �
z StreamingOutputArgument.__init__c C r r )r �r% r r r �cli_name< s z StreamingOutputArgument.cli_namec C s dS )Nr r r( r r r �
cli_type_nameC s z%StreamingOutputArgument.cli_type_namec C r r �r" r( r r r �requiredG � z StreamingOutputArgument.requiredc C s
|| _ d S r r+ )r% �valuer r r r, K s
c C r r )�HELPr( r r r �
documentationO r- z%StreamingOutputArgument.documentationc C s |j | j| j| jd� d S )N)�metavar�help)� add_argumentr �py_namer/ )r% �parserr r r �
add_to_parserS s
�z%StreamingOutputArgument.add_to_parserc C s8 || _ | jjj�� }| jj}| j�d||f | j� d S )Nzafter-call.%s.%s) r! r# �
service_model�
service_id� hyphenizer r$ �register� save_file)r% �
parametersr. r8 �operation_namer r r �
add_to_paramsW s ��z%StreamingOutputArgument.add_to_paramsc K s| | j |vrd S || j }| j}t| jd��}|�|�}|r)|�|� |�|�}|s W d � n1 s3w Y || j = d S )N�wb)r r �openr! �read�write)r% �parsedr
�bodyr& �fp�datar r r r; ^ s
�� � z!StreamingOutputArgument.save_filer )�__name__�
__module__� __qualname__r r/ r' �propertyr) r* r, �setterr0 r6 r> r; r r r r r & s$
�
r N)�botocore.modelr �awscli.argumentsr r r r
r r r r r �<module> s