File: //opt/alt/python37/lib64/python3.7/site-packages/aiohttp/__pycache__/client_proto.cpython-37.pyc
B
�k�a � @ s� d dl Z d dlmZ d dlmZmZmZ ddlmZ ddl m
Z
m Z m Z m
Z
ddlmZ ddlmZmZ dd lmZmZmZ G d
d � d eeeeef �ZdS ) � N)�suppress)�Any�Optional�Tuple� )� BaseProtocol)�
ClientOSError�ClientPayloadError�ServerDisconnectedError�ServerTimeoutError)�BaseTimerContext)�HttpResponseParser�RawResponseMessage)�
EMPTY_PAYLOAD� DataQueue� StreamReaderc sP e Zd ZdZejdd�dd�Zeed�dd��Z eed�d d
��Z
dd�d d �Z dd�d
d�Z ed�dd�Z
ee dd�� fdd� Zdd�dd�Zdd�� fdd� Zdd�� fdd� Zedd�� fdd� Zeedd �d d �Zdd d d dd!d"�ee eeeee edd#�d$d%�Zdd�d&d'�Zdd�d(d)�Z dd�d*d+�Z e dd,�d-d.�Z � Z S )/�ResponseHandlerz8Helper class to adapt between Protocol and StreamReader.N)�loop�returnc C sZ t j| |d� t�| |� d| _d | _d| _d | _d | _d| _d| _ d | _
d | _ d | _ d S )N)r F� )
r �__init__r �
_should_close�_payload�
_skip_payload�_payload_parserZ_timer�_tail� _upgraded�_parser�
_read_timeout�_read_timeout_handle)�selfr � r! �G/opt/alt/python37/lib64/python3.7/site-packages/aiohttp/client_proto.pyr s zResponseHandler.__init__)r c C s | j S )N)r )r r! r! r" �upgraded' s zResponseHandler.upgradedc C sV | j d k r| j �� r| jr dS | jpT| jpT| �� d k pT| jd k pTt| �dkpTt| j�S )NTr ) r �is_eofr r � exceptionr �len�boolr )r r! r! r" � should_close+ s
z ResponseHandler.should_closec C s
d| _ d S )NT)r )r r! r! r" � force_close9 s zResponseHandler.force_closec C s. | j }|d k r*|�� d | _ d | _| �� d S )N)� transport�closer �
_drop_timeout)r r* r! r! r" r+ < s zResponseHandler.closec C s | j d k o| j �� S )N)r* Z
is_closing)r r! r! r" � is_connectedD s z ResponseHandler.is_connected)�excr c s� | � � | jd k r0tt�� | j�� W d Q R X d }| jd k r|y| j�� }W n. tk
rz | jd k rv| j�td�� Y nX | � � s�t
|t �r�t |j
� }|d kr�t|�}| �|� d| _d | _d | _d | _d| _t� �|� d S )Nz!Response payload is not completedTF)r, r r � ExceptionZfeed_eofr r �
set_exceptionr r$ �
isinstance�OSErrorr �argsr
r Z_reading_paused�super�connection_lost)r r. Z uncompleted)� __class__r! r" r5 G s0
z ResponseHandler.connection_lostc C s | � � d S )N)r, )r r! r! r" � eof_receivedi s z ResponseHandler.eof_receivedc s t � �� | �� d S )N)r4 �
pause_readingr, )r )r6 r! r" r8 m s
z ResponseHandler.pause_readingc s t � �� | �� d S )N)r4 �resume_reading�_reschedule_timeout)r )r6 r! r" r9 q s
z ResponseHandler.resume_readingc s d| _ | �� t� �|� d S )NT)r r, r4 r0 )r r. )r6 r! r" r0 u s z ResponseHandler.set_exception)�parser�payloadr c C s6 || _ || _| �� | jr2| jd }| _| �|� d S )Nr )r r r, r �
data_received)r r; r< �datar! r! r" �
set_parserz s zResponseHandler.set_parserFTi )�timer� skip_payload�read_until_eof�auto_decompress� read_timeout� read_bufsize)r@ rA rB rC rD rE r c
C sT || _ || _| �� t| | j||t| ||d�| _| jrP| jd }| _| �|� d S )N)r@ Zpayload_exceptionZresponse_with_bodyrB rC r ) r r r: r
�_loopr r r r= )r r@ rA rB rC rD rE r> r! r! r" �set_response_params� s
z#ResponseHandler.set_response_paramsc C s | j d k r| j �� d | _ d S )N)r �cancel)r r! r! r" r, � s
z ResponseHandler._drop_timeoutc C s< | j }| jd k r| j�� |r2| j�|| j�| _nd | _d S )N)r r rH rF Z
call_later�_on_read_timeout)r �timeoutr! r! r" r: � s
z#ResponseHandler._reschedule_timeoutc C s, t d�}| �|� | jd k r(| j�|� d S )Nz#Timeout on reading data from socket)r r0 r )r r. r! r! r" rI � s
z ResponseHandler._on_read_timeout)r> r c
C sv | � � |sd S | jd k rL| j�|�\}}|rHd | _d | _|rH| �|� d S | js\| jd krn| j|7 _�ny| j�|�\}}}W n@ tk
r� } z"| j d k r�| j �
� | � |� d S d }~X Y nX || _d }xX|D ]P\}}|j r�d| _
|| _| j�s|jdk�r| �|tfd� q�| �||fd� q�W |d k �rT|tk �rL|�| j� n| �� |�rr|�rl| �|� n|| _d S )NT)�� i0 r )r: r Z feed_datar r= r r r �
BaseExceptionr* r+ r0 r( r r �coder Zon_eofr, ) r r> �eof�tail�messagesr# r. r< �messager! r! r" r= � sJ
z ResponseHandler.data_received)!�__name__�
__module__� __qualname__�__doc__�asyncioZAbstractEventLoopr �propertyr' r# r( r) r+ r- r rL r5 r7 r8 r9 r0 r r? r �float�intrG r, r: rI �bytesr= �
__classcell__r! r! )r6 r" r s4
" r )rV �
contextlibr �typingr r r Z
base_protocolr Zclient_exceptionsr r r
r Zhelpersr �httpr
r Zstreamsr r r r r! r! r! r" �<module> s