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 Exceptionlog 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 }|}|d k st|j}|d k st| jD ]0}t j|||| jjd d d}| I d H qt }z| jr| |j } | jj rF| | jj krFd| _| j d| nB|t krz| j t! krzd| _| j d| n| " 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 {})loggerZkeepalive_timeout
access_logaccess_log_formatd _ )Z ssl_contextZshutdown_timeoutFz Max requests, shutting down: %sz!Parent changed, shutting down: %s)%
isinstanceZwsgir
r! iscoroutinefunction RuntimeErrorformatcfgZ accesslogr+ r2 r
Z AppRunnerZ keepalive_get_valid_log_formatr3 setupZis_ssl_create_ssl_contextAssertionErrorserverZsocketsZSockSiteZgraceful_timeoutstartosgetpidalivenotifyZrequests_countZ max_requestsinfoZppidgetppid_wait_next_notify
BaseExceptioncleanup)
r Zappr2 Zrunnerctxr? socksitepidZcntr r r r) B s\
zGunicornWebWorker._runzasyncio.Future[bool]c C s> | | j}|d k st| | _}| jd| j | |S )Ng ?)_notify_waiter_doner$ r> Z
create_futurer Z
call_later)r r$ waiterr r r rG z s z#GunicornWebWorker._wait_next_notify)rO r c C s4 |d kr| j }|d k r t|d || j kr0d | _ d S )NT)r r )r rO r r r rN 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 rP )rC r: Z
worker_intrN r r\ r] r r r rS s z GunicornWebWorker.handle_quitc C s& d| _ d| _| j| td d S )NFr )rC r r: Z worker_abortr- r/ r^ r r r rY s z GunicornWebWorker.handle_abort
SSLContext)r: 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.)
sslr8 r_ Z ssl_versionZload_cert_chainZcertfileZkeyfileZ cert_reqsZ verify_modeZca_certsZload_verify_locationsZciphersZ set_ciphers)r: rJ r r r r= 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 ra r r r r; s
z'GunicornWebWorker._get_valid_log_format)N)__name__
__module__ __qualname__r Z
LOG_FORMATrc GunicornAccessLogFormatdefaultrb r r r% r0 r) rG r rN r[ intr rS rY staticmethodr= strr;
__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 rp r r r r% s z$GunicornUVLoopWebWorker.init_processrg rh ri r% ro 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# rq rr r r% )r rt r r r r% s z#GunicornTokioWebWorker.init_processrs r r r r r s r )$__doc__r! rA rd rQ r- typesr typingr r r r r Zgunicorn.configr rj Zgunicorn.workersr aiohttpr
Zhelpersr Zweb_appr
Zweb_logr r` r_ ImportErrorobject__all__ZWorkerr r r r r r r