ܲe=ddlmZddlZddlZddlmZmZmZddlm Z m Z m Z m Z ddl mZmZddlmZddlmZddlmZmZmZmZmZmZmZmZd d lmZd d lm Z m!Z!m"Z"d d l#m$Z$d d l%m&Z&m'Z'd dl(m)Z)d dl*m+Z+ej,dkr ddlm-Z-m.Z.nddl/m-Z-m.Z.edZ0eddZ1e-dZ2d.dZ3d/dZ4Gdd ee1eZ5Gd!d"e+Z6Gd#d$Z7e d0d1d+Z8d2d-Z9dS)3) annotationsN) AwaitableCallable Generator)FIRST_COMPLETEDFutureThreadPoolExecutorwait)AbstractContextManagercontextmanager) isawaitable) TracebackType)AnyAsyncContextManagerContextManagerGenericIterableTypeVarcastoverload) _eventloop)get_async_backendget_cancelled_exc_class threadlocals)Event) CancelScopecreate_task_group) AsyncBackend) TaskStatus) ) TypeVarTupleUnpackT_RetvalT_coT) covariantPosArgsTfunc1Callable[[Unpack[PosArgsT]], Awaitable[T_Retval]]argsUnpack[PosArgsT]returnc tj}tj}n#t$rt ddwxYw||||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 threadNtoken)rcurrent_async_backend current_tokenAttributeError RuntimeErrorrun_async_from_threadr)r+ async_backendr1s 1lib/python3.11/site-packages/anyio/from_thread.pyrunr:&si$: *  G     . .tT . G GG6&Callable[[Unpack[PosArgsT]], T_Retval]c tj}tj}n#t$rt ddwxYw||||S)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/Nr0)rr2r3r4r5run_sync_from_threadr7s r9run_syncr?<si$: *  G     - -dD - F FFr;c\eZdZUded<ded<ded<dZded <ddZddZddZddZdS) _BlockingAsyncContextManagerz Future[T_co] _enter_futurezFuture[bool | None] _exit_futurer _exit_event)NNNzMtuple[type[BaseException] | None, BaseException | None, TracebackType | None]_exit_exc_infoasync_cmAsyncContextManager[T_co]portalBlockingPortalc"||_||_dSN) _async_cm_portal)selfrFrHs r9__init__z%_BlockingAsyncContextManager.__init__Zs! r- bool | NonecK t|_|jd{V}|j|n-#t $r }|j|d}~wwxYw |jd{V|jj |j d{V}|S#|jj |j d{V}|ccYSxYwrK) rrDrL __aenter__rB set_result BaseException set_exceptionr __aexit__rE)rNvalueexcresults r9 run_async_cmz)_BlockingAsyncContextManager.run_async_cm^s* 1$wwD .3355555555E   ) )% 0 0 0 0       , ,S 1 1 1    "'')) ) ) ) ) ) ) ) 44>3T5HIIIIIIIFM44>3T5HIIIIIIIFMMMMMMMMs#2A A;A66A;?B::Cr&ct|_|j|j|_|jSrK)rrBrMstart_task_soonr[rCrZrNs r9 __enter__z&_BlockingAsyncContextManager.__enter__us>#XX L889JKK!((***rP&_BlockingAsyncContextManager__exc_typetype[BaseException] | None'_BlockingAsyncContextManager__exc_valueBaseException | None'_BlockingAsyncContextManager__tracebackTracebackType | Nonec|||f|_|j|jj|jSrK)rErMcallrDsetrCrZ)rNr`rbrds r9__exit__z%_BlockingAsyncContextManager.__exit__zsD )+{B $*./// '')))rPN)rFrGrHrI)r-rQ)r-r&)r`rarbrcrdrer-rQ) __name__ __module__ __qualname____annotations__rErOr[r_rirPr9rArARs%%%% .++++ ******rPrAc eZdZd dZd d d ZdS) _BlockingPortalTaskStatusfuturerc||_dSrK)_future)rNrqs r9rOz"_BlockingPortalTaskStatus.__init__s  rPNrXobjectr-Nonec:|j|dSrK)rsrT)rNrXs r9startedz!_BlockingPortalTaskStatus.starteds &&&&&rP)rqrrK)rXrtr-ru)rjrkrlrOrwrnrPr9rprpsA'''''''rPrpceZdZdZd4dZd5dZd4dZd6dZd5dZd5dZ d7d8dZ d9dZ d:d Z e d;d$Ze dd*Ze d(d)d?d+Zd(d)d@d,Zd(d)dAd/ZdBd3Zd(S)CrIzLAn object that lets external threads run code in an asynchronous event loop.r-cBtSrK)rcreate_blocking_portal)clss r9__new__zBlockingPortal.__new__s ""99;;;rPructj|_t|_t |_t|_dSrK) threading get_ident_event_loop_thread_idr _stop_eventr _task_groupr_cancelled_exc_classr^s r9rOzBlockingPortal.__init__sA1:1D1F1F" 77,..$;$=$=!!!rPcHK|jd{V|SrK)rrSr^s r9rSzBlockingPortal.__aenter__s2))+++++++++ rPexc_typeraexc_valrcexc_tbrerQc~K|d{V|j|||d{VSrK)stoprrW)rNrrrs r9rWzBlockingPortal.__aexit__sU iikk%//'6JJJJJJJJJrPc|jtd|jtjkrtddS)NzThis portal is not runningz7This method cannot be called from the event loop thread)rr5r~rr^s r9_check_runningzBlockingPortal._check_runningsN  % -;<< <  %)<)>)> > >I  ? >rPcHK|jd{VdS)z#Sleep until :meth:`stop` is called.N)rr r^s r9sleep_until_stoppedz"BlockingPortal.sleep_until_stoppeds3##%%%%%%%%%%%rPFcancel_remainingboolcKd|_|j|r |jjdSdS)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)rrrhr cancel_scopecancel)rNrs r9rzBlockingPortal.stopsU&*"   3   ) 0 0 2 2 2 2 2 3 3rPr).callbacksg{{}} (!;#%%D"" %,'''''  ( (""rP)rrr-ru) r rrradd_done_callbackrTrset_running_or_notify_cancelrUrV isinstance Exception) rNr)r+rrqrretval_or_awaitableretvalrYrs ` @r9 _call_funczBlockingPortal._call_funcs ( ( ( ( ( ( ( "&$"7"7"7 .// - ]]7e'')); 00:::#6666666F 777777777777777-##%% *!!&)))( 2 2 2 MMOOO  / / 1 1 1 1 1   ##%% *$$S)))c9--        EEEDELLLLs_%B=AB9 B=B  B= B  B=*E =2E/E 1 E:?D>9E >EE Enamertct)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)rNr)r+rrrqs r9_spawn_task_from_threadz&BlockingPortal._spawn_task_from_threads *"!rPr*r,r%cdSrKrnrNr)r+s r9rgzBlockingPortal.calls rPr<cdSrKrnrs r9rgzBlockingPortal.call s  rPchtt|j|g|RS)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 )rr%r]rZrs r9rgzBlockingPortal.calls6H2d24?$???FFHHIIIrPN)rcdSrKrnrNr)rr+s r9r]zBlockingPortal.start_task_soon"  rPcdSrKrnrs r9r]zBlockingPortal.start_task_soon+rrPc||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 )rrr)rNr)rr+rs r9r]zBlockingPortal.start_task_soon4s@0 $hh $$T4T1===rP"Callable[..., Awaitable[T_Retval]]tuple[Future[T_Retval], Any]cdfd }|tt}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 rqrr-rucJs|rdS|r)|dSt d}|dSdS)Nz1Task exited without calling task_status.started())donerr exceptionrVr5)rqrYtask_status_futures r9 task_donez,BlockingPortal.start_task..task_donehs%**,, :##%%:&--/////%%'':&44V5E5E5G5GHHHHH&KC'44S99999 : :rP task_status)rqrr-ru)rrrprrrZ)rNr)rr+rrrrs @r9 start_taskzBlockingPortal.start_taskQs. : : : : : : %+XX/0BCC HH I&&& $$T4-1MtUVWWW$++----rPcmrGContextManager[T_co]c"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 )rA)rNrs r9wrap_async_context_managerz)BlockingPortal.wrap_async_context_manager|s,B555rP)r-rIr-ru)rrarrcrrer-rQ)F)rrr-ru) r)rr+rrrrqrr-ru) r)rr+rrrrrtrqrr-rur)r*r+r,r-r%r)r<r+r,r-r%)r)rr+r,r-r%)r)r*r+r,rrtr-r)r)r<r+r,rrtr-r)r)rr+r,rrtr-r)r)rr+rtrrtr-r)rrGr-r)rjrkrl__doc__r|rOrSrWrrrrrrrgr]rrrnrPr9rIrIsVV<<<<>>>> KKKK&&&&33333 ((((T"""".   X    X JJJJ"       X        X  B ).).).).).).V666666rPrIasynciobackendstrbackend_optionsdict[str, Any] | None$Generator[BlockingPortal, Any, None]c#Kd fd }ttd5}|tj|||} t t tt|gtn7#t$r* | wxYw r }d} |Vn#t$rd}wxYw | |j|nA#t$rYn5wxYw# | |j|w#t$rYwwxYwxYw| d d d d S#1swxYwYd S) 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-rucKt4d{V}r/||d{Vdddd{VdS#1d{VswxYwYdSrK)rIrrTr)portal_rqs r9 run_portalz)start_blocking_portal..run_portals!## 4 4 4 4 4 4 4w2244 4!!'***11333333333 4 4 4 4 4 4 4 4 4 4 4 4 4 4 4 4 4 4 4 4 4 4 4 4 4 4 4 4 4 4sAA// A9<A9r)rr) return_whenFTNr)rr submitrr:r rrrrUrrrZrgrr5)rrrexecutor run_futurerHcancel_remaining_tasksrqs @r9start_blocking_portalrs3$444444 &,XXF A  (__ N + %    Xf% F';<<+         MMOOO         ;;== ]]__F%* "      )-&  KK -CDDDD#DKK -CDDDD#D ?s$E; 6BE;4B77-E;%C*)D)*C88D)=DE; D&#E;%D&&E;)E+EE E EE EE;;E?E?ruc tj}n#t$rtddwxYw|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)rr2r4r5check_cancelled)r8s r9rrs^&2&H  G    !!#####s *rr)rN)rrrrr-rr): __future__rsysr~collections.abcrrrconcurrent.futuresrrr r contextlibr r inspectr typesrtypingrrrrrrrr_corer_core._eventlooprrr_core._synchronizationr _core._tasksrrabcr abc._tasksr version_infor#r$typing_extensionsr%r&r(r:r?rArprIrrrnrPr9rs"""""" ::::::::::PPPPPPPPPPPP========                    VVVVVVVVVV))))))88888888""""""w+++++++++66666666 7:  wv&&& < # #HHHH,GGGG,0*0*0*0*0*74=2H0*0*0*f''''' '''~6~6~6~6~6~6~6~6BGK77777t$$$$$$rP