B
    Ce                  @   sR  U d dl Z d dlmZmZmZmZmZmZmZm	Z	m
Z
 ddlmZmZmZmZ ddlmZmZmZmZmZ ddlmZ ddlmZmZmZmZmZmZmZm Z  ddl!m"Z"m#Z#m$Z$m%Z% d	gZ&e 'e(d
Z)e 'dZ*ee+ ee+ dddZ,ee+ eee+e+f  dddZ-e 'e(d
Z.eee dddZ/e 'e(d
Z0ee
eedf dddZ1G dd dZ2e 'e(d
Z3G dd dZ4G dd dZ5eddddZ6ee
e	e$ ee	e$ e	e$ f f e
edef ee7edef f f f Z8eefe/e efe1e efe1eefe6eefe6eefe6e efe6e efe6e efe6ee4e2e5di
Z9e8e:d	< dS )     N)	AnyCallableDictIterableNoReturnOptionalTupleTypeUnion   )chunk_headerheader_fieldrequest_linestatus_line)DataEndOfMessageInformationalResponseRequestResponse)ReceiveBuffer)CLIENTCLOSEDDONEIDLE
MUST_CLOSE	SEND_BODYSEND_RESPONSESERVER)LocalProtocolErrorRemoteProtocolErrorSentinelvalidateREADERSasciis   [ \t]+)linesreturnc             c   s   t | }d }xl|D ]d}t|}|rd|d kr4tdt|tsFt|}|d7 }||| d  7 }q|d k	rr|V  |}qW |d k	r|V  d S )Nz%continuation line at start of headers    )iterobs_fold_rematchr   
isinstance	bytearrayend)r$   itlastliner)    r0   `/var/www/downstreamdata.science/rtclock/rtclock-venv/lib/python3.7/site-packages/h11/_readers.py_obsolete_line_fold+   s     


r2   c             c   s6   x0t | D ]$}tt|d|}|d |d fV  q
W d S )Nzillegal header line: {!r}
field_nameZfield_value)r2   r!   header_field_re)r$   r/   matchesr0   r0   r1   _decode_header_lines@   s    r6   )bufr%   c             C   sj   |   }|d kr$|  r tdd S |s0tdtt|d d|d }tf tt|dd  dd|S )Nzillegal request linezno request line receivedr   zillegal request line: {!r}r   T)headers_parsed)maybe_extract_lines+is_next_line_obviously_invalid_request_liner   r!   request_line_rer   listr6   )r7   r$   r5   r0   r0   r1   maybe_read_from_IDLE_clientK   s    r>   c             C   s   |   }|d kr$|  r tdd S |s0tdtt|d d|d }|d d krVdn|d }|d d krndn|d }t|d	 }|d
k rtnt}|tt	|dd  d|||dS )Nzillegal request linezno response line receivedr   zillegal status line: {!r}http_versions   1.1reason    status_code   r   T)r8   r9   rB   r@   r?   )
r:   r;   r   r!   status_line_reintr   r   r=   r6   )r7   r$   r5   r?   r@   rB   class_r0   r0   r1   $maybe_read_from_SEND_RESPONSE_server^   s$    rG   c               @   sD   e Zd ZeddddZeeeedf dddZ	e
dd	d
ZdS )ContentLengthReaderN)lengthr%   c             C   s   || _ || _d S )N)_length
_remaining)selfrI   r0   r0   r1   __init__{   s    zContentLengthReader.__init__)r7   r%   c             C   sD   | j dkrt S || j }|d kr(d S |  j t|8  _ t|dS )Nr   )data)rK   r   maybe_extract_at_mostlenr   )rL   r7   rN   r0   r0   r1   __call__   s    
zContentLengthReader.__call__)r%   c             C   s   t d| j| j | jd S )Nz]peer closed connection without sending complete message body (received {} bytes, expected {}))r   formatrJ   rK   )rL   r0   r0   r1   read_eof   s    zContentLengthReader.read_eof)__name__
__module____qualname__rE   rM   r   r
   r   r   rQ   r   rS   r0   r0   r0   r1   rH   z   s   	rH   c               @   sB   e Zd ZddddZeeeedf dddZe	ddd	Z
dS )
ChunkedReaderN)r%   c             C   s   d| _ d| _d| _d S )Nr   F)_bytes_in_chunk_bytes_to_discard_reading_trailer)rL   r0   r0   r1   rM      s    zChunkedReader.__init__)r7   r%   c             C   s@  | j r,| }|d krd S ttt|dS | jdkrn|| j}|d krNd S |  jt|8  _| jdkrnd S | jdks|t| j	dkr|
 }|d krd S tt|d|}t|d dd| _	| j	dkrd| _ | |S d}nd}| j	dkst|| j	}|d krd S |  j	t|8  _	| j	dkr.d	| _d}nd}t|||d
S )N)r8   r   zillegal chunk header: {!r}
chunk_size   )baseTF   )rN   chunk_start	chunk_end)rZ   r:   r   r=   r6   rY   rO   rP   AssertionErrorrX   Zmaybe_extract_next_liner!   chunk_header_rerE   r   )rL   r7   r$   rN   r   r5   r_   r`   r0   r0   r1   rQ      sL    




zChunkedReader.__call__c             C   s   t dd S )NzVpeer closed connection without sending complete message body (incomplete chunked read))r   )rL   r0   r0   r1   rS      s    zChunkedReader.read_eof)rT   rU   rV   rM   r   r
   r   r   rQ   r   rS   r0   r0   r0   r1   rW      s   .rW   c               @   s.   e Zd Zeee dddZedddZdS )Http10Reader)r7   r%   c             C   s    | d}|d krd S t|dS )Niɚ;)rN   )rO   r   )rL   r7   rN   r0   r0   r1   rQ      s    
zHttp10Reader.__call__)r%   c             C   s   t  S )N)r   )rL   r0   r0   r1   rS      s    zHttp10Reader.read_eofN)	rT   rU   rV   r   r   r   rQ   r   rS   r0   r0   r0   r1   rc      s   rc   c             C   s   | rt dd S )NzGot data when expecting EOF)r   )r7   r0   r0   r1   expect_nothing   s    rd   .)chunkedzcontent-lengthzhttp/1.0);retypingr   r   r   r   r   r   r   r	   r
   Z_abnfr   r   r   r   Z_eventsr   r   r   r   r   Z_receivebufferr   _stater   r   r   r   r   r   r   r   Z_utilr   r   r    r!   __all__compileencoder4   r(   bytesr2   r6   r<   r>   rD   rG   rH   rb   rW   rc   rd   strZReadersTyper"   __annotations__r0   r0   r0   r1   <module>   sN    ,(

>(