Edit File: client_proto.cpython-37.opt-1.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 �timeoutr! 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 rK r4 r6 r7 r8 r/ r r> r �float�intrF r+ r9 rH �bytesr< � __classcell__r! r! )r5 r" r s4 " r )rT � 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