Z @ s d d l Z d d l Z d d l m Z d d l m Z m Z m Z d d l m Z d d l
m Z m Z Gd d d e e j
j
Z d S) N)suppress )
ClientOSErrorClientPayloadErrorServerDisconnectedError)HttpResponseParser)
EMPTY_PAYLOAD DataQueuec
s e Z d Z d Z d d d d Z e d d Z e d d Z d
d Z d d
Z d d Z
f d d Z d d Z d d Z
d d Z f d d Z d d Z d d d d d d d d! d" d# Z d$ d% Z S)&ResponseHandlerz8Helper class to adapt between Protocol and StreamReader.loopNc C s t j j j | d | t j | d | d | _ d | _ d | _ d | _ d | _ d | _
d | _ d | _ d | _
d | _ d | _ d S)Nr F )asynciostreamsFlowControlMixin__init__r transport
_should_close_message_payload
_skip_payload_payload_parser_reading_pausedZ_timer_tail _upgraded_parser)selfr r /client_proto.pyr s zResponseHandler.__init__c C s | j S)N)r )r r r r upgraded! s zResponseHandler.upgradedc C sr | j d k r | j j s( | j r, d S| j pq | j pq | j d k pq | j d k pq t | pq | j S)NT)r is_eofr r exceptionr lenr )r r r r should_close% s z ResponseHandler.should_closec C s5 | j } | d k r1 | j d | _ d | _ | S)N)r closer )r r r r r r# 0 s
zResponseHandler.closec C s
| j d k S)N)r )r r r r is_connected8 s z ResponseHandler.is_connectedc C s
| | _ d S)N)r )r r r r r connection_made; s z ResponseHandler.connection_madec s% | j d k r0 t t | j j Wd QRXy | j j } Wn= t k
r d } | j d k r~ | j j t d Yn X| j s t | t
r t | j } | d k r t
| } | j | d | _ d | _ d | _ d | _ d | _ d | _ d | _ t j | d S)Nz!Response payload is not completedTF)r r ExceptionZfeed_eofr r
set_exceptionr r
isinstanceOSErrorr argsr r r r r superconnection_lost)r excZ uncompleted) __class__r r r, > s0
z ResponseHandler.connection_lostc C s d S)Nr )r r r r eof_received^ s z ResponseHandler.eof_receivedc
C sE | j sA y | j j Wn t t t f k
r7 Yn Xd | _ d S)NT)r r
pause_readingAttributeErrorNotImplementedError RuntimeError)r r r r r0 a s z ResponseHandler.pause_readingc
C sE | j rA y | j j Wn t t t f k
r7 Yn Xd | _ d S)NF)r r resume_readingr1 r2 r3 )r r r r r4 i s z ResponseHandler.resume_readingc s d | _ t j | d S)NT)r r+ r' )r r- )r. r r r' q s z ResponseHandler.set_exceptionc C s? | | _ | | _ | j r; | j d } | _ | j | d S)Nr )r r r
data_received)r parserpayloaddatar r r
set_parseru s
zResponseHandler.set_parsertimer skip_payloadFread_until_eofauto_decompressTc C sc | | _ t | | j d | d t d | d | | _ | j r_ | j d } | _ | j | d S)Nr: Zpayload_exceptionr< r= r )r r Z_loopr r r r5 )r r: r; r< r= r8 r r r set_response_params} s z#ResponseHandler.set_response_paramsc C s | s
d S| j d k r` | j j | \ } } | r\ d | _ d | _ | r\ | j | d S| j sx | j d k r | j | 7_ ny | j j | \ } } } WnC t k
r } z# | j j | j
| d SWYd d } ~ Xn X| | _ x | D]w \ } } | j r d | _ | | _
| | _ | j sG| j d k r`| j | t f d q | j | | f d q W| r| r| j | n | | _ d S)NT 0 r )r? r@ )r Z feed_datar r5 r r r
BaseExceptionr r# r' r" r r r coder ) r r8 eoftailZmessagesr r- messager7 r r r r5 s>
z ResponseHandler.data_received)__name__
__module__ __qualname____doc__r propertyr r" r# r$ r% r, r/ r0 r4 r' r9 r> r5 r r )r. r r
s$ r
)r
Zasyncio.streams
contextlibr Zclient_exceptionsr r r Zhttpr r r r r r
r r r r <module> s