File: //lib/python3.6/site-packages/awscli/customizations/__pycache__/streamingoutputarg.cpython-36.pyc
3
�T_< � @ 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 �(/usr/lib/python3.6/streamingoutputarg.py�add_streaming_output_arg s r c C s | j S )N)Zhas_streaming_output)�modelr r r
r s r c C s | j jd S )NZpayload)Z output_shapeZ
serialization)r 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 )N)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 )N)r )r r r r
�requiredG s z StreamingOutputArgument.requiredc C s
|| _ d S )N)r )r �valuer r r
r K s c C s | j S )N)�HELP)r 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 || _ | jjjj� }| jj}| jjd||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 sj | j |krd S || j }| j}t| jd��.}|j|�}x |rR|j|� |j|�}q8W W d 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