File: //opt/cppython/lib/python3.8/site-packages/botocore/__pycache__/response.cpython-38.pyc
U
KD�g; � @ s� d dl Z d dlmZ d dlmZ d dlmZ d dlm Z d dl
m Z d dl m
Z
mZmZ d dlmZ d d l
mZ d d
lmZ e �e�ZG d d � d e�Zd
d� ZdS )� N)�IOBase)�
ProtocolError)�ReadTimeoutError)�parsers)�set_socket_timeout)�IncompleteReadErrorr �ResponseStreamingError)� ScalarTypes)�
XMLParseError)�first_non_none_responsec @ s� e Zd ZdZdZdd� Zdd� Zdd� Zd d
� Zd#d d
�Z dd� Z
dd� Z dd� Z dd� Z
dd� Ze Zedfdd�Zefdd �Zd d � Zd d � Zd!d"� Zd S )$�
StreamingBodyae Wrapper class for an http response body.
This provides a few additional conveniences that do not exist
in the urllib3 model:
* Set the timeout on the socket (i.e read() timeouts)
* Auto validation of content length, if the amount of bytes
we read does not match the content length, an exception
is raised.
i c C s || _ || _d| _d S )Nr )� _raw_stream�_content_length� _amount_read)�selfZ
raw_stream�content_length� r �>/opt/cppython/lib/python3.8/site-packages/botocore/response.py�__init__5 s zStreamingBody.__init__c C s d S �Nr �r r r r �__del__: s zStreamingBody.__del__c C s: zt | j|� W n$ tk
r4 tjddd� � Y nX dS )z&Set the timeout seconds on the socket.zbCannot access the socket object of a streaming response. It's possible the interface has changed.T)�exc_infoN)r r
�AttributeError�logger�error)r �timeoutr r r r @ s �z StreamingBody.set_socket_timeoutc C s( z | j �� W S tk
r" Y dS X d S )NF)r
�readabler r r r r r W s zStreamingBody.readableNc
C s� z| j �|�}W nZ tk
r@ } zt|j|d��W 5 d}~X Y n, tk
rj } zt|d��W 5 d}~X Y nX | jt|�7 _|dks�|s�|dkr�| � � |S )zhRead at most amt bytes from the stream.
If the amt argument is omitted, read all data.
)Z endpoint_urlr N)r r )
r
�read�URLLib3ReadTimeoutErrorr �url�URLLib3ProtocolErrorr r �len�_verify_content_length)r �amt�chunk�er r r r ] s zStreamingBody.readc C s
| j �� S r )r
� readlinesr r r r r' q s zStreamingBody.readlinesc C s | � | j�S )z:Return an iterator to yield 1k chunks from the raw stream.)� iter_chunks�_DEFAULT_CHUNK_SIZEr r r r �__iter__t s zStreamingBody.__iter__c C s | � | j�}|r|S t� �dS )z-Return the next 1k chunk from the raw stream.N)r r) �
StopIteration)r �
current_chunkr r r �__next__x s zStreamingBody.__next__c C s | j S r )r
r r r r � __enter__ s zStreamingBody.__enter__c C s | j �� d S r �r
�close)r �type�value� tracebackr r r �__exit__� s zStreamingBody.__exit__Fc c sd d}| � |�D ]<}|| �d�}|dd� D ]}|�|�d V q,|d }q|r`|�|�d V dS )z�Return an iterator to yield lines from the raw stream.
This is achieved by reading chunk of bytes (of size chunk_size) at a
time from the raw stream, and then yielding lines from there.
� TN���r )r( �
splitlines)r �
chunk_size�keepends�pendingr% �lines�liner r r �
iter_lines� s
zStreamingBody.iter_linesc c s | � |�}|dkrq |V q dS )z\Return an iterator to yield chunks of chunk_size bytes from the raw
stream.
r5 N)r )r r8 r, r r r r( � s
zStreamingBody.iter_chunksc C s2 | j d k r.| jt| j �kr.t| jt| j �d��d S )N)Z actual_bytesZexpected_bytes)r r �intr r r r r r# � s ��z$StreamingBody._verify_content_lengthc C s
| j �� S r )r
�tellr r r r r? � s zStreamingBody.tellc C s | j �� dS )z*Close the underlying http response stream.Nr/ r r r r r0 � s zStreamingBody.close)N)�__name__�
__module__� __qualname__�__doc__r) r r r r r r' r* r- r. r4 �nextr= r( r# r? r0 r r r r r & s$
r c C sx | j d }|j|jd�}|d dkr0|j|d< n,| jrRt|j|d �d��|d< n
|j|d< t� |�}||�
|| j �fS )N�protocol)�headers� status_coderG i, �bodyrF zcontent-length) �metadatarF rG �contentZhas_streaming_outputr �raw�getr Z
create_parser�parseZ output_shape)Zoperation_model�
http_responserE Z
response_dict�parserr r r � get_response� s"
� �
�rP )�logging�ior Zurllib3.exceptionsr r! r r Zbotocorer Zbotocore.compatr Zbotocore.exceptionsr r r r
Zbotocore.hooksr � getLoggerr@ r r rP r r r r �<module> s