File: //usr/local/lib/python3.10/dist-packages/anyio/__pycache__/from_thread.cpython-310.pyc
o
���gFD � @ s� d dl mZ d dlZd dlmZmZmZ d dlmZ d dl m
Z
m Z m Z d dl
mZmZ d dlmZ d dlmZmZmZ d d lmZ d d
lmZmZmZm Z m Z d d l m Z d d
l m!Z!m"Z"m#Z# d dl$m%Z% d dl&m'Z'm(Z( d dl)m*Z* d dl+m,Z, ej-dkr�d dlm.Z.m/Z/ nd dl0m.Z.m/Z/ ed�Z1eddd�Z2e.d�Z3d6d d �Z4d7d!d"�Z5G d#d$� d$ee2 e �Z6G d%d&� d&e,�Z7G d'd(� d(�Z8eG d)d*� d*��Z9e d8d9d1d2��Z:d:d4d5�Z;dS );� )� annotationsN)� Awaitable�Callable� Generator)�Future)�AbstractAsyncContextManager�AbstractContextManager�contextmanager)� dataclass�field)� isawaitable)�Lock�Thread� get_ident)�
TracebackType)�Any�Generic�TypeVar�cast�overload� )�
_eventloop)�get_async_backend�get_cancelled_exc_class� threadlocals)�Event)� CancelScope�create_task_group)� AsyncBackend)�
TaskStatus)� � )� TypeVarTuple�Unpack�T_Retval�T_coT)� covariant�PosArgsT�func�1Callable[[Unpack[PosArgsT]], Awaitable[T_Retval]]�args�Unpack[PosArgsT]�returnc G �: zt j}t j}W n ty td�d�w |j| ||d�S )z�
Call a coroutine function from a worker thread.
:param func: a coroutine function
:param args: positional arguments for the callable
:return: the return value of the coroutine function
�9This function can only be run from an AnyIO worker threadN��token)r �current_async_backend�
current_token�AttributeError� RuntimeError�run_async_from_thread�r( r* �
async_backendr0 � r8 �</usr/local/lib/python3.10/dist-packages/anyio/from_thread.py�run( �
���r: �&Callable[[Unpack[PosArgsT]], T_Retval]c G r- )z�
Call a function in the event loop thread from a worker thread.
:param func: a callable
:param args: positional arguments for the callable
:return: the return value of the callable
r. Nr/ )r r1 r2 r3 r4 �run_sync_from_threadr6 r8 r8 r9 �run_sync> r; r> c @ sZ e Zd ZU ded< ded< ded< dZded < d dd�Zd!dd�Zd"dd�Zd#d d �Zd S )$� _BlockingAsyncContextManagerz Future[T_co]�
_enter_futurezFuture[bool | None]� _exit_futurer � _exit_event)NNNzMtuple[type[BaseException] | None, BaseException | None, TracebackType | None]�_exit_exc_info�async_cm�!AbstractAsyncContextManager[T_co]�portal�BlockingPortalc C s || _ || _d S �N)� _async_cm�_portal)�selfrD rF r8 r8 r9 �__init__\ s
z%_BlockingAsyncContextManager.__init__r, � bool | Nonec
� s� �zt � | _| j�� I d H }W n ty" } z| j�|� � d }~ww | j�|� z| j�� I d H W | jj | j
� I d H }|S | jj | j
� I d H }| Y S rH ) r rB rI �
__aenter__�
BaseExceptionr@ �
set_exception�
set_result�wait� __aexit__rC )rK �value�exc�resultr8 r8 r9 � run_async_cmb s � �� �z)_BlockingAsyncContextManager.run_async_cmr% c C s" t � | _| j�| j�| _| j�� S rH )r r@ rJ �start_task_soonrW rA rV �rK r8 r8 r9 � __enter__y s
z&_BlockingAsyncContextManager.__enter__�&_BlockingAsyncContextManager__exc_type�type[BaseException] | None�'_BlockingAsyncContextManager__exc_value�BaseException | None�'_BlockingAsyncContextManager__traceback�TracebackType | Nonec C s& |||f| _ | j�| jj� | j�� S rH )rC rJ �callrB �setrA rV )rK r[ r] r_ r8 r8 r9 �__exit__~ s
z%_BlockingAsyncContextManager.__exit__N)rD rE rF rG )r, rM )r, r% )r[ r\ r] r^ r_ r` r, rM ) �__name__�
__module__� __qualname__�__annotations__rC rL rW rZ rc r8 r8 r8 r9 r? T s
�
r? c @ s" e Zd Zd dd�Zd
dd
d �ZdS )�_BlockingPortalTaskStatus�futurer c C s
|| _ d S rH )�_future)rK ri r8 r8 r9 rL � �
z"_BlockingPortalTaskStatus.__init__NrT �objectr, �Nonec C s | j �|� d S rH )rj rQ )rK rT r8 r8 r9 �started� s z!_BlockingPortalTaskStatus.started)ri r rH )rT rl r, rm )rd re rf rL rn r8 r8 r8 r9 rh � s
rh c @ s� e Zd ZdZdAdd�ZdBdd�ZdAdd �ZdCdd�ZdBdd�ZdBdd�Z dDdEdd�Z
dFd$d%�Z dGd(d)�Z e
dHd-d.��Ze
dId0d.��ZdJd1d.�Ze
d2d3�dKd4d5��Ze
d2d3�dLd6d5��Zd2d3�dMd7d5�Zd2d3�dNd:d;�ZdOd?d@�Zd2S )PrG zLAn object that lets external threads run code in an asynchronous event loop.r, c C s
t � �� S rH )r �create_blocking_portal)�clsr8 r8 r9 �__new__� rk zBlockingPortal.__new__rm c C s$ t � | _t� | _t� | _t� | _d S rH )r �_event_loop_thread_idr � _stop_eventr � _task_groupr �_cancelled_exc_classrY r8 r8 r9 rL � s zBlockingPortal.__init__c � s �| j �� I d H | S rH )rt rN rY r8 r8 r9 rN � s �zBlockingPortal.__aenter__�exc_typer\ �exc_valr^ �exc_tbr` rM c � s& �| � � I d H | j�|||�I d H S rH )�stoprt rS )rK rv rw rx r8 r8 r9 rS � s �zBlockingPortal.__aexit__c C s* | j d u r td��| j t� krtd��d S )NzThis portal is not runningz7This method cannot be called from the event loop thread)rr r4 r rY r8 r8 r9 �_check_running� s
��z BlockingPortal._check_runningc � s �| j �� I dH dS )z#Sleep until :meth:`stop` is called.N)rs rR rY r8 r8 r9 �sleep_until_stopped� s �z"BlockingPortal.sleep_until_stoppedF�cancel_remaining�boolc � s* �d| _ | j�� |r| jj�� dS dS )a.
Signal the portal to shut down.
This marks the portal as no longer accepting new calls and exits from
:meth:`sleep_until_stopped`.
:param cancel_remaining: ``True`` to cancel all the remaining tasks, ``False``
to let them finish before returning
N)rr rs rb rt � cancel_scope�cancel)rK r| r8 r8 r9 ry � s �
�zBlockingPortal.stopr( �<Callable[[Unpack[PosArgsT]], Awaitable[T_Retval] | T_Retval]r* �tuple[Unpack[PosArgsT]]�kwargs�dict[str, Any]ri �Future[T_Retval]c
� s& �d� �fdd� }z�z6||i |��}t |�r=t� �� |�� r#� �� n|�|� |I d H }W d � n1 s7w Y n|}W n2 �jyR |�� |�� Y n4 tyr } z|�� sb|�|� t |t
�sh� W Y d }~n d }~ww |�� s�|� |� W d � d S W d � d S W d � d S W d � d S d � w )N�fr� r, rm c s0 | � � r�jd t� fvr��� j� d S d S d S rH )� cancelledrr r ra r )r� ��scoperK r8 r9 �callback� s ��z+BlockingPortal._call_func.<locals>.callback)r� r� r, rm ) r r r� r �add_done_callbackru � set_running_or_notify_cancelrO rP �
isinstance� ExceptionrQ ) rK r( r* r� ri r� �retval_or_awaitable�retvalrU r8 r� r9 �
_call_func� sD �
���
��� �� �zBlockingPortal._call_func�namerl c C s t �)a%
Spawn a new task using the given callable.
Implementors must ensure that the future is resolved when the task finishes.
:param func: a callable
:param args: positional arguments to be passed to the callable
:param kwargs: keyword arguments to be passed to the callable
:param name: name of the task (will be coerced to a string if not ``None``)
:param future: a future that will resolve to the return value of the callable,
or the exception raised during its execution
)�NotImplementedError)rK r( r* r� r� ri r8 r8 r9 �_spawn_task_from_thread� s z&BlockingPortal._spawn_task_from_threadr) r+ r$ c G � d S rH r8 �rK r( r* r8 r8 r9 ra s zBlockingPortal.callr< c G r� rH r8 r� r8 r8 r9 ra s c G s t t| j|g|�R � �� �S )a3
Call the given function in the event loop thread.
If the callable returns a coroutine object, it is awaited on.
:param func: any callable
:raises RuntimeError: if the portal is not running or if this method is called
from within the event loop thread
)r r$ rX rV r� r8 r8 r9 ra s N)r� c G r� rH r8 �rK r( r� r* r8 r8 r9 rX $ � z BlockingPortal.start_task_soonc G r� rH r8 r� r8 r8 r9 rX , r� c G s$ | � � t� }| �||i ||� |S )a�
Start a task in the portal's task group.
The task will be run inside a cancel scope which can be cancelled by cancelling
the returned future.
:param func: the target function
:param args: positional arguments passed to ``func``
:param name: name of the task (will be coerced to a string if not ``None``)
:return: a future that resolves with the return value of the callable if the
task completes successfully, or with the exception raised in the task
:raises RuntimeError: if the portal is not running or if this method is called
from within the event loop thread
:rtype: concurrent.futures.Future[T_Retval]
.. versionadded:: 3.0
)rz r r� )rK r( r� r* r� r8 r8 r9 rX 4 s �"Callable[..., Awaitable[T_Retval]]� tuple[Future[T_Retval], Any]c sV d � fdd� }| � � t� � t� �}t� }|�|� | �||d|i||� |� �� fS )
a�
Start a task in the portal's task group and wait until it signals for readiness.
This method works the same way as :meth:`.abc.TaskGroup.start`.
:param func: the target function
:param args: positional arguments passed to ``func``
:param name: name of the task (will be coerced to a string if not ``None``)
:return: a tuple of (future, task_status_value) where the ``task_status_value``
is the value passed to ``task_status.started()`` from within the target
function
:rtype: tuple[concurrent.futures.Future[T_Retval], Any]
.. versionadded:: 3.0
ri r� r, rm c sP � � � s&| �� r� �� d S | �� r� �| �� � d S td�}� �|� d S d S )Nz1Task exited without calling task_status.started())�doner� r � exceptionrP r4 )ri rU ��task_status_futurer8 r9 � task_doneh s ��z,BlockingPortal.start_task.<locals>.task_done� task_statusN)ri r� r, rm )rz r rh r� r� rV )rK r( r� r* r� r� r� r8 r� r9 �
start_taskQ s
zBlockingPortal.start_task�cmrE � AbstractContextManager[T_co]c C s
t || �S )a�
Wrap an async context manager as a synchronous context manager via this portal.
Spawns a task that will call both ``__aenter__()`` and ``__aexit__()``, stopping
in the middle until the synchronous context manager exits.
:param cm: an asynchronous context manager
:return: a synchronous context manager
.. versionadded:: 2.1
)r? )rK r� r8 r8 r9 �wrap_async_context_manager| s
z)BlockingPortal.wrap_async_context_manager�r, rG �r, rm )rv r\ rw r^ rx r` r, rM )F)r| r} r, rm )
r( r� r* r� r� r� ri r� r, rm ) r( r� r* r� r� r� r� rl ri r� r, rm �r( r) r* r+ r, r$ �r( r< r* r+ r, r$ )r( r� r* r+ r, r$ )r( r) r* r+ r� rl r, r� )r( r< r* r+ r� rl r, r� )r( r� r* r+ r� rl r, r� )r( r� r* rl r� rl r, r� )r� rE r, r� )rd re rf �__doc__rq rL rN rS rz r{ ry r� r� r ra rX r� r� r8 r8 r8 r9 rG � s6
*
�� �!�+rG c @ s� e Zd ZU dZdZded< dZded< eded �Z d
ed < edd d
�Z
ded< edd�Z ded< eddd
�Z ded< d!dd�Z
d"d d �ZdS )#�BlockingPortalProvidera=
A manager for a blocking portal. Used as a context manager. The first thread to
enter this context manager causes a blocking portal to be started with the specific
parameters, and the last thread to exit causes the portal to be shut down. Thus,
there will be exactly one blocking portal running in this context as long as at
least one thread has entered this context manager.
The parameters are the same as for :func:`~anyio.run`.
:param backend: name of the backend
:param backend_options: backend options
.. versionadded:: 4.4
�asyncio�str�backendN�dict[str, Any] | None�backend_optionsF)�init�default_factoryr
�_lockr )r� �default�int�_leases)r� rG rJ z-AbstractContextManager[BlockingPortal] | None�
_portal_cmr, c C sd | j �% | jd u rt| j| j�| _| j�� | _| jd7 _| jW d � S 1 s+w Y d S )Nr )r� r� �start_blocking_portalr� r� rZ rJ r� rY r8 r8 r9 rZ � s
� $�z BlockingPortalProvider.__enter__rv r\ rw r^ rx r` rm c C s� d }| j �& | js J �| jdksJ �| jd8 _| js$| j}d | _| `W d � n1 s.w Y |r>|�d d d � d S d S )Nr r )r� r� r� rJ rc )rK rv rw rx � portal_cmr8 r8 r9 rc � s
� � �z BlockingPortalProvider.__exit__r� )rv r\ rw r^ rx r` r, rm )rd re rf r� r� rg r� r r
r� r� rJ r� rZ rc r8 r8 r8 r9 r� � s
�
r� r� r� r� r� r� �$Generator[BlockingPortal, Any, None]c # s� � d �fdd� �d � ���fdd� }t � �t|dd�}|�� zLd }��� }z$z|V W n
ty6 d}� w W z |�|j|� W n tyJ Y nw z |�|j|� W w ty] Y w w W |�� d
S W |�� d
S |�� w ) a|
Start a new event loop in a new thread and run a blocking portal in its main task.
The parameters are the same as for :func:`~anyio.run`.
:param backend: name of the backend
:param backend_options: backend options
:return: a context manager that yields a blocking portal
.. versionchanged:: 3.0
Usage as a context manager is now required.
r, rm c � sZ �t � 4 I d H �} � �| � | �� I d H W d �I d H d S 1 I d H s&w Y d S rH )rG rQ r{ )�portal_)ri r8 r9 �
run_portal� s
�
.�z)start_blocking_portal.<locals>.run_portalc
sj �� � r3z tj�� �d� W d S ty2 } z��� s'��| � W Y d } ~ d S W Y d } ~ d S d } ~ ww d S )N)r� r� )r� r r: rO r� rP )rU �r� r� ri r� r8 r9 �run_blocking_portal� s �����z2start_blocking_portal.<locals>.run_blocking_portalT)�target�daemonFNr� ) r r �startrV rO ra ry r4 �join)r� r� r� �thread�cancel_remaining_tasksrF r8 r� r9 r� � s< �
�� �� �� �r� rm c C s0 zt j} W n ty td�d�w | �� dS )aa
Check if the cancel scope of the host task's running the current worker thread has
been cancelled.
If the host task's current cancel scope has indeed been cancelled, the
backend-specific cancellation exception will be raised.
:raises RuntimeError: if the current thread was not spawned by
:func:`.to_thread.run_sync`
r. N)r r1 r3 r4 �check_cancelled)r7 r8 r8 r9 r� � s
��� r� r� r� )r� N)r� r� r� r� r, r� r� )<�
__future__r �sys�collections.abcr r r �concurrent.futuresr �
contextlibr r r � dataclassesr
r �inspectr � threadingr
r r �typesr �typingr r r r r �_corer �_core._eventloopr r r �_core._synchronizationr � _core._tasksr r �abcr �
abc._tasksr � version_infor" r# �typing_extensionsr$ r% r' r: r> r? rh rG r� r� r� r8 r8 r8 r9 �<module> sD
5 ~8�4