Format: application/x-python-code
1 o
2
3 ;'hFD � @ s� d dl mZ d dlZd dlmZmZmZ d dlmZ d dl m
4 Z
5 m Z m Z d dl
6 mZmZ d dlmZ d dlmZmZmZ d d lmZ d d
7 lmZmZmZm Z m Z d d l m Z d d
8 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)�
9 TracebackType)�Any�Generic�TypeVar�cast�overload� )�
10 _eventloop)�get_async_backend�get_cancelled_exc_class� threadlocals)�Event)� CancelScope�create_task_group)� AsyncBackend)�
11 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�
12 Call a coroutine function from a worker thread.
13
14 :param func: a coroutine function
15 :param args: positional arguments for the callable
16 :return: the return value of the coroutine function
17
18 �9This function can only be run from an AnyIO worker threadN��token)r �current_async_backend�
19 current_token�AttributeError� RuntimeErrorZrun_async_from_thread�r( r* �
20 async_backendr0 � r7 �GC:\Users\shubh\Videos\agent\nova\Lib\site-packages\anyio/from_thread.py�run( �
21 ���r9 �&Callable[[Unpack[PosArgsT]], T_Retval]c G r- )z�
22 Call a function in the event loop thread from a worker thread.
23
24 :param func: a callable
25 :param args: positional arguments for the callable
26 :return: the return value of the callable
27
28 r. Nr/ )r r1 r2 r3 r4 Zrun_sync_from_threadr5 r7 r7 r8 �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]�
29 _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)�selfrB rD r7 r7 r8 �__init__\ s
30 z%_BlockingAsyncContextManager.__init__r, � bool | Nonec
31 � s� �zt � | _| j�� I d H }W n ty" } z| j�|� � d }~ww | j�|� z| j�� I d H W | jj | j
32 � I d H }|S | jj | j
33 � I d H }| Y S rF ) r r@ rG �
34 __aenter__�
35 BaseExceptionr> �
36 set_exception�
37 set_result�wait� __aexit__rA )rI �value�exc�resultr7 r7 r8 � run_async_cmb s � �� �z)_BlockingAsyncContextManager.run_async_cmr% c C s" t � | _| j�| j�| _| j�� S rF )r r> rH �start_task_soonrU r? rT �rI r7 r7 r8 � __enter__y s
38 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 rF )rA rH �callr@ �setr? rT )rI rY r[ r] r7 r7 r8 �__exit__~ s
39 z%_BlockingAsyncContextManager.__exit__N)rB rC rD rE )r, rK )r, r% )rY rZ r[ r\ r] r^ r, rK ) �__name__�
40 __module__� __qualname__�__annotations__rA rJ rU rX ra r7 r7 r7 r8 r= T s
41
42 �
43
44
45 r= c @ s" e Zd Zd dd�Zd
46 dd
47 d �ZdS )�_BlockingPortalTaskStatus�futurer c C s
48 || _ d S rF )�_future)rI rg r7 r7 r8 rJ � �
49 z"_BlockingPortalTaskStatus.__init__NrR �objectr, �Nonec C s | j �|� d S rF )rh rO )rI rR r7 r7 r8 �started� s z!_BlockingPortalTaskStatus.started)rg r rF )rR rj r, rk )rb rc rd rJ rl r7 r7 r7 r8 rf � s
50 rf c @ s� e Zd ZdZdAdd�ZdBdd�ZdAdd �ZdCdd�ZdBdd�ZdBdd�Z dDdEdd�Z
51 dFd$d%�Z dGd(d)�Z e
52 dHd-d.��Ze
53 dId0d.��ZdJd1d.�Ze
54 d2d3�dKd4d5��Ze
55 d2d3�dLd6d5��Zd2d3�dMd7d5�Zd2d3�dNd:d;�ZdOd?d@�Zd2S )PrE zLAn object that lets external threads run code in an asynchronous event loop.r, c C s
56 t � �� S rF )r Zcreate_blocking_portal)�clsr7 r7 r8 �__new__� ri zBlockingPortal.__new__rk c C s$ t � | _t� | _t� | _t� | _d S rF )r �_event_loop_thread_idr � _stop_eventr � _task_groupr �_cancelled_exc_classrW r7 r7 r8 rJ � s zBlockingPortal.__init__c � s �| j �� I d H | S rF )rq rL rW r7 r7 r8 rL � s �zBlockingPortal.__aenter__�exc_typerZ �exc_valr\ �exc_tbr^ rK c � s& �| � � I d H | j�|||�I d H S rF )�stoprq rQ )rI rs rt ru r7 r7 r8 rQ � 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)ro r4 r rW r7 r7 r8 �_check_running� s
57 ��z BlockingPortal._check_runningc � s �| j �� I dH dS )z#Sleep until :meth:`stop` is called.N)rp rP rW r7 r7 r8 �sleep_until_stopped� s �z"BlockingPortal.sleep_until_stoppedF�cancel_remaining�boolc � s* �d| _ | j�� |r| jj�� dS dS )a.
58 Signal the portal to shut down.
59
60 This marks the portal as no longer accepting new calls and exits from
61 :meth:`sleep_until_stopped`.
62
63 :param cancel_remaining: ``True`` to cancel all the remaining tasks, ``False``
64 to let them finish before returning
65
66 N)ro rp r` rq Z cancel_scope�cancel)rI ry r7 r7 r8 rv � s �
67 �zBlockingPortal.stopr( �<Callable[[Unpack[PosArgsT]], Awaitable[T_Retval] | T_Retval]r* �tuple[Unpack[PosArgsT]]�kwargs�dict[str, Any]rg �Future[T_Retval]c
68 � 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
69 �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, rk c s0 | � � r�jd t� fvr��� j� d S d S d S rF )� cancelledro r r_ r{ )r� ��scoperI r7 r8 �callback� s ��z+BlockingPortal._call_func.<locals>.callback)r� r� r, rk ) r r r� r{ �add_done_callbackrr � set_running_or_notify_cancelrM rN �
70 isinstance� ExceptionrO ) rI r( r* r~ rg r� Zretval_or_awaitable�retvalrS r7 r� r8 �
71 _call_func� sD �
72
73 ���
74
75 ��� �� �zBlockingPortal._call_func�namerj c C s t �)a%
76 Spawn a new task using the given callable.
77
78 Implementers must ensure that the future is resolved when the task finishes.
79
80 :param func: a callable
81 :param args: positional arguments to be passed to the callable
82 :param kwargs: keyword arguments to be passed to the callable
83 :param name: name of the task (will be coerced to a string if not ``None``)
84 :param future: a future that will resolve to the return value of the callable,
85 or the exception raised during its execution
86
87 )�NotImplementedError)rI r( r* r~ r� rg r7 r7 r8 �_spawn_task_from_thread� s z&BlockingPortal._spawn_task_from_threadr) r+ r$ c G � d S rF r7 �rI r( r* r7 r7 r8 r_ s zBlockingPortal.callr; c G r� rF r7 r� r7 r7 r8 r_ s c G s t t| j|g|�R � �� �S )a3
88 Call the given function in the event loop thread.
89
90 If the callable returns a coroutine object, it is awaited on.
91
92 :param func: any callable
93 :raises RuntimeError: if the portal is not running or if this method is called
94 from within the event loop thread
95
96 )r r$ rV rT r� r7 r7 r8 r_ s N)r� c G r� rF r7 �rI r( r� r* r7 r7 r8 rV $ � z BlockingPortal.start_task_soonc G r� rF r7 r� r7 r7 r8 rV , r� c G s$ | � � t� }| �||i ||� |S )a�
97 Start a task in the portal's task group.
98
99 The task will be run inside a cancel scope which can be cancelled by cancelling
100 the returned future.
101
102 :param func: the target function
103 :param args: positional arguments passed to ``func``
104 :param name: name of the task (will be coerced to a string if not ``None``)
105 :return: a future that resolves with the return value of the callable if the
106 task completes successfully, or with the exception raised in the task
107 :raises RuntimeError: if the portal is not running or if this method is called
108 from within the event loop thread
109 :rtype: concurrent.futures.Future[T_Retval]
110
111 .. versionadded:: 3.0
112
113 )rw r r� )rI r( r� r* r� r7 r7 r8 rV 4 s �"Callable[..., Awaitable[T_Retval]]� tuple[Future[T_Retval], Any]c sV d � fdd� }| � � t� � t� �}t� }|�|� | �||d|i||� |� �� fS )
114 a�
115 Start a task in the portal's task group and wait until it signals for readiness.
116
117 This method works the same way as :meth:`.abc.TaskGroup.start`.
118
119 :param func: the target function
120 :param args: positional arguments passed to ``func``
121 :param name: name of the task (will be coerced to a string if not ``None``)
122 :return: a tuple of (future, task_status_value) where the ``task_status_value``
123 is the value passed to ``task_status.started()`` from within the target
124 function
125 :rtype: tuple[concurrent.futures.Future[T_Retval], Any]
126
127 .. versionadded:: 3.0
128
129 rg r� r, rk c sP � � � s&| �� r� �� d S | �� r� �| �� � d S td�}� �|� d S d S )Nz1Task exited without calling task_status.started())�doner� r{ � exceptionrN r4 )rg rS �Ztask_status_futurer7 r8 � task_doneh s ��z,BlockingPortal.start_task.<locals>.task_done� task_statusN)rg r� r, rk )rw r rf r� r� rT )rI r( r� r* r� r� r� r7 r� r8 �
130 start_taskQ s
131 zBlockingPortal.start_task�cmrC � AbstractContextManager[T_co]c C s
132 t || �S )a�
133 Wrap an async context manager as a synchronous context manager via this portal.
134
135 Spawns a task that will call both ``__aenter__()`` and ``__aexit__()``, stopping
136 in the middle until the synchronous context manager exits.
137
138 :param cm: an asynchronous context manager
139 :return: a synchronous context manager
140
141 .. versionadded:: 2.1
142
143 )r= )rI r� r7 r7 r8 �wrap_async_context_manager| s
144 z)BlockingPortal.wrap_async_context_manager�r, rE �r, rk )rs rZ rt r\ ru r^ r, rK )F)ry rz r, rk )
145 r( r| r* r} r~ r rg r� r, rk ) r( r| r* r} r~ r r� rj rg r� r, rk �r( r) r* r+ r, r$ �r( r; r* r+ r, r$ )r( r| r* r+ r, r$ )r( r) r* r+ r� rj r, r� )r( r; r* r+ r� rj r, r� )r( r| r* r+ r� rj r, r� )r( r� r* rj r� rj r, r� )r� rC r, r� )rb rc rd �__doc__rn rJ rL rQ rw rx rv r� r� r r_ rV r� r� r7 r7 r7 r8 rE � s6
146
147
148
149
150
151
152
153 *
154 �� �!�+rE c @ s� e Zd ZU dZdZded< dZded< eded �Z d
155 ed < edd d
156 �Z
157 ded< edd�Z ded< eddd
158 �Z ded< d!dd�Z
159 d"d d �ZdS )#�BlockingPortalProvidera=
160 A manager for a blocking portal. Used as a context manager. The first thread to
161 enter this context manager causes a blocking portal to be started with the specific
162 parameters, and the last thread to exit causes the portal to be shut down. Thus,
163 there will be exactly one blocking portal running in this context as long as at
164 least one thread has entered this context manager.
165
166 The parameters are the same as for :func:`~anyio.run`.
167
168 :param backend: name of the backend
169 :param backend_options: backend options
170
171 .. versionadded:: 4.4
172 �asyncio�str�backendN�dict[str, Any] | None�backend_optionsF)�init�default_factoryr
173 �_lockr )r� �default�int�_leases)r� rE rH z-AbstractContextManager[BlockingPortal] | None�
174 _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� rX rH r� rW r7 r7 r8 rX � s
175 � $�z BlockingPortalProvider.__enter__rs rZ rt r\ ru r^ rk 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� rH ra )rI rs rt ru Z portal_cmr7 r7 r8 ra � s
176 � � �z BlockingPortalProvider.__exit__r� )rs rZ rt r\ ru r^ r, rk )rb rc rd r� r� re r� r r
177 r� r� rH r� rX ra r7 r7 r7 r8 r� � s
178 �
179 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
180 ty6 d}� w W z |�|j|� W n tyJ Y nw z |�|j|� W w ty] Y w w W |�� d
181 S W |�� d
182 S |�� w ) a|
183 Start a new event loop in a new thread and run a blocking portal in its main task.
184
185 The parameters are the same as for :func:`~anyio.run`.
186
187 :param backend: name of the backend
188 :param backend_options: backend options
189 :return: a context manager that yields a blocking portal
190
191 .. versionchanged:: 3.0
192 Usage as a context manager is now required.
193
194 r, rk 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 rF )rE rO rx )Zportal_)rg r7 r8 �
195 run_portal� s
196 �
197 .�z)start_blocking_portal.<locals>.run_portalc
198 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 r9 rM r� rN )rS �r� r� rg r� r7 r8 �run_blocking_portal� s �����z2start_blocking_portal.<locals>.run_blocking_portalT)�target�daemonFNr� ) r r �startrT rM r_ rv r4 �join)r� r� r� �threadZcancel_remaining_tasksrD r7 r� r8 r� � s< �
199
200 �� �� �� �r� rk c C s0 zt j} W n ty td�d�w | �� dS )aa
201 Check if the cancel scope of the host task's running the current worker thread has
202 been cancelled.
203
204 If the host task's current cancel scope has indeed been cancelled, the
205 backend-specific cancellation exception will be raised.
206
207 :raises RuntimeError: if the current thread was not spawned by
208 :func:`.to_thread.run_sync`
209
210 r. N)r r1 r3 r4 �check_cancelled)r6 r7 r7 r8 r� � s
211 ��� r� r� r� )r� N)r� r� r� r� r, r� r� )<�
212 __future__r �sys�collections.abcr r r �concurrent.futuresr �
213 contextlibr r r � dataclassesr
214 r �inspectr � threadingr
215 r r �typesr �typingr r r r r Z_corer Z_core._eventloopr r r Z_core._synchronizationr Z _core._tasksr r �abcr Z
216 abc._tasksr � version_infor" r# Ztyping_extensionsr$ r% r' r9 r< r= rf rE r� r� r� r7 r7 r7 r8 �<module> sD
217
218
219 5 ~8�4