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 Z 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" r4 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)r3 �
pause_readingr+ )r )r5 r! r" r7 m s
z ResponseHandler.pause_readingc s t � �� | �� d S )N)r3 �resume_reading�_reschedule_timeout)r )r5 r! r" r8 q s
z ResponseHandler.resume_readingc s d| _ | �� t� �|� d S )NT)r r+ r3 r/ )r r- )r5 r! r" r/ 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? r@ rA rB rC rD r c
C sT || _ || _| �� t| | j||t| ||d�| _| jrP| jd }| _| �|� d S )N)r? Zpayload_exceptionZresponse_with_bodyrA rB r ) r r r9 r
�_loopr r r r< )r r? r@ rA rB rC rD 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 rG rE Z
call_later�_on_read_timeout)r Ztimeoutr! r! r" r9 � s
z#ResponseHandler._reschedule_timeoutc C s, t d�}| �|� | jd k r(| j�|� d S )Nz#Timeout on reading data from socket)r r/ r )r r- r! r! r" rH � 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 )r9 r Z feed_datar r< r r r �
BaseExceptionr) r* r/ r' r r �coder Zon_eofr+ ) r r= �eof�tailZmessagesr# r- r; �messager! r! r" r< � sJ
z ResponseHandler.data_received)!�__name__�
__module__� __qualname__�__doc__�asyncioZAbstractEventLoopr �propertyr& r# r' r( r* r, r rJ r4 r6 r7 r8 r/ r r> r �float�intrF r+ r9 rH �bytesr< �
__classcell__r! r! )r5 r" r s4
" r )rS �
contextlibr �typingr r r Z
base_protocolr Zclient_exceptionsr r r
r Zhelpersr Zhttpr
r Zstreamsr r r r r! r! r! r" �<module> s