File: //lib/python3/dist-packages/awscli/customizations/__pycache__/streamingoutputarg.cpython-38.pyc
U
<�^< � @ sD d dl mZ d dlmZ dd� Zdd� Zdd� ZG d d
� d
e�Zd S ) � )�Shape)�BaseCLIArgumentc K s( t |�r$t|�}t|||dd�| d< d S )NZoutfile)� response_key�operation_model�session�name)�_has_streaming_output� _get_streaming_argument_name�StreamingOutputArgument)Zargument_tabler r �kwargsZstreaming_argument_name� r �J/usr/lib/python3/dist-packages/awscli/customizations/streamingoutputarg.py�add_streaming_output_arg s �r c C s | j S �N)Zhas_streaming_output�Zmodelr r r
r s r c C s | j jd S )NZpayload)Z output_shapeZ
serializationr r r r
r " s r c @ st e Zd ZdZdZddd�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 kr$| j}|| _|| _d | _|| _ d| _|| _|| _ d S )Nr
�type�stringT)
�_namer Zargument_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 s | j S 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 s | j S r �r r r r r
�requiredG s z StreamingOutputArgument.requiredc C s
|| _ d S r r! )r �valuer r r
r" K s c C s | j S r )�HELPr r r r
�
documentationO s z%StreamingOutputArgument.documentationc C s |j | j| j| jd� d S )N)�metavar�help)� add_argumentr Zpy_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 Z
service_model�
service_idZ hyphenizer r �register� save_file)r Z
parametersr# r+ Zoperation_namer r r
�
add_to_paramsW s ��z%StreamingOutputArgument.add_to_paramsc K sf | j |krd S || j }| j}t| jd��*}|�|�}|rP|�|� |�|�}q6W 5 Q R X || j = d S )N�wb)r r �openr �read�write)r Zparsedr Zbodyr �fp�datar r r
r- ^ s
z!StreamingOutputArgument.save_file)N)�__name__�
__module__� __qualname__r r$ r �propertyr r r" �setterr% r* r. r- r r r r
r
& s" �
r
N)Zbotocore.modelr Zawscli.argumentsr r r r r
r r r r
�<module>
s