a
    њh                     @  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mZ d dlm	Z	m
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)MutableMapping)AnyCallable)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, 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_INFOZ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-lengthZCONTENT_LENGTHzcontent-typeZ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   T/wd/webapps/venvs/v2025_4um/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Callable[..., Any]None)appr   c                 C  s
   || _ d S N)r9   )selfr9   r3   r3   r4   __init__K   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__N   s    zWSGIMiddleware.__call__N)__name__
__module____qualname__r<   rA   r3   r3   r3   r4   r6   J   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@   z*ObjectSendStream[MutableMapping[str, Any]]stream_sendz-ObjectReceiveStream[MutableMapping[str, 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<   X   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   a   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   q   s    zWSGIResponder.senderNstrzlist[tuple[str, str]]r   )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   v   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@   T   s   
		 r@   )
__future__r   r)   rJ   r+   warningscollections.abcr   typingr   r   rI   Z	anyio.abcr   r   Zstarlette.typesr   r	   r
   warnDeprecationWarningr5   r6   r@   r3   r3   r3   r4   <module>   s    4
