File: //lib/python3/dist-packages/awscli/customizations/__pycache__/s3events.cpython-38.pyc
U
<�^�
� @ 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/dist-packages/awscli/customizations/s3events.pyr s r c C s | � dt� | �dt� 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 sZ | j }d}|dkrBz |�� }W q
tk
r> td| ��Y q
X q
|�d� |�d� 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 || _ | j�d| 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 s` | j |krd S || j }t| jd��*}|D ] }d|kr*|�|d d � q*W 5 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