o
    bi                     @  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mZ d dl	m
Z
 d dlZd dlmZmZ d dlmZmZmZ ejdedd	 dddZG dd dZG dd dZdS )    )annotationsN)CallableMutableMapping)Any)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.   )
stacklevelscoper	   bodybytesreturndict[str, Any]c           	      C  sL  |  dddd}| d dd}||r#|t|d }| d ||| d d	d
| d  d|  ddt|tjdddd}|  dpLd}|d |d< |d |d< |  drf| d d |d< |  dg D ]7\}}|d}|dkr|d}n|dkrd}nd| 	 
d d!}|d}||v r|| d" | }|||< ql|S )#zE
    Builds a scope and request body into a WSGI environ object.
    	root_path utf8latin1pathNmethodquery_stringasciizHTTP/http_version)   r   schemehttpTF)REQUEST_METHODSCRIPT_NAME	PATH_INFOQUERY_STRINGSERVER_PROTOCOLzwsgi.versionzwsgi.url_schemez
wsgi.inputzwsgi.errorszwsgi.multithreadzwsgi.multiprocesszwsgi.run_onceserver)	localhostP   r   SERVER_NAMEr   SERVER_PORTclientREMOTE_ADDRheaderszcontent-lengthCONTENT_LENGTHzcontent-typeCONTENT_TYPEHTTP_-_,)getencodedecode
startswithlenioBytesIOsysstdoutupperreplace)	r   r   script_name	path_infoenvironr#   namevaluecorrected_name rB   [/sda-disk/www/egybert/egybert_env/lib/python3.10/site-packages/starlette/middleware/wsgi.pybuild_environ   sD   





rD   c                   @  s    e Zd ZdddZdddZdS )WSGIMiddlewareappCallable[..., Any]r   Nonec                 C  s
   || _ d S N)rF   )selfrF   rB   rB   rC   __init__L   s   
zWSGIMiddleware.__init__r   r	   receiver   sendr
   c                   s2   |d dks	J t | j|}|||I d H  d S )Ntyper   )WSGIResponderrF   )rJ   r   rL   rM   	responderrB   rB   rC   __call__O   s   zWSGIMiddleware.__call__N)rF   rG   r   rH   )r   r	   rL   r   rM   r
   r   rH   )__name__
__module____qualname__rK   rQ   rB   rB   rB   rC   rE   K   s    
rE   c                   @  sT   e Zd ZU ded< ded< d#ddZd$ddZd%ddZ	d&d'ddZd(d!d"ZdS ))rO   z*ObjectSendStream[MutableMapping[str, Any]]stream_sendz-ObjectReceiveStream[MutableMapping[str, Any]]stream_receiverF   rG   r   r	   r   rH   c                 C  s<   || _ || _d | _d | _ttj\| _| _	d| _
d | _d S )NF)rF   r   statusresponse_headersanyiocreate_memory_object_streammathinfrU   rV   response_startedexc_info)rJ   rF   r   rB   rB   rC   rK   Y   s   
zWSGIResponder.__init__rL   r   rM   r
   c              
     s  d}d}|r| I d H }|| dd7 }| dd}|st| j|}t 4 I d H <}|| j| | j4 I d H  tj	| j
|| jI d H  W d   I d H  n1 I d H sXw   Y  W d   I d H  n1 I d H smw   Y  | jd ur| jd | jd | jd d S )	N    Tr   	more_bodyFr   r   r   )r1   rD   r   rY   create_task_group
start_soonsenderrU   	to_threadrun_syncwsgistart_responser^   with_traceback)rJ   rL   rM   r   r`   messager>   
task_grouprB   rB   rC   rQ   b   s$   *(
zWSGIResponder.__call__c              	     sf   | j 4 I d H  | j 2 z3 d H W }||I d H  q6 W d   I d H  d S 1 I d H s,w   Y  d S rI   )rV   )rJ   rM   ri   rB   rB   rC   rc   r   s   .zWSGIResponder.senderNrW   strrX   list[tuple[str, str]]r^   r   c                 C  sZ   || _ | js+d| _|dd\}}t|}dd |D }tj| jjd||d d S d S )NT r   c                 S  s0   g | ]\}}|  d  |  d fqS )r   )stripr2   lower).0r?   r@   rB   rB   rC   
<listcomp>   s    z0WSGIResponder.start_response.<locals>.<listcomp>zhttp.response.start)rN   rW   r*   )	r^   r]   splitintrY   from_threadrunrU   rM   )rJ   rW   rX   r^   status_code_stringr/   status_coder*   rB   rB   rC   rg   w   s    zWSGIResponder.start_responser>   r   rg   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)rN   r   r`   r_   )rN   r   )rF   rY   rt   ru   rU   rM   )rJ   r>   rg   chunkrB   rB   rC   rf      s   
zWSGIResponder.wsgi)rF   rG   r   r	   r   rH   )rL   r   rM   r
   r   rH   )rM   r
   r   rH   rI   )rW   rk   rX   rl   r^   r   r   rH   )r>   r   rg   rG   r   rH   )	rR   rS   rT   __annotations__rK   rQ   rc   rg   rf   rB   rB   rB   rC   rO   U   s   
 

	
	rO   )r   r	   r   r   r   r   )
__future__r   r6   r[   r8   warningscollections.abcr   r   typingr   rY   	anyio.abcr   r   starlette.typesr   r	   r
   warnDeprecationWarningrD   rE   rO   rB   rB   rB   rC   <module>   s$    
4
