���Z� � @ s� d Z d d l Z d d l Z d d l Z d d l m Z d d l m Z d
Z e j d d
d g � Z
e
d d � Z e
d d � Z Gd d � d e � Z
d S)z"Http related parsers and protocol.� N� )�AbstractStreamWriter)�noop� StreamWriter� HttpVersion�
HttpVersion10�
HttpVersion11�major�minorc @ s� e Z d Z d d � Z e d d � � Z e d d � � Z d d � Z d d
d � Z d d
� Z d d d d d d � Z
d d d d � Z d d d � Z d d � Z
d S) r c C s^ | | _ | | _ | | _ d | _ d | _ d | _ d | _ d | _ d | _ d | _ d S)NFr )
� _protocol�
_transport�loop�length�chunked� buffer_size� output_size�_eof� _compressZ
_drain_waiter)�self�protocol� transportr
� r �/http_writer.py�__init__ s zStreamWriter.__init__c C s | j S)N)r )r r r r r " s zStreamWriter.transportc C s | j S)N)r )r r r r r & s zStreamWriter.protocolc C s
d | _ d S)NT)r )r r r r �enable_chunking* s z StreamWriter.enable_chunkingZdeflatec C s< | d k r d t j n t j } t j d | � | _ d S)N�gzip� Zwbits)�zlib� MAX_WBITS� compressobjr )r �encodingZ zlib_moder r r �enable_compression- s #z StreamWriter.enable_compressionc C sk t | � } | j | 7_ | j | 7_ | j d k sH | j j � rW t j d � � | j j | � d S)Nz!Cannot write to closing transport)�lenr r r Z
is_closing�asyncio�CancelledError�write)r �chunk�sizer r r �_write2 s zStreamWriter._write�drainT�LIMIT�@ i c C s | j d k r. | j j | � } | s. t � S| j d k r� t | � } | j | k rk | j | | _ n) | d | j � } d | _ | s� t � S| r� | j r� d t | � j d � } | | d } | j | � | j | k r� | r� d | _ | j � St � S)z�Writes chunk of data to a stream.
write_eof() indicates end of stream.
writer can't be used after write_eof() method being called.
write() return drain future.
Nr z%x
�asciis
)
r �compressr r r" r �encoder( r r) )r r& r) r* � chunk_lenr r r r% ; s*
zStreamWriter.writez: z
c sS | d j � � f d d � | j � D� � } | j d � d } | j | � d S)z*Write request/response status and headers.� c s( g | ] \ } } | � | � � q Sr r )�.0�k�v)�END�SEPr r �
<listcomp>b s z.StreamWriter.write_headers.<locals>.<listcomp>zutf-8s
N)�join�itemsr. r( )r Z status_lineZheadersr5 r4 r )r4 r5 r �
write_headers^ s &zStreamWriter.write_headers� c � s� | j r
d S| j rz | r. | j j | � } | | j j � } | r� | j r� d t | � j d � } | | d } n? | j r� | r� d t | � j d � } | | d } n d } | r� | j | � | j � Id Hd | _ d | _ d S)Nz%x
r, s
0
s 0
T)
r r r- �flushr r" r. r( r) r )r r& r/ r r r � write_eoff s&
zStreamWriter.write_eofc � s( | j j d k r$ | j j � Id Hd S)zFlush the write buffer.
The intended use is to write
await w.write(data)
await w.drain()
N)r r Z
_drain_helper)r r r r r) � s zStreamWriter.drainNi )�__name__�
__module__� __qualname__r �propertyr r r r! r( r% r9 r< r) r r r r r s # )r r r r )�__doc__r# � collectionsr �abcr Zhelpersr �__all__�
namedtupler r r r r r r r �<module> s