ruạṛ
a �u:j�J � @ s d dl mZ dZd dlZd dlmZmZmZ d dlm Z d dl m Z m Z m Z d dlmZmZ d dlmZ d d lmZ d d lmZmZmZmZ d d lmZ d d l m Z m Z m Z m Z m!Z! d dl"m#Z#m$Z$ d dl"m%Z& d dl'm(Z( d dl)m*Z* d dl+m,Z,m-Z- d dl.m/Z/ d dl0m1Z1m2Z2 ej3dk�r2d dl m4Z4m5Z5 nd dl6m4Z4m5Z5 e d�Z7e ddd�Z8e4d�Z9d d d �d d �Z:dd!�d"d#d dd$�d%d&�Z%dd!�d'd#d dd$�d(d)�Z;G d*d+� d+e e8 e �Z<G d,d-� d-e/�Z=G d.d/� d/�Z>eG d0d1� d1��Z?e d?dd3�d4d5d6d7d8�d9d:��Z@d;d<�d=d>�ZAdS )@� )� annotations)�BlockingPortal�BlockingPortalProvider�check_cancelled�run�run_sync�start_blocking_portalN)� Awaitable�Callable� Generator)�Future)�AbstractAsyncContextManager�AbstractContextManager�contextmanager)� dataclass�field)�partial)� isawaitable)�Lock�Thread�current_thread� get_ident)� TracebackType)�Any�Generic�TypeVar�cast�overload� )�get_cancelled_exc_class� threadlocals)r )�NoEventLoopError)�Event)� CancelScope�create_task_group)� TaskStatus)�EventLoopToken� current_token)� � )� TypeVarTuple�Unpack�T_Retval�T_coT)� covariant�PosArgsTzEventLoopToken | Noner& )�token�returnc C s6 | d ur | S zt jW S ty0 td�d �Y n0 d S )NzONot running inside an AnyIO worker thread, and no event loop token was provided)r r' �AttributeErrorr! �r0 � r4 �D/opt/bart/bart_venv/lib/python3.9/site-packages/anyio/from_thread.py�_token_or_error6 s ��r6 r3 �1Callable[[Unpack[PosArgsT]], Awaitable[T_Retval]]�Unpack[PosArgsT])�func�argsr0 r1 c G s, |du}t |�}|jj| ||r$|jndd�S )a Call a coroutine function from a worker thread. :param func: a coroutine function :param args: positional arguments for the callable :param token: an event loop token to use to get back to the event loop thread (required if calling this function from outside an AnyIO worker thread) :return: the return value of the coroutine function :raises MissingTokenError: if no token was provided and called from outside an AnyIO worker thread :raises RunFinishedError: if the event loop tied to ``token`` is no longer running .. versionchanged:: 4.11.0 Added the ``token`` parameter. Nr3 )r6 � backend_classZrun_async_from_thread� native_token�r9 r0 r: Zexplicit_tokenr4 r4 r5 r C s �r �&Callable[[Unpack[PosArgsT]], T_Retval]c G s, |du}t |�}|jj| ||r$|jndd�S )az Call a function in the event loop thread from a worker thread. :param func: a callable :param args: positional arguments for the callable :param token: an event loop token to use to get back to the event loop thread (required if calling this function from outside an AnyIO worker thread) :return: the return value of the callable :raises MissingTokenError: if no token was provided and called from outside an AnyIO worker thread :raises RunFinishedError: if the event loop tied to ``token`` is no longer running .. versionchanged:: 4.11.0 Added the ``token`` parameter. Nr3 )r6 r; Zrun_sync_from_threadr< r= r4 r4 r5 r _ s �r c @ sr e Zd ZU ded< ded< ded< dZded < d d d �d d�Zdd�dd�Zdd�dd�Zddddd�dd�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�!AbstractAsyncContextManager[T_co]r )�async_cm�portalc C s || _ || _d S �N)� _async_cm�_portal)�selfrE rF r4 r4 r5 �__init__� s z%_BlockingAsyncContextManager.__init__z bool | None�r1 c � s� z t � | _| j�� I d H }W n2 tyN } z| j�|� � W Y d }~nd }~0 0 | j�|� z(| j�� I d H W | jj | j � I d H }n| jj | j � I d H }0 |S rG ) r" rB rH � __aenter__� BaseExceptionr@ � set_exception� set_result�wait� __aexit__rC )rJ �value�exc�resultr4 r4 r5 � run_async_cm� s ,z)_BlockingAsyncContextManager.run_async_cmr- c C s" t � | _| j�| j�| _| j�� S rG )r r@ rI �start_task_soonrV rA rU �rJ r4 r4 r5 � __enter__� s z&_BlockingAsyncContextManager.__enter__�type[BaseException] | None�BaseException | None�TracebackType | None)�&_BlockingAsyncContextManager__exc_type�'_BlockingAsyncContextManager__exc_value�'_BlockingAsyncContextManager__tracebackr1 c C s& |||f| _ | j�| jj� | j�� S rG )rC rI �callrB �setrA rU )rJ r] r^ r_ r4 r4 r5 �__exit__� s z%_BlockingAsyncContextManager.__exit__N) �__name__� __module__� __qualname__�__annotations__rC rK rV rY rb r4 r4 r4 r5 r? { s � r? c @ s, e Zd Zdd�dd�Zd ddd�d d �ZdS ) �_BlockingPortalTaskStatusr )�futurec C s || _ d S rG )�_future)rJ rh r4 r4 r5 rK � s z"_BlockingPortalTaskStatus.__init__N�object�None)rS r1 c C s | j �|� d S rG )ri rP )rJ rS r4 r4 r5 �started� s z!_BlockingPortalTaskStatus.started)N)rc rd re rK rl r4 r4 r4 r5 rg � s rg c @ sZ e Zd ZdZdd�dd�Zd d�dd�Zdd d d d �d d�Zdd�dd�Zdd�dd�Zd;d dd�dd�Z dddddd�d d �Z dddd ddd �d d!�Z e d"d#d$d%�d&d'��Z e d(d#d$d%�d)d'��Z dd#d$d%�d*d'�Z e d+d,�d"d#d dd-�d.d/��Ze d+d,�d(d#d dd-�d0d/��Zd+d,�dd#d dd-�d1d/�Zd+d,�d2d d d3d-�d4d5�Zd6d7d8�d9d:�Zd+S )<r z� An object that lets external threads run code in an asynchronous event loop. :raises NoEventLoopError: if no supported asynchronous event loop is running in the current thread rk rL c C s$ t � | _t� | _t� | _t� | _d S rG )r' �_tokenr �_event_loop_thread_idr" � _stop_eventr$ � _task_grouprX r4 r4 r5 rK � s zBlockingPortal.__init__c � s | j �� I d H | S rG )rp rM rX r4 r4 r5 rM � s zBlockingPortal.__aenter__rZ r[ r\ �bool��exc_type�exc_val�exc_tbr1 c � s$ | � � I d H | j�|||�I d H S rG )�stoprp rR )rJ rs rt ru r4 r4 r5 rR � s zBlockingPortal.__aexit__c C s* | j d u rtd��| j t� kr&td��d S )NzThis portal is not runningz7This method cannot be called from the event loop thread)rn � RuntimeErrorr rX r4 r4 r5 �_check_running� s �z BlockingPortal._check_runningc � s | j �� I dH dS )z#Sleep until :meth:`stop` is called.N)ro rQ rX r4 r4 r5 �sleep_until_stopped� s z"BlockingPortal.sleep_until_stoppedF)�cancel_remainingr1 c � s& d| _ | j�� |r"| jj�d� 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 Nz$the blocking portal is shutting down)rn ro ra rp Z cancel_scope�cancel)rJ rz r4 r4 r5 rv � s zBlockingPortal.stopz<Callable[[Unpack[PosArgsT]], Awaitable[T_Retval] | T_Retval]ztuple[Unpack[PosArgsT]]zdict[str, Any]�Future[T_Retval])r9 r: �kwargsrh r1 c � s� ddd�� �fdd� }z�zZ||i |��}t |�rlt� �$� |�|� |I d H }W d � qp1 s`0 Y n|}W nd t� y� |�� |�� Y nT ty� } z*|�� s�|�|� t |t �s W Y d }~n d }~0 0 |�� s�|� |� W d � nd � 0 d S )Nr| rk )�fr1 c s< | � � r8�jt� kr � �d� n�jd ur8��� jd� d S )Nzthe future was cancelled)� cancelledrn r r{ r` )r~ ��scoperJ r4 r5 �callback� s z+BlockingPortal._call_func.<locals>.callback) r r# �add_done_callbackr r{ � set_running_or_notify_cancelrN r rO � isinstance� ExceptionrP ) rJ r9 r: r} rh r� Zretval_or_awaitable�retvalrT r4 r� r5 � _call_func� s& * zBlockingPortal._call_funcrj )r9 r: r} �namerh r1 c C s* t t| jj|d�| j||||| jd� dS )a� Spawn a new task using the given callable. :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 �r� r3 N)r r rp Z start_soonr� rm )rJ r9 r: r} r� rh r4 r4 r5 �_spawn_task_from_thread s �z&BlockingPortal._spawn_task_from_threadr7 r8 r, )r9 r: r1 c G s d S rG r4 �rJ r9 r: r4 r4 r5 r` 3 s zBlockingPortal.callr> c G s d S rG r4 r� r4 r4 r5 r` : 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, rW rU r� r4 r4 r5 r` ? s Nr� )r9 r: r� r1 c G s d S rG r4 �rJ r9 r� r: r4 r4 r5 rW P s z BlockingPortal.start_task_soonc G s d S rG r4 r� r4 r4 r5 rW X s 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 )rx r r� )rJ r9 r� r: r~ r4 r4 r5 rW ` s z"Callable[..., Awaitable[T_Retval]]z tuple[Future[T_Retval], Any]c s\ ddd�� 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 r| rk )rh r1 c sH � � � sD| �� r� �� n*| �� r2� �| �� � ntd�}� �|� d S )Nz1Task exited without calling task_status.started())�doner r{ � exceptionrO rw )rh rT �Ztask_status_futurer4 r5 � task_done� s �z,BlockingPortal.start_task.<locals>.task_done� task_status)rx r rg r� r� rU )rJ r9 r� r: r� r� r~ r4 r� r5 � start_task} s zBlockingPortal.start_taskrD z AbstractContextManager[T_co])�cmr1 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? )rJ r� r4 r4 r5 �wrap_async_context_manager� s z)BlockingPortal.wrap_async_context_manager)F)rc rd re �__doc__rK rM rR rx ry rv r� r� r r` rW r� r� r4 r4 r4 r5 r � s2 & �� �!�+r c @ s� e Zd ZU dZdZded< dZded< eded �Z d ed < edd d �Z ded< edd�Z ded< eddd �Z ded< dd�dd�Z ddddd �d d �ZdS ) r a= 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� r rI z-AbstractContextManager[BlockingPortal] | None� _portal_cmrL c C sd | j �J | jd u r.t| j| j�| _| j�� | _| jd7 _| jW d � S 1 sV0 Y d S )Nr )r� r� r r� r� rY rI r� rX r4 r4 r5 rY � s � z BlockingPortalProvider.__enter__rZ r[ r\ rk rr c C s| d }| j �L | jsJ �| jdks$J �| jd8 _| jsH| j}d | _| `W d � n1 s\0 Y |rx|�d d d � d S )Nr r )r� r� r� rI rb )rJ rs rt ru Z portal_cmr4 r4 r5 rb � s "z BlockingPortalProvider.__exit__)rc rd re r� r� rf r� r r r� r� rI r� rY rb r4 r4 r4 r5 r � s � r r� r� r� r� z str | Nonez$Generator[BlockingPortal, Any, None])r� r� r� r1 c # s� dd�� ��fdd� �dd�� ���fdd� }t � �t|d�d�}|�� z�d }��� }zNz |V W n ty| d}� Y n0 W z|�|j|� W q� ty� Y q�0 n(z|�|j|� W n ty� Y n0 0 W |�� n |�� 0 d 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 :param name: name of the thread :return: a context manager that yields a blocking portal .. versionchanged:: 3.0 Usage as a context manager is now required. rk rL c � sv t � 4 I d H �N} �d u r0� � dt| �d��t� _��| � | �� I d H W d �I d H qr1 I d H sh0 Y d S )Nz-portal-�x)r �idr r� rP ry )Zportal_)r� rh r� r4 r5 � run_portal s z)start_blocking_portal.<locals>.run_portalc sV �� � rRzt�� �d� W n6 tyP } z ��� s<��| � W Y d } ~ n d } ~ 0 0 d S )N)r� r� )r� � run_eventlooprN r� rO )rT )r� r� rh r� r4 r5 �run_blocking_portal s � z2start_blocking_portal.<locals>.run_blocking_portalT)�target�daemonr� FN) r r �startrU rN r` rv rw �join)r� r� r� r� �threadZcancel_remaining_tasksrF r4 )r� r� rh r� r� r5 r � s, � r rk rL c C s6 z t j} W n ty& td�d�Y n0 | j�� 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` z>This function can only be called inside an AnyIO worker threadN)r r' r2 r! r; r r3 r4 r4 r5 r / s ��r )r� N)B� __future__r �__all__�sys�collections.abcr r r �concurrent.futuresr � contextlibr r r � dataclassesr r � functoolsr �inspectr � threadingr r r r �typesr �typingr r r r r Z_core._eventloopr r r r� Z_core._exceptionsr! Z_core._synchronizationr" Z _core._tasksr# r$ Z abc._tasksr% Zlowlevelr& r' � version_infor* r+ Ztyping_extensionsr, r- r/ r6 r r? rg r r r r r4 r4 r4 r5 �<module> sV � � 6 8 ��;
cải xoăn