U
    i
                     @  s   d dl mZ d dlZd dlZd dlZd dlZd dlZd dlZd dlm	Z	m
Z
 d dlmZmZmZ ede dddd	d
dZG dd dZG dd dZdS )    )annotationsN)ObjectReceiveStreamObjectSendStream)ReceiveScopeSendzstarlette.middleware.wsgi is deprecated and will be removed in a future release. Please refer to https://github.com/abersheeran/a2wsgi as a replacement.r   bytesdict[str, typing.Any])scopebodyreturnc           	      C  sP  |  dddd}| d dd}||rF|t|d }| d ||| d d	d
| d  d|  ddt|tjdddd}|  dpd}|d |d< |d |d< |  dr| d d |d< |  dg D ]r\}}|d}|dkrd}n&|dkrd}nd| 	 
d d!}|d}||krB|| d" | }|||< q|S )#zE
    Builds a scope and request body into a WSGI environ object.
    	root_path utf8latin1pathNmethodZquery_stringasciizHTTP/http_version)   r   schemehttpTF)REQUEST_METHODZSCRIPT_NAMEZ	PATH_INFOQUERY_STRINGZSERVER_PROTOCOLzwsgi.versionzwsgi.url_schemez
wsgi.inputzwsgi.errorszwsgi.multithreadzwsgi.multiprocesszwsgi.run_onceserver)	localhostP   r   ZSERVER_NAMEr   ZSERVER_PORTclientZREMOTE_ADDRheaderszcontent-lengthCONTENT_LENGTHzcontent-typeCONTENT_TYPEZHTTP_-_,)getencodedecode
startswithlenioBytesIOsysstdoutupperreplace)	r
   r   script_nameZ	path_infoenvironr   namevalueZcorrected_name r3   =/tmp/pip-unpacked-wheel-1q9scis0/starlette/middleware/wsgi.pybuild_environ   sD    







r5   c                   @  s0   e Zd ZdddddZddddd	d
dZdS )WSGIMiddleware typing.Callable[..., typing.Any]None)appr   c                 C  s
   || _ d S N)r9   )selfr9   r3   r3   r4   __init__J   s    zWSGIMiddleware.__init__r   r   r   )r
   receivesendr   c                   s0   |d dkst t| j|}|||I d H  d S )Ntyper   )AssertionErrorWSGIResponderr9   )r;   r
   r=   r>   Z	responderr3   r3   r4   __call__M   s    zWSGIMiddleware.__call__N)__name__
__module____qualname__r<   rB   r3   r3   r3   r4   r6   I   s   r6   c                   @  sz   e Zd ZU ded< ded< ddddd	d
ZddddddZdddddZddddddddZddddddZdS )rA   z8ObjectSendStream[typing.MutableMapping[str, typing.Any]]stream_sendz;ObjectReceiveStream[typing.MutableMapping[str, typing.Any]]stream_receiver7   r   r8   )r9   r
   r   c                 C  s<   || _ || _d | _d | _ttj\| _| _	d| _
d | _d S )NF)r9   r
   statusresponse_headersanyioZcreate_memory_object_streammathinfrF   rG   response_startedexc_info)r;   r9   r
   r3   r3   r4   r<   W   s    zWSGIResponder.__init__r   r   )r=   r>   r   c                   s   d}d}|r6| I d H }|| dd7 }| dd}qt| j|}t 4 I d H N}|| j| | j4 I d H   tj	| j
|| jI d H  W 5 Q I d H R X W 5 Q I d H R X | jd k	r| jd | jd | jd d S )	N    Tr   	more_bodyFr   r      )r$   r5   r
   rJ   Zcreate_task_groupZ
start_soonsenderrF   Z	to_threadZrun_syncwsgistart_responserN   with_traceback)r;   r=   r>   r   rP   messager0   Z
task_groupr3   r3   r4   rB   `   s    :
zWSGIResponder.__call__)r>   r   c              
     sH   | j 4 I d H * | j 2 z3 d H W }||I d H  q6 W 5 Q I d H R X d S r:   )rG   )r;   r>   rV   r3   r3   r4   rR   p   s    zWSGIResponder.senderNstrzlist[tuple[str, str]]z
typing.Any)rH   rI   rN   r   c                 C  sV   || _ | jsRd| _|dd\}}t|}dd |D }tj| jjd||d d S )NT r   c                 S  s0   g | ](\}}|  d  |  d fqS )r   )stripr%   lower).0r1   r2   r3   r3   r4   
<listcomp>   s   z0WSGIResponder.start_response.<locals>.<listcomp>zhttp.response.start)r?   rH   r   )	rN   rM   splitintrJ   from_threadrunrF   r>   )r;   rH   rI   rN   Zstatus_code_stringr"   status_coder   r3   r3   r4   rT   u   s    zWSGIResponder.start_responser	   )r0   rT   r   c                 C  sH   |  ||D ]}tj| jjd|dd qtj| jjddd d S )Nzhttp.response.bodyT)r?   r   rP   rO   )r?   r   )r9   rJ   r_   r`   rF   r>   )r;   r0   rT   chunkr3   r3   r4   rS      s    
zWSGIResponder.wsgi)N)	rC   rD   rE   __annotations__r<   rB   rR   rT   rS   r3   r3   r3   r4   rA   S   s   
		 rA   )
__future__r   r)   rK   r+   typingwarningsrJ   Z	anyio.abcr   r   Zstarlette.typesr   r   r   warnDeprecationWarningr5   r6   rA   r3   r3   r3   r4   <module>   s   4
