U
    eS                     @  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	 d dl
mZ d dlmZ d dlmZ G dd	 d	eZG d
d deZdS )    )annotationsN)Any)Arbiter)Worker)Config)Serverc                      s   e Zd ZU dZdddZded< dddd fd	d
Zdd fddZddddZddddZ	ddddZ
ddddZddddZ  ZS )UvicornWorkerzv
    A worker class for Gunicorn that interfaces with an ASGI consumer callable,
    rather than a WSGI callable.
    autoloophttpzdict[str, Any]CONFIG_KWARGSr   None)argskwargsreturnc              	     s2  t  j|| td}| jjj|_|| jjj d|_	td}| jj
j|_|| jj
j d|_	d d | jj| j| j| j| jjd}| jjr| jjd| jjd| jjd| jjd| jjd	| jjd
| jjdd}|| | jjd jr| jjd j|d< || j tf || _d S )Nzuvicorn.errorFzuvicorn.access)appZ
log_configZtimeout_keep_aliveZtimeout_notifycallback_notifyZlimit_max_requestsforwarded_allow_ipskeyfilecertfilepasswordssl_version	cert_reqsca_certsciphers)Zssl_keyfileZssl_certfileZssl_keyfile_passwordr   Zssl_cert_reqsZssl_ca_certsZssl_ciphersbacklog)super__init__logging	getLoggerlogZ	error_loghandlerssetLevellevel	propagateZ
access_logcfg	keepalivetimeoutr   Zmax_requestsr   Zis_sslZssl_optionsgetupdatesettingsvaluer   r   config)selfr   r   loggerZconfig_kwargsZ
ssl_kwargs	__class__ 3/tmp/pip-unpacked-wheel-cq1xfpla/uvicorn/workers.pyr      s>    


	
zUvicornWorker.__init__)r   c                   s   | j   t   d S N)r-   Zsetup_event_loopr   init_processr.   r0   r2   r3   r5   B   s    
zUvicornWorker.init_processc                 C  s<   | j D ]}t|tj qttj| j ttjd d S )NF)ZSIGNALSsignalSIG_DFLSIGUSR1Zhandle_usr1siginterrupt)r.   sr2   r2   r3   init_signalsF   s    
zUvicornWorker.init_signalsc                 C  s"   t  }|tj| jtjd dS )zInstall a SIGQUIT handler on workers.

        - https://github.com/encode/uvicorn/issues/1116
        - https://github.com/benoitc/gunicorn/issues/2604
        N)asyncioZget_running_loopZadd_signal_handlerr7   SIGQUITZhandle_exit)r.   r   r2   r2   r3   _install_sigquit_handlerQ   s    z&UvicornWorker._install_sigquit_handlerc                   sH   | j | j_t| jd}|   |j| jdI d H  |jsDt	t
j d S )N)r-   )sockets)Zwsgir-   r   r   r?   Zserver@   startedsysexitr   ZWORKER_BOOT_ERROR)r.   serverr2   r2   r3   _serve[   s    
zUvicornWorker._servec                 C  s   t |  S r4   )r=   runrE   r6   r2   r2   r3   rF   c   s    zUvicornWorker.runc                   s   |    d S r4   )notifyr6   r2   r2   r3   r   f   s    zUvicornWorker.callback_notify)__name__
__module____qualname____doc__r   __annotations__r   r5   r<   r?   rE   rF   r   __classcell__r2   r2   r0   r3   r      s   
*
r   c                   @  s   e Zd ZdddZdS )UvicornH11Workerr=   Zh11r
   N)rH   rI   rJ   r   r2   r2   r2   r3   rN   j   s   rN   )
__future__r   r=   r   r7   rB   typingr   Zgunicorn.arbiterr   Zgunicorn.workers.baser   Zuvicorn.configr   Zuvicorn.mainr   r   rN   r2   r2   r2   r3   <module>   s   Z