U
'7`V @ s d Z ddlZddlZddlZddlZddlZddlmZ ddlm Z m
Z
m Z m Z m
Z
ddlmZ ddlmZ ddlmZ dd lmZ dd
lmZ dd lmZ zddlZej 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 AwaitableCallableOptionalUnion)AccessLogFormat)base)web )
set_result) Application) AccessLogger)GunicornWebWorkerGunicornUVLoopWebWorkerGunicornTokioWebWorkerc s e Zd ZejZejZe e dd fdd Z
dd fdd Z dddd Z ddd
d Z
d dd
dZd ed ddddZddddZeeddddZeeddddZee dddd Zeed d d Z ZS )!r N)argskwreturnc s$ t j|| d | _d| _d | _d S Nr )super__init___task exit_code_notify_waiter)selfr r __class__ A/opt/alt/python38/lib64/python3.8/site-packages/aiohttp/worker.pyr % s zGunicornWebWorker.__init__r c s0 t t | _t | j t d S )N)asyncioget_event_loopcloseZnew_event_looploopZset_event_loopr init_processr r r r r% , s
z GunicornWebWorker.init_processc C s| | j | | _z| j | j W n tk
rD | jd Y nX tj dkrb| j | j
| j t | j
d S )Nz Exception in gunicorn worker) )r$ Z create_task_runr Zrun_until_complete ExceptionlogZ exceptionsys version_infoZshutdown_asyncgensr# exitr r& r r r run5 s
zGunicornWebWorker.runc
s t | jtr| j}n,t| jr0| I d H }ntd| j| jjrP| j j
nd }t j || j | jj
|| | jjd}| I d H | jjr| | jnd }|}|j}| jD ]0}t j|||| jjd d d}| I d H qt }z| jrr| |j } | jj r.| | jj kr.d| _| j d| q|t krb| j t krbd| _| j d| q| ! I d H qW 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_logaccess_log_formatd _ )Z ssl_contextZshutdown_timeoutFz Max requests, shutting down: %sz!Parent changed, shutting down: %s)$
isinstanceZwsgir
r! Ziscoroutinefunction RuntimeErrorformatcfgZ accesslogr+ r0 r
Z AppRunnerZ keepalive_get_valid_log_formatr1 ZsetupZis_ssl_create_ssl_contextserverZsocketsZSockSiteZgraceful_timeoutstartosgetpidaliveZnotifyZrequests_countZ max_requestsinfoZppidgetppid_wait_next_notify
BaseExceptionZcleanup)
r Zappr0 Zrunnerctxr: ZsockZsitepidZcntr r r r) B sX
zGunicornWebWorker._runzasyncio.Future[bool]c C s2 | | j}| | _}| jd| j | |S )Ng ?)_notify_waiter_doner$ Z
create_futurer Z
call_later)r r$ waiterr r r rA z s
z#GunicornWebWorker._wait_next_notify)rF r c C s4 |d kr| j }|d k r t|d || j kr0d | _ d S )NT)r r )r rF r r r rE 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 ttjd ttj
d d S NF)r$ Zadd_signal_handlersignalSIGQUIT handle_quitSIGTERMZ handle_exitSIGINTSIGWINCHZ handle_winchSIGUSR1Z handle_usr1SIGABRT handle_abort siginterruptr& r r r init_signals sL z GunicornWebWorker.init_signals)sigframer c C s d| _ | j| | d S rG )r> r7 Z
worker_intrE r rS rT r r r rJ s z GunicornWebWorker.handle_quitc C s& d| _ d| _| j| td d S )NFr )r> r r7 Z worker_abortr, r. rU r r r rP s z GunicornWebWorker.handle_abort
SSLContext)r7 r c C s\ t dkrtdt | 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.)
sslr5 rV Z ssl_versionZload_cert_chainZcertfileZkeyfileZ cert_reqsZ verify_modeZca_certsZload_verify_locationsZciphersZ set_ciphers)r7 rC r r r r9 s z%GunicornWebWorker._create_ssl_context)
source_formatr c C s. || j kr| jS td|r&tdn|S d S )Nz %\([^\)]+\)zGunicorn'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_FORMATDEFAULT_AIOHTTP_LOG_FORMATresearch
ValueError)r rX r r r r8 s
z'GunicornWebWorker._get_valid_log_format)N)__name__
__module__ __qualname__r Z
LOG_FORMATrZ GunicornAccessLogFormatdefaultrY r r r% r/ r) rA r rE rR intr rJ rP staticmethodr9 strr8
__classcell__r r r r r s$
8 r c s" e Zd Zdd fdd Z ZS )r Nr c s0 dd l }t t| t d S r )uvloopr! r" r# set_event_loop_policyEventLoopPolicyr r% )r rg r r r r% s z$GunicornUVLoopWebWorker.init_processr^ r_ r` r% rf r r r r r s r c s" e Zd Zdd fdd Z ZS )r Nr c s0 dd l }t t| t d S r )tokior! r" r# rh ri r r% )r rk r r r r% s z#GunicornTokioWebWorker.init_processrj r r r r r s r )$__doc__r! r< r[ rH r, typesr typingr r r r r Zgunicorn.configr ra Zgunicorn.workersr Zaiohttpr
Zhelpersr Zweb_appr
Zweb_logr rW rV ImportErrorobject__all__ZWorkerr r r r r r r <module> s0
@