File: //home/arjun/projects/env/lib/python3.10/site-packages/aiohttp/__pycache__/worker.cpython-310.pyc
o
+we � @ s� d Z ddlZddlZddlZddlZddlZddlmZ ddlm Z m
Z
m Z m Z m
Z
ddlmZ ddlmZ ddlmZ dd lmZ dd
lmZ dd lmZ z ddlZej Z W n
e yd dZe Z Y nw d Z G d
d� dej �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�GunicornUVLoopWebWorkerc s� e Zd ZejZejZde de ddf� fdd� Z
d"� fdd� Z d"d d
�Z d"d d �Z
d#dd�Z d$ded
ddfdd�Zd"dd�Zdedee ddfdd�Zdedee ddfdd�Zede dd fd d ��Zd edefd d!�Z� ZS )%r �args�kw�returnNc s( t � j|i |�� d | _d| _d | _d S �Nr )�super�__init__�_task� exit_code�_notify_waiter)�selfr r �� __class__� �G/home/arjun/projects/env/lib/python3.10/site-packages/aiohttp/worker.pyr % s
zGunicornWebWorker.__init__c s0 t �� �� t �� | _t �| j� t� �� d S �N)�asyncio�get_event_loop�close�new_event_loop�loop�set_event_loopr � init_process�r r r r r& , s
z GunicornWebWorker.init_processc C sp | j �| �� �| _z | j �| j� W n ty! | j�d� Y nw | j �| j �� � | j � � t
� | j � d S )Nz Exception in gunicorn worker)
r$ � create_task�_runr �run_until_complete� Exception�log� exception�shutdown_asyncgensr" �sys�exitr r' r r r �run5 s �
zGunicornWebWorker.runc � s� �d }t | jt�r
| j}n$t�| j�r)| �� I d H }t |tj�r&|}|j}n |}ntd� | j���|d u rX| j
j r=| j j
nd }tj|| j | j
j|| �| j
j�| j
jd d d�}|�� I d H | j
jri| �| j
�nd }|}|d ussJ �|j}|d us|J �| jD ]}tj|||d�}|�� I d H qt�� } zA| jr�| � � |j }
| j r�|
| j kr�d| _| j � d| � n | t�� kr�| j t�!� kr�d| _| j � d| � n| �"� I d H | js�W n t#y� Y nw |�$� I d H d S ) NzUwsgi app should be either Application or async function returning Application, got {}�d �_ )�logger�keepalive_timeout�
access_log�access_log_format�shutdown_timeout)� ssl_contextFz Max requests, shutting down: %sz!Parent changed, shutting down: %s)%�
isinstance�wsgir
r �iscoroutinefunctionr
� AppRunner�app� RuntimeError�format�cfg� accesslogr, r6 � keepalive�_get_valid_log_formatr7 �graceful_timeout�setup�is_ssl�_create_ssl_context�server�sockets�SockSite�start�os�getpid�alive�notify�requests_count� max_requests�info�ppid�getppid�_wait_next_notify�
BaseException�cleanup) r �runnerr> r; r6 �ctxrI �sock�site�pid�cntr r r r) A sr � ���
���
�zGunicornWebWorker._run�asyncio.Future[bool]c C s> | � � | j}|d us
J �|�� | _}| j�d| j |� |S )Ng �?)�_notify_waiter_doner$ �
create_futurer �
call_later)r r$ �waiterr r r rV � s z#GunicornWebWorker._wait_next_notifyrc c C s8 |d u r| j }|d urt|d� || j u rd | _ d S d S )NT)r r )r rc r r r r` � 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� d S �NF)r$ �add_signal_handler�signal�SIGQUIT� handle_quit�SIGTERM� handle_exit�SIGINT�SIGWINCH� handle_winch�SIGUSR1� handle_usr1�SIGABRT� handle_abort� siginterruptr' r r r � init_signals� s( ������z GunicornWebWorker.init_signals�sig�framec C s d| _ | j�| � | �� d S rd )rO rA �
worker_intr` �r rt ru r r r rh � s z GunicornWebWorker.handle_quitc C s& d| _ d| _| j�| � t�d� d S )NFr )rO r rA � worker_abortr/ r0 rw r r r rq � s z GunicornWebWorker.handle_abortrA �
SSLContextc C s\ t du rtd��t �| j�}|�| j| j� | j|_| j r#|�
| j � | j r,|� | j � |S )z~Creates SSLContext instance for usage in asyncio.create_server.
See ssl.SSLSocket.__init__ for more details.
NzSSL is not supported.)
�sslr? ry � ssl_version�load_cert_chain�certfile�keyfile� cert_reqs� verify_mode�ca_certs�load_verify_locations�ciphers� set_ciphers)rA rZ r r r rH � s z%GunicornWebWorker._create_ssl_context�
source_formatc C s( || j kr| jS t�d|�rtd��|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 rD � s
�z'GunicornWebWorker._get_valid_log_format�r N)r r_ r )�__name__�
__module__� __qualname__r �
LOG_FORMATr� �GunicornAccessLogFormat�defaultr� r r r& r1 r) rV r r` rs �intr rh rq � staticmethodrH �strrD �
__classcell__r r r r r s(
B ��
�
" r c s e Zd Zd� fdd� Z� ZS )r r Nc s0 dd l }t�� �� t�|�� � t� �� d S r )�uvloopr r! r" �set_event_loop_policy�EventLoopPolicyr r& )r r� r r r r& � s z$GunicornUVLoopWebWorker.init_processr� )r� r� r� r& r� r r r r r � s r )#�__doc__r rM r� rf r/ �typesr �typingr r r r r �gunicorn.configr r� �gunicorn.workersr �aiohttpr
�helpersr �web_appr
�web_logr rz ry � ImportError�object�__all__�Workerr r r r r r �<module> s2
� K