File: //lib/python3/dist-packages/awscli/customizations/__pycache__/s3events.cpython-310.pyc
o
.&�a�
� @ 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_arg�
register_last�replace_event_stream_docs)�event_handlersr r r �register_event_stream_arg s ��r c K s$ t dtdd|jjd |d�| d< d S )N�outfile�stringT�payload)�name� help_text�
cli_type_name�positional_arg�
stream_key�session)� S3SelectStreamOutputArgument�STREAM_HELP_TEXT� output_shape�
serialization)�argument_table�operation_modelr �kwargsr r r r
" s
�r
c K sZ | j }d}|dkr!z|�� }W n
ty td| ��w |dks |�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.
)�doc� pop_write�
IndexErrorr �write)� help_commandr r �currentr r r r + s ����
r c s0 e Zd ZdZ� fdd�Zdd� Zdd� Z� ZS )r Tc s, t t| �jdi |�� || _d | _|| _d S )Nr )�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+ �
parameters�valuer r r �
add_to_paramsG s �z*S3SelectStreamOutputArgument.add_to_paramsc K st | j |vrd S || j }t| jd��}|D ]}d|v r$|�|d d � qW d � n1 s/w Y || j = d S )N�wb�Records�Payload)r( �openr) r# )r+ �parsedr � event_stream�fp�eventr r r r. L s
�� � z&S3SelectStreamOutputArgument.save_file)r r r �_DOCUMENT_AS_REQUIREDr' r1 r. �
__classcell__r r r, r r < s
r N)
�__doc__�awscli.argumentsr r � Exceptionr r r
r r r r r r �<module> s