Edit File: 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