Edit File: worker.cpython-37.pyc
B �k�aK" � @ s d Z ddlZddlZddlZddlZddlZddlmZ ddlm Z m Z m Z m Z m Z ddlmZ ddlmZ ddlmZ dd lmZ dd lmZ dd lmZ yddlZej Z W n e k r� dZe Z Y nX d Z G d d� dej �Z!G dd� de!�Z"G dd� de!�Z#dS )z%Async gunicorn worker for aiohttp.web� N)� FrameType)�Any� Awaitable�Callable�Optional�Union)�AccessLogFormat)�base)�web� )� set_result)� Application)� AccessLogger)�GunicornWebWorker�GunicornUVLoopWebWorker�GunicornTokioWebWorkerc s� e Zd ZejZejZe e dd�� fdd� Z dd�� fdd� Z dd�dd �Z dd�d d �Z d d�d d�Zd ed dd�dd�Zdd�dd�Zeedd�dd�Zeedd�dd�Zee dd�dd ��Zeed �d d �Z� ZS )!r N)�args�kw�returnc s$ t � j||� d | _d| _d | _d S )Nr )�super�__init__�_task� exit_code�_notify_waiter)�selfr r )� __class__� �A/opt/alt/python37/lib64/python3.7/site-packages/aiohttp/worker.pyr % s zGunicornWebWorker.__init__)r c s0 t �� �� t �� | _t �| j� t� �� d S )N)�asyncio�get_event_loop�closeZnew_event_loop�loopZset_event_loopr � init_process)r )r r r r" , s z GunicornWebWorker.init_processc C sr | j �| �� �| _y| j �| j� W n tk rD | j�d� Y nX | j �| j �� � | j � � t � | j � d S )Nz Exception in gunicorn worker) r! Z create_task�_runr Zrun_until_complete� Exception�log� exceptionZshutdown_asyncgensr �sys�exitr )r r r r �run5 s zGunicornWebWorker.runc � s� d }t | jt�r| j}nHt�| j�rP| �� I d H }t |tj�rJ|}|j}q`|}ntd� | j���|d kr�| j j rx| j j nd }tj|| j | j j|| �| j j�d�}|�� I d H | j jr�| �| j �nd }|}|d k s�t�|j}|d k s�t�x:| jD ]0}tj|||| j jd d d�}|�� I d H q�W t�� } y�x�| j �r�| � � |j } | j j �rt| | j j k�rtd| _ | j � d| � nB| t�� k�r�| j!t�"� k�r�d| _ | j � d| � n| �#� I d H �q0W W n t$k �r� Y nX |�%� I d H d S ) NzUwsgi app should be either Application or async function returning Application, got {})�loggerZkeepalive_timeout� access_log�access_log_format�d �_ )Z ssl_contextZshutdown_timeoutFz Max requests, shutting down: %sz!Parent changed, shutting down: %s)&� isinstance�wsgir r �iscoroutinefunctionr Z AppRunner�app� RuntimeError�format�cfgZ accesslogr% r+ Z keepalive�_get_valid_log_formatr, �setupZis_ssl�_create_ssl_context�AssertionError�serverZsocketsZSockSiteZgraceful_timeout�start�os�getpid�alive�notifyZrequests_countZ max_requests�infoZppid�getppid�_wait_next_notify� BaseException�cleanup) r Zrunnerr2 r0 r+ �ctxr: �sock�site�pidZcntr r r r# A sb zGunicornWebWorker._runzasyncio.Future[bool]c C s> | � � | j}|d k st�|�� | _}| j�d| j |� |S )Ng �?)�_notify_waiter_doner! r9 Z create_futurer Z call_later)r r! �waiterr r r rB � s z#GunicornWebWorker._wait_next_notify)rJ r c C s4 |d kr| j }|d k r t|d� || j kr0d | _ d S )NT)r r )r rJ r r r rI � s z%GunicornWebWorker._notify_waiter_donec C s� | j �tj| jtjd � | j �tj| jtjd � | j �tj| jtjd � | j �tj| j tjd � | j �tj | j tj d � | j �tj | j tj d � t�tjd� t�tj d� tjdk r�t�tjtj� d S )NF)� � )r! Zadd_signal_handler�signal�SIGQUIT� handle_quit�SIGTERMZ handle_exit�SIGINT�SIGWINCHZ handle_winch�SIGUSR1Z handle_usr1�SIGABRT� handle_abort� siginterruptr' � version_info�SIGCHLD�SIG_DFL)r r r r � init_signals� s z GunicornWebWorker.init_signals)�sig�framer c C s d| _ | j�| � | �� d S )NF)r> r5 Z worker_intrI )r r[ r\ r r r rO � s z GunicornWebWorker.handle_quitc C s& d| _ d| _| j�| � t�d� d S )NFr )r> r r5 Z worker_abortr' r( )r r[ r\ r r r rU � s z GunicornWebWorker.handle_abort� SSLContext)r5 r c C s\ t dkrtd��t �| j�}|�| j| j� | j|_| j rF|� | j � | j rX|� | j � |S )z~Creates SSLContext instance for usage in asyncio.create_server. See ssl.SSLSocket.__init__ for more details. NzSSL is not supported.) �sslr3 r] Z ssl_versionZload_cert_chainZcertfileZkeyfileZ cert_reqsZ verify_modeZca_certsZload_verify_locationsZciphersZ set_ciphers)r5 rE r r r r8 � s z%GunicornWebWorker._create_ssl_context)� source_formatr c C s. || j kr| jS t�d|�r&td��n|S d S )Nz %\([^\)]+\)z�Gunicorn's style options in form of `%(name)s` are not supported for the log formatting. Please use aiohttp's format specification to configure access log formatting: http://docs.aiohttp.org/en/stable/logging.html#format-specification)�DEFAULT_GUNICORN_LOG_FORMAT�DEFAULT_AIOHTTP_LOG_FORMAT�re�search� ValueError)r r_ r r r r6 � s z'GunicornWebWorker._get_valid_log_format)N)�__name__� __module__� __qualname__r Z LOG_FORMATra �GunicornAccessLogFormat�defaultr` r r r" r) r# rB r rI rZ �intr rO rU � staticmethodr8 �strr6 � __classcell__r r )r r r s B ( r c s" e Zd Zdd�� fdd� Z� ZS )r N)r c s0 dd l }t�� �� t�|�� � t� �� d S )Nr )�uvloopr r r �set_event_loop_policy�EventLoopPolicyr r" )r rn )r r r r" � s z$GunicornUVLoopWebWorker.init_process)re rf rg r" rm r r )r r r � s r c s" e Zd Zdd�� fdd� Z� ZS )r N)r c s0 dd l }t�� �� t�|�� � t� �� d S )Nr )�tokior r r ro rp r r" )r rq )r r r r" s z#GunicornTokioWebWorker.init_process)re rf rg r" rm r r )r r r s r )$�__doc__r r< rb rM r' �typesr �typingr r r r r Zgunicorn.configr rh Zgunicorn.workersr Zaiohttpr Zhelpersr Zweb_appr Zweb_logr r^ r] � ImportError�object�__all__ZWorkerr r r r r r r �<module> s0 Q