a
    df                     @   s   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 d dlm	Z	m
Z
mZmZ d dlmZmZmZmZmZmZmZmZmZmZmZmZ ddejeddd	ZG d
d dZG dd dZzd dlmZ W n e y   eZY n0 dS )    N)deque)DequeIterableOptionalTuple)ASGIReceiveCallableASGIReceiveEventASGISendCallableASGISendEventEnvironExcInfoHTTPRequestEventHTTPResponseBodyEventHTTPResponseStartEvent	HTTPScopeStartResponseWSGIAppr   r   )scopemessagebodyreturnc                 C   s.  | d d| d  dd| d dd| d	  d
| dd|tjdddd}| d}|du rdd}|d |d< |d |d< | d}|dur|d |d< | dg D ]\}}|d}|dkrd}	n"|dkrd}	nd| dd  }	|d}
|	|v r ||	 }t|tsJ |d! |
 }
|
||	< q|S )"zH
    Builds a scope and request message into a WSGI environ object.
    method pathutf8latin1Zquery_stringasciizHTTP/%s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serverN)	localhostP   r   ZSERVER_NAMEr   ZSERVER_PORTclientZREMOTE_ADDRheaderszcontent-lengthCONTENT_LENGTHzcontent-typeCONTENT_TYPEzHTTP_%s-_,)	encodedecodegetsysstdoutupperreplace
isinstancestr)r   r   r   environr#   r&   namevalueZname_strZcorrected_nameZ	value_strZcorrected_name_environ r9   `/var/www/html/stable-diffusion-webui/venv/lib/python3.9/site-packages/uvicorn/middleware/wsgi.pybuild_environ   sF    







r;   c                   @   s2   e Zd ZdeedddZddddd	d
dZdS )_WSGIMiddleware
   )appworkersc                 C   s&   t dt || _tjj|d| _d S )NzgUvicorn's native WSGI implementation is deprecated, you should switch to a2wsgi (`pip install a2wsgi`).)max_workers)warningswarnDeprecationWarningr>   
concurrentfuturesThreadPoolExecutorexecutor)selfr>   r?   r9   r9   r:   __init__O   s    z_WSGIMiddleware.__init__r   r   r	   N)r   receivesendr   c                    s4   |d dksJ t | j| j|}|||I d H  d S )Ntyper    )WSGIResponderr>   rG   )rH   r   rJ   rK   instancer9   r9   r:   __call__X   s    z_WSGIMiddleware.__call__)r=   )__name__
__module____qualname__r   intrI   rO   r9   r9   r9   r:   r<   N   s   r<   c                   @   s|   e Zd ZeejjddddZddddd	d
ZdddddZ	de
eee
e
f  ee ddddZeeddddZdS )rM   r   )r>   rG   r   c                 C   sJ   || _ || _|| _d | _d | _t | _t | _	t
 | _d| _d | _d S )NF)r>   rG   r   statusresponse_headersasyncioEvent
send_eventr   
send_queueget_event_looploopresponse_startedexc_info)rH   r>   rG   r   r9   r9   r:   rI   d   s    

zWSGIResponder.__init__r   r	   N)rJ   rK   r   c           
   
      sP  | I d H }t |dd}|dd}|rv|dt j |rl| I d H }||dd |dd}q<|d t| j||}t	 | _
| j
| j| j|| j}| j
| |}	z>t|d I d H  W | jd  | j  t|	d I d H  n*| jd  | j  t|	d I d H  0 | jd urL| jd | jd | jd d S )Nr       	more_bodyFr   r      )ioBytesIOr/   seekSEEK_ENDwriter;   r   rV   rZ   r[   Zrun_in_executorrG   wsgistart_responseZcreate_tasksenderwait_forrY   appendrX   setr]   with_traceback)
rH   rJ   rK   r   r   r_   Zbody_messager6   rf   rh   r9   r9   r:   rO   u   s6    




zWSGIResponder.__call__)rK   r   c                    sL   | j r,| j  }|d u rd S ||I d H  q | j I d H  | j  q d S )N)rY   popleftrX   waitclear)rH   rK   r   r9   r9   r:   rh      s    
zWSGIResponder.sender)rT   rU   r]   r   c           	      C   sd   || _ | js`d| _|dd\}}t|}dd |D }d||d}| j| | j| jj	 d S )NT r   c                 S   s$   g | ]\}}| d | d fqS )r   )r-   ).0r7   r8   r9   r9   r:   
<listcomp>   s   z0WSGIResponder.start_response.<locals>.<listcomp>zhttp.response.start)rL   rT   r'   )
r]   r\   splitrS   rY   rj   r[   call_soon_threadsaferX   rk   )	rH   rT   rU   r]   Zstatus_code_strr+   status_coder'   Zhttp_response_start_eventr9   r9   r:   rg      s    zWSGIResponder.start_response)r6   rg   r   c                 C   sf   |  ||D ],}d|dd}| j| | j| jj qdddd}| j| | j| jj d S )Nzhttp.response.bodyT)rL   r   r_   r^   F)r>   rY   rj   r[   rt   rX   rk   )rH   r6   rg   chunkZresponse_bodyZ
empty_bodyr9   r9   r:   rf      s    zWSGIResponder.wsgi)N)rP   rQ   rR   r   rD   rE   rF   rI   rO   rh   r5   r   r   r   r   rg   r   r   rf   r9   r9   r9   r:   rM   c   s     rM   )WSGIMiddleware)!rV   concurrent.futuresrD   ra   r0   rA   collectionsr   typingr   r   r   r   Zuvicorn._typesr   r   r	   r
   r   r   r   r   r   r   r   r   rb   r;   r<   rM   Za2wsgirw   ModuleNotFoundErrorr9   r9   r9   r:   <module>   s    85e