Edit File: worker.cpython-37.opt-1.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�logZ 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 {})ZloggerZkeepalive_timeout� access_log�access_log_format�d �_ )Z ssl_contextZshutdown_timeoutFz Max requests, shutting down: %sz!Parent changed, shutting down: %s)&� isinstance�wsgir r Ziscoroutinefunctionr Z AppRunner�app� RuntimeError�format�cfgZ accesslogr% r) Z keepalive�_get_valid_log_formatr* ZsetupZis_ssl�_create_ssl_context�AssertionError�serverZsocketsZSockSiteZgraceful_timeout�start�os�getpid�alive�notifyZrequests_countZ max_requests�infoZppid�getppid�_wait_next_notify� BaseExceptionZcleanup) r Zrunnerr/ r. r) �ctxr6 ZsockZsite�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! r5 Z create_futurer Z call_later)r r! �waiterr r r r> � s z#GunicornWebWorker._wait_next_notify)rC r c C s4 |d kr| j }|d k r t|d� || j kr0d | _ d S )NT)r r )r rC r r r rB � 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: r2 Z worker_intrB )r rT rU r r r rH � s z GunicornWebWorker.handle_quitc C s& d| _ d| _| j�| � t�d� d S )NFr )r: r r2 Z worker_abortr& r' )r rT rU r r r rN � s z GunicornWebWorker.handle_abort� SSLContext)r2 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.) �sslr0 rV Z ssl_versionZload_cert_chainZcertfileZkeyfileZ cert_reqsZ verify_modeZca_certsZload_verify_locationsZciphersZ set_ciphers)r2 r@ r r r r4 � 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 rX r r r r3 � s z'GunicornWebWorker._get_valid_log_format)N)�__name__� __module__� __qualname__r Z LOG_FORMATrZ �GunicornAccessLogFormat�defaultrY r r r" r( r# r> r rB rS �intr rH rN � staticmethodr4 �strr3 � __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 rg )r r r r" � s z$GunicornUVLoopWebWorker.init_process)r^ r_ r` r" rf 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 rh ri r r" )r rj )r r r r" s z#GunicornTokioWebWorker.init_process)r^ r_ r` r" rf r r )r r r s r )$�__doc__r r8 r[ rF r& �typesr �typingr r r r r Zgunicorn.configr ra Zgunicorn.workersr Zaiohttpr Zhelpersr Zweb_appr Zweb_logr rW rV � ImportError�object�__all__ZWorkerr r r r r r r �<module> s0 Q