o

<&�aE�@s�dd
lZddl
mZ
ddlmZmZmZ
ddlmZ
ddlm	Z

ddlmZ
ddlm
Z

ddl
mZ
dd	lmZ
e�e�
ZGd
d�de�Zdd
�Zd
S)�N)
�set_socket_timeout)�IncompleteReadError�ReadTimeoutError�ResponseStreamingError)
r)
�
ProtocolError)
�parsers)
�ScalarTypes)
�
XMLParseError)
�first_non_none_responsec@sheZ
dZd
ZdZdd�Zdd�Zddd	�
Zd
d�Zdd
�Z	e	Z
ddd�
Zef
dd�
Zdd�Z
dd�ZdS)�
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.

    �cCs|
|_||_
d
|_dS)Nr
)�_raw_stream�_content_length�_amount_read)�self�
raw_stream�content_length�r�3/usr/lib/python3/dist-packages/botocore/response.py�__init__0s



zStreamingBody.__init__cCs2z	t|j
|
�
WdSty


tjd
dd�
�w)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)rr
�AttributeError�logger�error)r�timeoutrrrr5s

��z StreamingBody.set_socket_timeoutNc
Cs�z|j�
|
�
}Wn#ty
}
zt|j|d
��
d}~w
ty+
}
zt|d�
�
d}~w
w|jt|�
7_|
dus?|sC|
dkrC|�	�
|S)zhRead at most amt bytes from the stream.

        If the amt argument is omitted, read all data.
        )�endpoint_urlrN)
rr
)
r
�read�URLLib3ReadTimeoutErrorr�url�URLLib3ProtocolErrorrr�len�_verify_content_length)r�amt�chunk�
errrrLs

�


��

zStreamingBody.readc

Cs|�|j
�
S)
zCReturn an iterator to yield 1k chunks from the raw stream.
        )�iter_chunks�_DEFAULT_CHUNK_SIZE�
rrrr�__iter__`szStreamingBody.__iter__c
Cs|�|j
�
}
|
r
|
St��
)
z6Return the next 1k chunk from the raw stream.
        )rr&�
StopIteration)r�
current_chunkrrr�__next__es


zStreamingBody.__next__Fccsj�d
}|�|
�
D]}||�
d�
}|dd�D]
}|�
|�
dV
q|d}q|r3|�
|�
dV
dSdS)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�linerrr�
iter_linesos�







�zStreamingBody.iter_linesccs"�	|�|
�
}|dkr
dS|V
q)z\Return an iterator to yield chunks of chunk_size bytes from the raw
        stream.
        Tr,N)
r)rr/r*rrrr%~s�




�zStreamingBody.iter_chunksc

Cs6|jdu
r|j
t|j�
krt|j
t|j�
d
��
dSdS)N)�actual_bytes�expected_bytes)rr�intrr'rrrr!�s




��
z$StreamingBody._verify_content_lengthc

Cs|j�
�
d
S)z*Close the underlying http response stream.N)r
�closer'rrrr8�szStreamingBody.close)
N)rF)�__name__�
__module__�__qualname__�__doc__r&rrrr(r+�nextr4r%r!r8rrrrr"s




rcCsx|jd
}|
j
|
jd�}|ddkr|
j|d<n|jr)t|
j|d�d�
�|d<n|
j|d<t�	|�
}|
|�
||j�fS)N�protocol)�headers�status_coder@i,
�bodyr?zcontent-length)�metadatar?r@�content�has_streaming_outputr�raw�getr�
create_parser�parse�output_shape)�operation_model�
http_responser>�
response_dict�parserrrr�get_response�s


�




�



�rN)�logging�botocore.compatr�botocore.exceptionsrrr�urllib3.exceptionsrrr�botocorerrr	�botocore.hooksr
�	getLoggerr9r�objectrrNrrrr�<module>
s





u