o
    +i                     @  s  U d dl mZ dZd dlZd dlmZ d dlmZ d dlm	Z	m
Z
mZmZmZmZ d dlmZ dd	lmZ dd
lmZ edZedZd$ddZd$ddZd$ddZeedddG dd dZd%ddZe Zded< G dd dejZG d d! d!e
e Z G d"d# d#e
e Z!dS )&    )annotations)EventLoopTokenRunvarTokenRunVar
checkpointcheckpoint_if_cancelledcancel_shielded_checkpointcurrent_tokenN)	dataclass)TracebackType)AnyGenericLiteralTypeVarfinaloverload)WeakKeyDictionary   )get_async_backend)AsyncBackendTDreturnNonec                        t   I dH  dS )z
    Check for cancellation and allow the scheduler to switch to another task.

    Equivalent to (but more efficient than)::

        await checkpoint_if_cancelled()
        await cancel_shielded_checkpoint()


    .. versionadded:: 3.0

    N)r   r    r   r   A/var/www/html/venv/lib/python3.10/site-packages/anyio/lowlevel.pyr         r   c                     r   )z
    Enter a checkpoint if the enclosing cancel scope has been cancelled.

    This does not allow the scheduler to switch to a different task.

    .. versionadded:: 3.0

    N)r   r   r   r   r   r   r   *   s   	r   c                     r   )z
    Allow the scheduler to switch to another task but without checking for cancellation.

    Equivalent to (but potentially more efficient than)::

        with CancelScope(shield=True):
            await checkpoint()


    .. versionadded:: 3.0

    N)r   r   r   r   r   r   r   6   r   r   TF)frozenreprc                   @  s"   e Zd ZU dZded< ded< dS )r   za
    An opaque object that holds a reference to an event loop.

    .. versionadded:: 4.11.0
    ztype[AsyncBackend]backend_classobjectnative_tokenN)__name__
__module____qualname____doc____annotations__r   r   r   r   r   F   s   
 r   c                  C  s   t  } |  }t| |S )z
    Return a token object that can be used to call code in the current event loop from
    another thread.

    .. versionadded:: 4.11.0

    )r   r	   r   )r    	raw_tokenr   r   r   r	   S   s   
r	   z1WeakKeyDictionary[object, dict[RunVar[Any], Any]]	_run_varsc                   @  s   e Zd Ze ZdS )_NoValueSetN)r#   r$   r%   enumautoNO_VALUE_SETr   r   r   r   r*   c   s    r*   c                   @  s.   e Zd ZdZdddZdd
dZdddZdS )r   _var_value	_redeemedvar	RunVar[T]value%T | Literal[_NoValueSet.NO_VALUE_SET]c                 C  s   || _ || _d| _d S )NFr.   )selfr2   r4   r   r   r   __init__j   s   
zRunvarToken.__init__r   RunvarToken[T]c                 C  s   | S Nr   r6   r   r   r   	__enter__o      zRunvarToken.__enter__exc_typetype[BaseException] | Noneexc_valBaseException | Noneexc_tbTracebackType | Noner   c                 C  s   | j |  d S r9   )r/   reset)r6   r=   r?   rA   r   r   r   __exit__r   s   zRunvarToken.__exit__N)r2   r3   r4   r5   )r   r8   )r=   r>   r?   r@   rA   rB   r   r   )r#   r$   r%   	__slots__r7   r;   rD   r   r   r   r   r   g   s
    

r   c                   @  s   e Zd ZU dZdZejZded< efd"d	d
Ze	d#ddZ
ed$ddZed%ddZefd&ddZd'ddZd(ddZd)dd Zd!S )*r   a  
    Like a :class:`~contextvars.ContextVar`, except scoped to the running event loop.

    Can be used as a context manager, Just like :class:`~contextvars.ContextVar`, that
    will reset the variable to its previous value when the context block is exited.
    _name_defaultz!Literal[_NoValueSet.NO_VALUE_SET]r-   namestrdefaultr5   c                 C  s   || _ || _d S r9   rF   )r6   rI   rK   r   r   r   r7      s   
zRunVar.__init__r   dict[RunVar[T], T]c                 C  s6   t  j}zt| W S  ty   i  }t|< | Y S w r9   )r	   r"   r)   KeyError)r6   r"   run_varsr   r   r   _current_vars   s   
zRunVar._current_varsr   T | Dc                 C     d S r9   r   r6   rK   r   r   r   get   r<   z
RunVar.getr   c                 C  rQ   r9   r   r:   r   r   r   rS      r<   %D | Literal[_NoValueSet.NO_VALUE_SET]c                 C  sZ   z| j |  W S  ty#   |tjur| Y S | jtjur!| j Y S Y nw td| j d)NzRun variable "z!" has no value and no default set)rO   rM   r   r-   rH   LookupErrorrG   rR   r   r   r   rS      s   

r4   r8   c                 C  s&   | j }t| || tj}||| < |S r9   )rO   r   rS   r   r-   )r6   r4   current_varstokenr   r   r   set   s   z
RunVar.setrW   r   c                 C  sb   |j | ur	td|jrtd|jtju r&z| j| = W n ty%   Y nw |j| j| < d|_d S )Nz)This token does not belong to this RunVarz This token has already been usedT)r/   
ValueErrorr1   r0   r*   r-   rO   rM   )r6   rW   r   r   r   rC      s   

zRunVar.resetc                 C  s   d| j dS )Nz<RunVar name=>)rG   r:   r   r   r   __repr__   s   zRunVar.__repr__N)rI   rJ   rK   r5   )r   rL   )rK   r   r   rP   )r   r   )rK   rT   r   rP   )r4   r   r   r8   )rW   r8   r   r   )r   rJ   )r#   r$   r%   r&   rE   r*   r-   r'   r7   propertyrO   r   rS   rX   rC   r[   r   r   r   r   r   {   s"   
 

r   )r   r   )r   r   )"
__future__r   __all__r+   dataclassesr
   typesr   typingr   r   r   r   r   r   weakrefr   _core._eventloopr   abcr   r   r   r   r   r   r   r	   r)   r'   Enumr*   r   r   r   r   r   r   <module>   s,    
 




