File: //usr/local/CyberPanel/lib64/python3.10/site-packages/asgiref/__pycache__/sync.cpython-310.pyc
o
��h�O � @ s� U d dl Z d dlZ d dlZd dlZd dlZd dlZd dlZd dlZd dlZd dl Z d dl
m Z m Z d dl
mZmZmZmZmZmZmZmZmZmZmZmZ ddlmZ ddl m Z ej dkred dl
m Z nd dl m Z ersd d l!m"Z" ed
ed ef d �Z#e d
�Z$ed�Z%dej&ddfdd�Z'e(ed�r�ej)Z)ej*Z*ee#ge#f e+d< n e j)Z)de#de#fdd�Z*G dd� d�Z,G dd� dee$e%f �Z-G dd� dee$e%f �Z.ed d �d e/deeee$eeee%f f ee$ee% f f gee$e%f f fd d ��Z0ed d �d!eee$eeee%f f ee$ee% f f d e/dee$e%f fd"d ��Z0 d-d d �d!eeee$eeee%f f ee$ee% f f d e/deeeee$eeee%f f ee$ee% f f gee$e%f f ee$e%f f fd#d �Z0ed$dd%�d&e/d'ed( deee$e%f gee$eeee%f f f fd)d*��Z1ed$dd%�dee$e%f d&e/d'ed( dee$eeee%f f fd+d*��Z1 d-d$dd%�deee$e%f d&e/d'ed( deeee$e%f gee$eeee%f f f ee$eeee%f f f fd,d*�Z1dS ).� N)�Future�ThreadPoolExecutor) �
TYPE_CHECKING�Any� Awaitable�Callable� Coroutine�Dict�Generic�List�Optional�TypeVar�Union�overload� )�CurrentThreadExecutor)�Local)� �
)� ParamSpec)�
OptExcInfo�_F.)�bound�_P�_R�context�returnc C sN | D ]"}| � |�}z
|� � |kr|�|� W q ty$ |�|� Y qw d S �N)�get�set� LookupError)r �cvar�cvalue� r# �B/usr/local/CyberPanel/lib/python3.10/site-packages/asgiref/sync.py�_restore_context, s
� ��r% �markcoroutinefunction�funcc C s t jj| _| S r )�asyncio�
coroutines�
_is_coroutine)r' r# r# r$ r& C s
c @ s( e Zd ZdZdd� Zdd� Zdd� ZdS ) �ThreadSensitiveContextaY Async context manager to manage context for thread sensitive mode
This context manager controls which thread pool executor is used when in
thread sensitive mode. By default, a single thread pool executor is shared
within a process.
The ThreadSensitiveContext() context manager may be used to specify a
thread pool per context.
This context manager is re-entrant, so only the outer-most call to
ThreadSensitiveContext will set the context.
Usage:
>>> import time
>>> async with ThreadSensitiveContext():
... await sync_to_async(time.sleep, 1)()
c C s
d | _ d S r )�token��selfr# r# r$ �__init__\ s
z ThreadSensitiveContext.__init__c � s6 �zt j�� W | S ty t j�| �| _Y | S w r )� SyncToAsync�thread_sensitive_contextr r r r, r- r# r# r$ �
__aenter___ s � ��z!ThreadSensitiveContext.__aenter__c � s8 �| j sd S tj�| d �}|r|�� tj�| j � d S r )r, r0 �context_to_thread_executor�pop�shutdownr1 �reset)r. �exc�value�tb�executorr# r# r$ � __aexit__g s �z ThreadSensitiveContext.__aexit__N)�__name__�
__module__� __qualname__�__doc__r/ r2 r; r# r# r# r$ r+ H s
r+ c @ s� e Zd ZU dZe� Zded< i Zded< d dee e
e e e e
f f e e
ee
f f defd d
�Zd e
jd e
jd
e
fdd�Zde de d
e e
e
f fdd�Zdddddddeej dee e e e
f ee
f d
df d d �ZdS ) � AsyncToSynca�
Utility class which turns an awaitable that only works on the thread with
the event loop into a synchronous callable that works in a subthread.
If the call stack contains an async loop, the code runs there.
Otherwise, the code runs in a new loop in a new thread.
Either way, this thread then pauses and waits to run any thread_sensitive
code called from further down the call stack using SyncToAsync, before
finally exiting once the async task returns.
r � executorsz6Dict[asyncio.AbstractEventLoop, CurrentThreadExecutor]�loop_thread_executorsF� awaitable�force_new_loopc C s� t |�rt|�stt|d|��stjddd� || _z| jj| _W n ty* Y nw || _d | _ zt
� � | _ W d S t yC Y d S w )N�__call__z4async_to_sync was passed a non-async-marked callable� )�
stacklevel)
�callable�iscoroutinefunction�getattr�warnings�warnrC �__self__�AttributeErrorrD �main_event_loopr( �get_running_loop� RuntimeError)r. rC rD r# r# r$ r/ � s* ��� � �zAsyncToSync.__init__�args�kwargsr c
s� d}�j s �js ttjdd �}|r |t�� kr ttjdd ��_z t�� W t d�� t y1 Y nw t
� }t�j dd �}t |����j _
t�� g}ttjdd �}zv��|t�� ||�j|i |���� d� ��fdd � } �jd ur�z �j��jj� � W n t y� d
}
Y n w d}
��|� nd
}
|
s�td d �} | �tj| � �} ��| � | �� W t|d
� |�j _
|�� S W t|d
� |�j _
|�� S t|d
� |�j _
w )NT�main_event_loop_pidrO znYou cannot use AsyncToSync in the same thread as an async event loop - just await the async function directly.�current� task_contextr c � s8 �t �� } ��j| < z � I d H W �j| = d S �j| = w r )r( rP rB )�loop�rC �current_executorr. r# r$ �
new_loop_wrap� s �
z+AsyncToSync.__call__.<locals>.new_loop_wrapFr �� max_workersr )r N) rD rO rJ r0 � threadlocal�os�getpidr( rP rQ r rA r rU � contextvars� copy_context� main_wrap�sys�exc_inforC �call_soon_threadsafe� create_task�run_until_futurer �submit�run�resultr% )
r. rR rS �__traceback_hide__rT � call_result� old_executorr rV rZ �running_in_main_event_loop�
loop_executor� loop_futurer# rX r$ rE � sr ��
� ��
�
� �
� �
zAsyncToSync.__call__�parent�objtypec C � t �| j|�}t �|| j�S �z*
Include self for methods
)� functools�partialrE �update_wrapperrC �r. rq rr r' r# r# r$ �__get__ s zAsyncToSync.__get__rl z
Future[_R]rd r rV z!Optional[List[asyncio.Task[Any]]]r Nc
� s �d}|dur
t |d � t�� }|dur |dur |�|� zQz |d r7z|d � ty6 |I dH }Y nw |I dH }W n tyT } z |�| � W Y d} ~ n
d} ~ ww |�|� W |durh|durh|�|� t� � |d< dS |dur}|dur}|�|� t� � |d< w )zs
Wraps the awaitable with something that puts the result into the
result/exception future.
TNr r )
r% r( � current_task�append�
BaseException�
set_exception�
set_result�remover` ra )
r. rl rd rV r rC rk rz rj �er# r# r$ rb s8 �
�
���
�
�
zAsyncToSync.main_wrap)F)r< r= r> r? r rA �__annotations__rB r r r r r r r �boolr/ rR rS rE ry r r` �Contextrb r# r# r# r$ r@ q s:
����
� d������r@ c @ s� e Zd ZU dZe�� Zedd�Ze �
d�Z de d< e �
d�Z
de d< e�� Zde d <
d d eeef d
eded dd fdd�Zdejdejdefdd�Zdededeeeeeef f fdd�Z dd � Z d S ) r0 a
Utility class which turns a synchronous callable into an awaitable that
runs in a threadpool. It also sets a threadlocal inside the thread so
calls to AsyncToSync can escape it.
If thread_sensitive is passed, the code will run in the same thread as any
outer code. This is needed for underlying Python code that is not
threadsafe (for example, code which handles SQLite database connections).
If the outermost program is async (i.e. SyncToAsync is outermost), then
this will be a dedicated single sub-thread that all sync code runs in,
one after the other. If the outermost program is sync (i.e. AsyncToSync is
outermost), this will just be the main thread. This is achieved by idling
with a CurrentThreadExecutor while AsyncToSync is blocking its sync parent,
rather than just blocking.
If executor is passed in, that will be used instead of the loop's default executor.
In order to pass in an executor, thread_sensitive must be set to False, otherwise
a TypeError will be raised.
r r[ r1 z.contextvars.ContextVar[ThreadSensitiveContext]�deadlock_contextz contextvars.ContextVar[bool]zEweakref.WeakKeyDictionary[ThreadSensitiveContext, ThreadPoolExecutor]r3 TNr' �thread_sensitiver: r r c C s� t |�rt|�stt|d|��rtd��|| _t�| |� || _t| � |r.|d ur.td��|| _ z|j
| _
W d S t yB Y d S w )NrE z4sync_to_async can only be applied to sync functions.z6executor must not be set when thread_sensitive is True) rH rI rJ � TypeErrorr' ru rw �_thread_sensitiver& � _executorrM rN )r. r' r� r: r# r# r$ r/ g s& ��� �zSyncToAsync.__init__rR rS c � s� �d}t �� }| jrVttjdd �}|r|}nC| j�d �r7| j�� }|| jv r,| j| }n-t dd�}|| j|< n"|tj
v rBtj
| }n| j �d�rLt d��| j
}| j �d� n| j}t�� }tj| jg|�R i |��} |j}
g } |�|t�| j|t�� | |
| ��} ziz
t �| �I d H }
W nD t jy� d}z | d }|� � z |I d H d}W n
t jy� Y nw W n t y� Y nw | � � r |r�| � � | I d H }
Y nw W t |� | j �d� |
S W t |� | j �d� |
S t |� | j �d� w )NTrU r r[ Fz9Single thread executor already being used, would deadlockr ) r( rP r� rJ r@ rA r1 r r3 r rB r� rQ �single_thread_executorr r� r` ra ru rv r' ri �run_in_executor�thread_handlerrc rd �shield�CancelledError�cancel�
IndexError�doner% )r. rR rS rk rW �current_thread_executorr: r1 r �childr' rV � exec_coro�ret�
cancel_parent�taskr# r# r$ rE s� �
���
�� ��� � �zSyncToAsync.__call__rq rr c C rs rt )ru rv rE rw r' rx r# r# r$ ry � s zSyncToAsync.__get__c O s` d}|| j _t�� | j _|| j _|d r)z|d � ty( ||i |�� Y S w ||i |��S )zE
Wraps the sync application with exception handling.
Tr )r] rO r^ r_ rT rV r| )r. rW rd rV r' rR rS rk r# r# r$ r� � s �zSyncToAsync.thread_handler)TN) r< r= r> r? � threading�localr] r r� r` �
ContextVarr1 r� r� �weakref�WeakKeyDictionaryr3 r r r r� r r/ rR rS rE r r ry r� r# r# r# r$ r0 9 s>
� �
��
���
�P��
� r0 F�rD rD c C � d S r r# r� r# r# r$ �
async_to_sync� s r� rC c C r� r r# �rC rD r# r# r$ r� � s c s | d u r
� fdd�S t | � d�S )Nc s t | � d�S )Nr� �r@ ��fr� r# r$ �<lambda> s �z async_to_sync.<locals>.<lambda>r� r� r� r# r� r$ r� s �T�r� r: r� r: r c C r� r r# r� r# r# r$ �
sync_to_async" s r� c C r� r r# �r' r� r: r# r# r$ r� + s c s$ | d u r � �fdd�S t | �� d�S )Nc s t | �� d�S )Nr� �r0 r� �r: r� r# r$ r� ? s
�z sync_to_async.<locals>.<lambda>r� r� r� r# r� r$ r� 5 s �r )2r( �asyncio.coroutinesr` ru �inspectr^ rc r� rK r� �concurrent.futuresr r �typingr r r r r r r
r r r
r r r� r r� r � version_infor �typing_extensions� _typeshedr r r r r� r% �hasattrrI r& r� r+ r@ r0 r� r� r� r# r# r# r$ �<module> s�
8
) I 8��*
�� �
�����
�������� *
��
��
����& ��
��� �
�����&���