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)AnyOptionalTuple ) BaseProtocol)
ClientOSErrorClientPayloadErrorServerDisconnectedErrorServerTimeoutError)BaseTimerContext)HttpResponseParserRawResponseMessage)
EMPTY_PAYLOAD DataQueue StreamReaderc sP e Zd ZdZejddddZeedddZ eedd d
Z
ddd d Z ddd
dZ edddZ
ee dd fdd ZddddZdd fdd Zdd fdd Zedd fdd Zeedd d d Zdd d d dd!d"ee eeeee edd#d$d%Zddd&d'Zddd(d)Z ddd*d+Z e dd,d-d.Z Z S )/ResponseHandlerz8Helper class to adapt between Protocol and StreamReader.N)loopreturnc 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| jS )NTr ) r is_eofr r Z exceptionr lenboolr 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# ) transportcloser
_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| jtd Y nX | st
|t rt |j
}|d krt|}| | d| _d | _d | _d | _d| _t | d S )Nz!Response payload is not completedTF)r. r r ExceptionZfeed_eofr r
set_exceptionr r&
isinstanceOSErrorr argsr
r Z_reading_pausedsuperconnection_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)parserpayloadr 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_payloadread_until_eofauto_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 rd| _
|| _| js|jdkr| |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 eoftailZmessagesr% 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 floatintrK 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