���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� )�
ClientOSError�ClientPayloadError�ServerDisconnectedError)�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� )�asyncio�streams�FlowControlMixin�__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 �
isinstance�OSErrorr �argsr r r r r �super�connection_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_reading�AttributeError�NotImplementedError� 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 �parser�payload�datar r r �
set_parseru s
zResponseHandler.set_parser�timer� skip_payloadF�read_until_eof�auto_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 �eof�tailZmessagesr 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