U
'�7`� � @ 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/python38/lib64/python3.8/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 r# )� transport�closer �
_drop_timeout)r r, r! r! r" r- < s zResponseHandler.closec C s | j d k o| j �� S r# )r, Z
is_closingr$ r! r! r" � is_connectedD s z ResponseHandler.is_connected)�excr c s� | � � | jd k r0tt�� | j�� W 5 Q R X d }| jd k r|z| 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 r0 Z uncompleted�� __class__r! r" r7 G s2
�
z ResponseHandler.connection_lostc C s | � � d S r# )r. r$ r! r! r" � eof_receivedi s z ResponseHandler.eof_receivedc s t � �� | �� d S r# )r6 �
pause_readingr. r$ r8 r! r" r; m s
z ResponseHandler.pause_readingc s t � �� | �� d S r# )r6 �resume_reading�_reschedule_timeoutr$ r8 r! r" r< q s
z ResponseHandler.resume_readingc s d| _ | �� t� �|� d S r* )r r. r6 r2 �r r0 r8 r! r" r2 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)rD rE rF rG rH rI r c
C sT || _ || _| �� t| | j||t| ||d�| _| jrP| jd }| _| �|� d S )N)rD Zpayload_exceptionZresponse_with_bodyrF rG r ) r r r= r
�_loopr r r rA )r rD rE rF rG rH rI rB r! r! r" �set_response_params� s
� z#ResponseHandler.set_response_paramsc C s | j d k r| j �� d | _ d S r# )r �cancelr$ r! r! r" r. � s
z ResponseHandler._drop_timeoutc C s< | j }| jd k r| j�� |r2| j�|| j�| _nd | _d S r# )r r rL rJ Z
call_later�_on_read_timeout)r Ztimeoutr! 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 r2 r r> r! r! r" rM � s
z ResponseHandler._on_read_timeout)rB r c
C sx | � � |sd S | jd k rL| j�|�\}}|rHd | _d | _|rH| �|� d S | js\| jd krn| j|7 _�nz| j�|�\}}}W nF tk
r� } z(| j d k r�| j �
� | � |� W Y �d S d }~X Y nX || _d }|D ]P\}}|j r�d| _
|| _| j�s|jdk�r| �|tfd� q�| �||fd� q�|d k �rV|tk �rN|�| j� n| �� |�rt|�rn| �|� n|| _d S )NT)�� i0 r )r= r Z feed_datar rA r r r �
BaseExceptionr, r- r2 r) r r �coder Zon_eofr. ) r rB �eof�tailZmessagesr% r0 r@ �messager! r! r" rA � sJ
z ResponseHandler.data_received)!�__name__�
__module__� __qualname__�__doc__�asyncioZAbstractEventLoopr �propertyr( r% r) r+ r- r/ r rO r7 r: r; r< r2 r rC r �float�intrK r. r= rM �bytesrA �
__classcell__r! r! r8 r" r sD
"�� r )rX �
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