a
    c	g!                     @  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}||v 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   \/var/www/html/llm_bihealth/app/venv/lib/python3.9/site-packages/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J t | j|}|||I d H  d S )Ntyper   )WSGIResponderr9   )r;   r
   r=   r>   Z	responderr3   r3   r4   __call__M   s    zWSGIMiddleware.__call__N)__name__
__module____qualname__r<   rA   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 )r@   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infrE   rF   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 x}|| j| | j4 I d H 0 tj	| j
|| jI d H  W d   I d H  q1 I d H s0    Y  W d   I d H  q1 I d H s0    Y  | jd ur
| jd | jd | jd d S )	N    Tr   	more_bodyFr   r      )r$   r5   r
   rI   Zcreate_task_groupZ
start_soonsenderrE   Z	to_threadZrun_syncwsgistart_responserM   with_traceback)r;   r=   r>   r   rO   messager0   Z
task_groupr3   r3   r4   rA   b   s    nzWSGIResponder.__call__)r>   r   c              	     sb   | j 4 I d H : | j 2 z3 d H W }||I d H  q6 W d   I d H  q^1 I d H sT0    Y  d S r:   )rF   )r;   r>   rU   r3   r3   r4   rQ   r   s    zWSGIResponder.senderNstrzlist[tuple[str, str]]z
typing.Any)rG   rH   rM   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?   rG   r   )	rM   rL   splitintrI   from_threadrunrE   r>   )r;   rG   rH   rM   Zstatus_code_stringr"   status_coder   r3   r3   r4   rS   w   s    zWSGIResponder.start_responser	   )r0   rS   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   rO   rN   )r?   r   )r9   rI   r^   r_   rE   r>   )r;   r0   rS   chunkr3   r3   r4   rR      s    
zWSGIResponder.wsgi)N)	rB   rC   rD   __annotations__r<   rA   rQ   rS   rR   r3   r3   r3   r4   r@   S   s   
	 r@   )
__future__r   r)   rJ   r+   typingwarningsrI   Z	anyio.abcr   r   Zstarlette.typesr   r   r   warnDeprecationWarningr5   r6   r@   r3   r3   r3   r4   <module>   s   4
