File: //lib/python3.6/site-packages/awscli/customizations/__pycache__/s3events.cpython-36.pyc
3
�T_�
� @ sP d Z ddlmZ dZG dd� de�Zdd� Zdd � Zd
d � ZG d d
� d
e�Z dS )z+Add S3 specific event streaming output arg.� )�CustomArgumentz(Filename where the records will be savedc @ s e Zd ZdS )�DocSectionNotFoundErrorN)�__name__�
__module__� __qualname__� r r � /usr/lib/python3.6/s3events.pyr s r c C s | j dt� | jdt� d S )Nz3building-argument-table.s3api.select-object-contentz&doc-output.s3api.select-object-content)�register�add_event_stream_output_argZ
register_last�replace_event_stream_docs)Zevent_handlersr r r �register_event_stream_arg s r c K s$ t dtdd|jjd |d�| d< d S )NZoutfile�stringTZpayload)�nameZ help_textZ
cli_type_nameZpositional_arg�
stream_key�session)� S3SelectStreamOutputArgument�STREAM_HELP_TEXTZ output_shapeZ
serialization)Zargument_tableZoperation_modelr �kwargsr r r r
" s
r
c K s^ | j }d}x:|dkrDy |j� }W q tk
r@ td| ��Y q X q W |jd� |jd� d S )N� z======
Output
======z7Could not find the "output" section for the command: %sz======
Output
======
zdThis command generates no output. The selected object content is written to the specified outfile.
)�docZ pop_write�
IndexErrorr �write)Z help_commandr r Zcurrentr r r r + s
r c s0 e Zd ZdZ� fdd�Zdd� Zdd� Z� ZS )r Tc s( t t| �jf |� || _d | _|| _d S )N)�superr �__init__� _stream_key� _output_file�_session)�selfr r r )� __class__r r r ? s z%S3SelectStreamOutputArgument.__init__c C s || _ | jjd| j� d S )Nz!after-call.s3.SelectObjectContent)r r r � save_file)r Z
parameters�valuer r r �
add_to_paramsG s z*S3SelectStreamOutputArgument.add_to_paramsc K sd | j |krd S || j }t| jd��.}x&|D ] }d|kr,|j|d d � q,W W d Q R X || j = d S )N�wbZRecordsZPayload)r �openr r )r Zparsedr Z event_stream�fpZeventr r r r L s
z&S3SelectStreamOutputArgument.save_file)r r r Z_DOCUMENT_AS_REQUIREDr r! r �
__classcell__r r )r r r < s r N)
�__doc__Zawscli.argumentsr r � Exceptionr r r
r r r r r r �<module>
s