File: //usr/local/CyberCP/lib64/python3.10/site-packages/botocore/__pycache__/response.cpython-310.pyc
o
�h; � @ 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)�self�
raw_stream�content_length� r �D/usr/local/CyberCP/lib/python3.10/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 s2 z t | j|� W dS ty tjddd� � 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)r r
�AttributeError�logger�error)r �timeoutr r r r @ s ��z StreamingBody.set_socket_timeoutc C s" z| j �� W S ty Y dS w )NF)r
�readabler r r r r r W s
�zStreamingBody.readableNc
C s� z| j �|�}W n# ty } zt|j|d��d}~w ty+ } zt|d��d}~ww | jt|�7 _|du s?|sC|dkrC| � � |S )zhRead at most amt bytes from the stream.
If the amt argument is omitted, read all data.
)� 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 �
| j �� S r )r
� readlinesr r r r r* q �
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� �)z-Return the next 1k chunk from the raw stream.)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 sj � d}| � |�D ] }|| �d�}|dd� D ]
}|�|�d V q|d }q|r3|�|�d V dS 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" � | � |�}|dkr
dS |V q)z\Return an iterator to yield chunks of chunk_size bytes from the raw
stream.
Tr9 N)r )r r< r0 r r r r, � s �
�zStreamingBody.iter_chunksc C s6 | j d ur| jt| j �krt| jt| j �d��d S d S )N)� actual_bytes�expected_bytes)r r �intr r r r r r% � s ���z$StreamingBody._verify_content_lengthc C r) r )r
�tellr r r r rE � r+ zStreamingBody.tellc C s | j �� dS )z*Close the underlying http response stream.Nr3 r r r r r4 � s zStreamingBody.closer )�__name__�
__module__� __qualname__�__doc__r- r r r r r r* r. r1 r2 r8 �nextrA r, r% rE r4 r r r r r & s&
r c C sx | j d }|j|jd�}|d dkr|j|d< n| jr)t|j|d �d��|d< n|j|d< t� |�}||�
|| j �fS )N�protocol)�headers� status_coderM i, �bodyrL zcontent-length) �metadatarL rM �content�has_streaming_outputr �raw�getr �
create_parser�parse� output_shape)�operation_model�
http_responserK �
response_dict�parserr r r � get_response� s
�
�
�r[ )�logging�ior �urllib3.exceptionsr r# r r! �botocorer �botocore.compatr �botocore.exceptionsr r r r
�botocore.hooksr � getLoggerrF r r r[ r r r r �<module> s